Handle failed connection more gracefully.

Author: akenmorris

Studio/ShapeWorksMONAI/MonaiLabelJob.cpp

@@ -73,6 +73,8 @@ void MonaiLabelJob::setCurrentSampleNumber(int n) { sample_number_ = n; }
 void MonaiLabelJob::initializeClient() {
   SW_DEBUG("Initializing MONAI Client with server: {} tmp dir: {} client_id: {}", server_url_, tmp_dir_, client_id_);
   try {
+    error_occurred_ = false;
+
     py::module monai_label = py::module::import("MONAILabel");
     py::object monai_client_class = monai_label.attr("MONAILabelClient");
     py::str py_server_url(server_url_);
@@ -89,7 +91,7 @@ void MonaiLabelJob::initializeClient() {
     }
     model_name_ = getModelName(model_type_);
     models_available_[model_type_] = {model_name_};
-    Q_EMIT triggerClientInitialized();
+    triggerClientInitialized(!error_occurred_);
   } catch (std::exception &e) {
     SW_ERROR("Error importing MONAILabel or initializing MONAILabelClient: {}", e.what());
     return;
@@ -107,10 +109,13 @@ py::dict MonaiLabelJob::getInfo() {
     response = (*monai_client_).attr("info")();
   } catch (const py::error_already_set &e) {
     SW_ERROR("Python error: {}", e.what());
+    error_occurred_ = true;
   } catch (const std::exception &e) {
     SW_ERROR("Exception occurred: {}", e.what());
+    error_occurred_ = true;
   } catch (...) {
     SW_ERROR("An unknown error occurred!");
+    error_occurred_ = true;
   }
   return response;
 }

Studio/ShapeWorksMONAI/MonaiLabelJob.h

@@ -74,7 +74,7 @@ class MonaiLabelJob : public Job {
 
  Q_SIGNALS:
   void triggerUpdateView();
-  void triggerClientInitialized();
+  void triggerClientInitialized(bool success);
   void triggerUploadSampleCompleted();
   void triggerSegmentationCompleted();
   void triggerSubmitLabelCompleted();
@@ -108,5 +108,7 @@ class MonaiLabelJob : public Job {
 
   QSharedPointer<Session> session_;
   ProjectHandle project_;
+
+  bool error_occurred_ = false;
 };
 }  // namespace monailabel

Studio/ShapeWorksMONAI/MonaiLabelTool.cpp

@@ -68,18 +68,23 @@ MonaiLabelTool::~MonaiLabelTool() {}
 //---------------------------------------------------------------------------
 void MonaiLabelTool::triggerUpdateView() { Q_EMIT update_view(); }
 
+//---------------------------------------------------------------------------
+void MonaiLabelTool::set_connect_button() {
+  Style::apply_normal_button_style(ui_->connectServerButton);
+  ui_->connectServerButton->setText("Connect Server");
+  ui_->connectServerButton->setIcon(QIcon(":/Studio/Images/connect.png"));
+  ui_->connectServerButton->setEnabled(true);
+  ui_->serverAddressField->setEnabled(true);
+}
+
 //---------------------------------------------------------------------------
 void MonaiLabelTool::onConnectServer() {
   if (tool_is_running_ && monai_label_job_) {
     ui_->connectServerButton->setText("Disconnecting...");
     monai_label_job_->abort();
     shutdown();
     SW_STATUS("Server disconnected successfully.");
-    Style::apply_normal_button_style(ui_->connectServerButton);
-    ui_->connectServerButton->setText("Connect Server");
-    ui_->connectServerButton->setIcon(QIcon(":/Studio/Images/connect.png"));
-    ui_->connectServerButton->setEnabled(true);
-    ui_->serverAddressField->setEnabled(true);
+    set_connect_button();
     enable_actions();
     session_->get_project()->save();
     monai_label_job_ = nullptr;
@@ -118,7 +123,9 @@ void MonaiLabelTool::enable_actions() {
   ui_->runSegmentationButton->setEnabled(false);
   ui_->submitLabelButton->setEnabled(false);
   std::string feature_name = MonaiLabelUtils::getFeatureName(session_);
-  if (!feature_name.empty()) session_->set_image_name(feature_name);
+  if (!feature_name.empty()) {
+    session_->set_image_name(feature_name);
+  }
 }
 
 //---------------------------------------------------------------------------
@@ -174,7 +181,12 @@ void MonaiLabelTool::runSegmentationTool() {
 }
 
 //---------------------------------------------------------------------------
-void MonaiLabelTool::handleClientInitialized() {
+void MonaiLabelTool::handleClientInitialized(bool success) {
+  if (!success) {
+    SW_LOG("Connection failed");
+    set_connect_button();
+    return;
+  }
   SW_LOG("✅ Connection successfully established to the server, continue with segmentation!");
   tool_is_running_ = true;
   if (session_->get_shapes().size() > 1) {

Studio/ShapeWorksMONAI/MonaiLabelTool.h

@@ -57,7 +57,7 @@ class MonaiLabelTool : public QWidget {
   void triggerUpdateView();
   void handle_progress(int val, QString message);
   void handleSampleNumberChanged();
-  void handleClientInitialized();
+  void handleClientInitialized(bool success);
   void handleUploadSampleCompleted();
   void handleSegmentationCompleted();
   void handleSubmitLabelCompleted();
@@ -68,6 +68,9 @@ class MonaiLabelTool : public QWidget {
   void sampleChanged();
 
  private:
+
+  void set_connect_button();
+
   Preferences& preferences_;
   Ui_MonaiLabelTool* ui_;
   QSharedPointer<Session> session_;