Better status and error handling

app.py

@@ -34,6 +34,7 @@ class DynamicState:
         self.in_cot = True
         self.current_language = "en"
         self.waiting_api = False  # 新增等待状态标志
+        self.label_passthrough = False
 
     def control_button_handler(self):
         original_state = self.should_stream
@@ -55,8 +56,10 @@ class DynamicState:
         )
         control_variant = "secondary" if self.should_stream else "primary"
         # 处理等待状态显示
-        if self.waiting_api:
+        if self.waiting_api and self.should_stream:
             status_suffix = lang_data["waiting_api"]
+        elif self.waiting_api and not self.should_stream:
+            status_suffix = lang_data["api_retry"]
         else:
             status_suffix = (
                 lang_data["completed"]
@@ -64,22 +67,24 @@ class DynamicState:
                 else lang_data["interrupted"]
             )
         editor_label = f"{lang_data['editor_label']} - {status_suffix}"
-
-        return (
+        output = (
             gr.update(value=control_value, variant=control_variant),
-            gr.update(label=editor_label),
+            gr.update() if self.label_passthrough else gr.update(label=editor_label),
             gr.update(interactive=not self.should_stream),
         )
-
-    def reset_workspace(self):
+        self.label_passthrough = False
+        return output
+    
+    def reset_workspace(self, chatbot_value=None):
         """重置工作区状态"""
         self.stream_completed = False
         self.should_stream = False
         self.in_cot = True
+        self.waiting_api = False
         return self.ui_state_controller() + (
             "",
             "",
-            LANGUAGE_CONFIG["en"]["bot_default"],
+            LANGUAGE_CONFIG["en"]["bot_default"] if chatbot_value is None else chatbot_value,
         )
 
 
@@ -114,6 +119,7 @@ class ConvoState:
         self.current_language = "en"
         self.convo = []
         self.initialize_new_round()
+        self.is_error = False
 
     def initialize_new_round(self):
         self.current = {}
@@ -217,18 +223,18 @@ class ConvoState:
                         time.sleep(0.005)
 
         except Exception as e:
-            error_msg = LANGUAGE_CONFIG[self.current_language].get("error", "Error")
-            full_response += f"\n\n[{error_msg}: {str(e)}]"
-            editor_label = f"{lang_data['editor_label']} - {error_msg}"
-            yield full_response, gr.update(
-                label=editor_label
-            ), self.flatten_output() + [
-                {
-                    "role": "assistant",
-                    "content": error_msg,
-                    "metadata": {"title": f"❌Error"},
-                }
-            ]
+            if str(e) == "list index out of range":
+                dynamic_state.stream_completed = True                
+            else:
+                if str(e) == "The read operation timed out":
+                    error_msg = lang_data["api_interrupted"]
+                else:
+                    error_msg = "❓ " + str(e)
+                # full_response += f"\n\n[{error_msg}: {str(e)}]"
+                editor_label_error = f"{lang_data['editor_label']} - {error_msg}"
+                self.is_error = True
+                dynamic_state.label_passthrough = True
+
 
         finally:
             dynamic_state.should_stream = False
@@ -240,7 +246,18 @@ class ConvoState:
                 else lang_data["interrupted"]
             )
             editor_label = f"{lang_data['editor_label']} - {final_status}"
-            yield full_response, gr.update(label=editor_label), self.flatten_output()
+            if not self.is_error:
+                yield full_response, gr.update(label=editor_label), self.flatten_output()
+            else:
+                yield full_response, gr.update(label=editor_label_error), self.flatten_output() + [
+                {
+                    "role": "assistant",
+                    "content": error_msg,
+                    "metadata": {"title": f"❌Error"},
+                }
+            ]
+            self.is_error =  False
+            
 
 
 def update_interface_language(selected_lang, convo_state, dynamic_state):
@@ -291,6 +308,14 @@ with gr.Blocks(theme=theme, css_paths="styles.css") as demo:
     convo_state = gr.State(ConvoState)
     dynamic_state = gr.State(DynamicState)
 
+    bot_default = LANGUAGE_CONFIG["en"]["bot_default"] + [
+                    {
+                        "role": "assistant",
+                        "content": f"Running `{os.getenv('API_MODEL')}` @ {os.getenv('API_URL')}  \n Proformance subjects to API provider situation",
+                        "metadata": {"title": f"API INFO"},
+                    }
+                ]
+
     with gr.Row(variant=""):
         title_md = gr.Markdown(
             f"## {LANGUAGE_CONFIG['en']['title']} \n GitHub: https://github.com/Intelligent-Internet/CoT-Lab-Demo",
@@ -332,14 +357,7 @@ with gr.Blocks(theme=theme, css_paths="styles.css") as demo:
             chatbot = gr.Chatbot(
                 type="messages",
                 height=300,
-                value=LANGUAGE_CONFIG["en"]["bot_default"]
-                + [
-                    {
-                        "role": "assistant",
-                        "content": f"Running `{os.getenv('API_MODEL')}` @ {os.getenv('API_URL')}  \n Proformance subjects to API provider situation",
-                        "metadata": {"title": f"API INFO"},
-                    }
-                ],
+                value=bot_default,
                 group_consecutive_messages=False,
                 show_copy_all_button=True,
                 show_share_button=True,
@@ -409,10 +427,11 @@ with gr.Blocks(theme=theme, css_paths="styles.css") as demo:
     )
 
     next_turn_btn.click(
-        lambda d: d.reset_workspace(),
+        lambda d: d.reset_workspace(bot_default),
         [dynamic_state],
         stateful_ui + (thought_editor, prompt_input, chatbot),
         concurrency_limit=None,
+        show_progress=False
     )
 
     lang_selector.change(

lang.py

@@ -33,7 +33,10 @@ LANGUAGE_CONFIG = {
             },
         ],
         "editor_default": "AI thought will start with this, leave blank to think freely",
-        "waiting_api": "⏳ Waiting for Deepseek API response",
+        "waiting_api": "⏳ Waiting for API response",
+        "api_retry": "🔁 API no response, hit Shift+Enter to try again.",
+        "api_interrupted": "⚠️ Pasued, API connection interrupted. Hit Shift+Enter to reconnect"
+
     },
     "zh": {
         "title": "CoT-Lab: 人机协同思维实验室\n在一轮对话中跟随、学习、迭代思维链。克隆Space并使用自己的API KEY可以获得更好的体验。",
@@ -69,6 +72,8 @@ LANGUAGE_CONFIG = {
             {"role": "assistant", "content": "**Shift+Enter** 可以暂停/继续AI生成"},
         ],
         "editor_default": "AI思维会以此开头，留空即为默认思考",
-        "waiting_api": "⏳ 等待Deepseek API响应",
+        "waiting_api": "⏳ 等待API响应",
+        "api_retry": "🔁 API无响应, Shift+Enter 重试一次试试？",
+        "api_interrupted": "⚠️ 暂停，API连接意外中断，Shift+Enter 可重连"
     },
 }