update7

app.py

@@ -20,13 +20,18 @@ if current_dir not in sys.path:
 if src_dir not in sys.path:
     sys.path.append(src_dir)
 
-# 创建全局任务队列和任务状态跟踪
+# 添加全局任务队列和任务状态跟踪
 task_queue = queue.Queue()
 task_results = {}
 active_tasks = {}
 completed_tasks = []
 task_id_counter = 0
 task_lock = threading.Lock()
+update_event = threading.Event()  # 通知UI需要刷新
+
+def trigger_ui_update():
+    """触发UI更新事件"""
+    update_event.set()
 
 def get_next_task_id():
     global task_id_counter
@@ -61,6 +66,9 @@ def submit_task(input_data):
             active_tasks[task_id] = task_info
             task_queue.put(task_info)
             
+            # 触发UI更新
+            trigger_ui_update()
+            
         # 如果这是第一个任务，启动处理线程
         if len(active_tasks) == 1:
             threading.Thread(target=process_task_queue, daemon=True).start()
@@ -108,6 +116,7 @@ def process_task_queue():
             # 更新任务状态
             with task_lock:
                 active_tasks[task_id]["status"] = "processing"
+                trigger_ui_update()  # 状态变更为处理中时更新UI
             
             # 处理任务
             result = evaluate(task_info["data"])
@@ -125,11 +134,26 @@ def process_task_queue():
                 # 保留最近的20个已完成任务
                 if len(completed_tasks) > 20:
                     completed_tasks.pop(0)
+                
+                trigger_ui_update()  # 任务完成时更新UI
             
             task_queue.task_done()
             
         except Exception as e:
             print(f"Error processing task queue: {str(e)}")
+            # 出错时也触发UI更新
+            with task_lock:
+                if task_id in active_tasks:
+                    active_tasks[task_id]["status"] = "error"
+                    active_tasks[task_id]["error"] = str(e)
+                    active_tasks[task_id]["completed_at"] = datetime.now()
+                    # 将任务移至已完成列表
+                    completed_tasks.append(active_tasks[task_id])
+                    del active_tasks[task_id]
+                    # 保留最近的20个已完成任务
+                    if len(completed_tasks) > 20:
+                        completed_tasks.pop(0)
+                trigger_ui_update()
             time.sleep(1)
 
 def evaluate(input_data):
@@ -380,6 +404,13 @@ def refresh_ui():
     """定期刷新UI函数"""
     return render_queue_status()
 
+def check_for_updates():
+    """检查是否有UI更新请求"""
+    if update_event.is_set():
+        update_event.clear()
+        return render_queue_status()
+    return gr.update()
+
 def submit_json_data(json_data):
     """提交JSON数据处理接口"""
     try:
@@ -390,9 +421,12 @@ def submit_json_data(json_data):
             data = json_data
             
         result = submit_task(data)
-        return json.dumps(result, ensure_ascii=False, indent=2)
+        response = json.dumps(result, ensure_ascii=False, indent=2)
+        
+        # 返回任务提交结果和最新的队列状态
+        return response, render_queue_status()
     except Exception as e:
-        return json.dumps({"status": "error", "message": str(e)}, ensure_ascii=False, indent=2)
+        return json.dumps({"status": "error", "message": str(e)}, ensure_ascii=False, indent=2), render_queue_status()
 
 # 创建Gradio接口
 with gr.Blocks(title="代码评估服务", theme=gr.themes.Soft()) as demo:
@@ -405,10 +439,14 @@ with gr.Blocks(title="代码评估服务", theme=gr.themes.Soft()) as demo:
         status_html = gr.HTML(render_queue_status)
         refresh_button = gr.Button("刷新状态")
         refresh_button.click(fn=refresh_ui, outputs=status_html)
-        # 使用JavaScript实现自动刷新，代替不兼容的every参数
+        
+        # 设置自动刷新 - 使用Gradio内置的轮询功能
+        demo.load(check_for_updates, None, status_html, every=1)
+        
+        # JavaScript刷新功能保留作为备份
         gr.HTML("""
         <script>
-            // 使用JavaScript实现自动刷新
+            // 使用JavaScript作为备份的自动刷新
             function setupAutoRefresh() {
                 const refreshInterval = 5000; // 5秒
                 
@@ -447,7 +485,7 @@ with gr.Blocks(title="代码评估服务", theme=gr.themes.Soft()) as demo:
             with gr.Column():
                 result_output = gr.Textbox(label="提交结果", lines=5)
                 
-        submit_button.click(fn=submit_json_data, inputs=json_input, outputs=result_output)
+        submit_button.click(fn=submit_json_data, inputs=json_input, outputs=[result_output, status_html])
     
     with gr.Tab("API文档"):
         gr.Markdown("""