update39

app.py

@@ -44,8 +44,16 @@ def queue_processor():
         try:
             task_id, input_data, request_time = task_queue.get(timeout=0.1)
             with lock:
-                task_status[task_id]['status'] = 'processing'
-                task_status[task_id]['start_time'] = time.time()
+                if task_id in task_status:
+                    task_status[task_id]['status'] = 'processing'
+                    task_status[task_id]['start_time'] = time.time()
+                else:
+                    # Create task status entry if it doesn't exist
+                    task_status[task_id] = {
+                        'status': 'processing',
+                        'queued_time': request_time,
+                        'start_time': time.time()
+                    }
             
             if isinstance(input_data, list) and len(input_data) > 0:
                 sample_task = input_data[0]
@@ -99,9 +107,16 @@ def queue_processor():
         except Exception as e:
             if 'task_id' in locals():
                 with lock:
-                    task_status[task_id]['status'] = 'error'
-                    task_status[task_id]['error'] = str(e)
-                    task_status[task_id]['end_time'] = time.time()
+                    if task_id in task_status:
+                        task_status[task_id]['status'] = 'error'
+                        task_status[task_id]['error'] = str(e)
+                        task_status[task_id]['end_time'] = time.time()
+                    else:
+                        task_status[task_id] = {
+                            'status': 'error',
+                            'error': str(e),
+                            'end_time': time.time()
+                        }
             task_queue.task_done()
 
 def _estimate_task_complexity(tasks):
@@ -233,6 +248,7 @@ def synchronous_evaluate(input_data):
     task_id = str(uuid.uuid4())
     request_time = time.time()
     
+    # Add task to queue and update UI first
     with lock:
         task_status[task_id] = {
             'status': 'queued',
@@ -247,19 +263,32 @@ def synchronous_evaluate(input_data):
             'estimated_time': estimated_total_time
         }
     
+    # Show task in UI for at least 1 second to ensure visibility
+    time.sleep(1)
+    
+    # Add to queue for processing
     task_queue.put((task_id, input_data, request_time))
     
+    # Wait for task to complete
     while True:
         with lock:
             if task_id in task_status:
                 status = task_status[task_id]['status']
                 if status == 'completed':
                     result = task_status[task_id]['result']
-                    task_status.pop(task_id, None)
+                    # Keep the result in status for a short time to ensure it shows in history
+                    if 'end_time' not in task_status[task_id]:
+                        task_status[task_id]['end_time'] = time.time()
+                    elif time.time() - task_status[task_id]['end_time'] > 5:
+                        task_status.pop(task_id, None)
                     return result
                 elif status == 'error':
                     error = task_status[task_id].get('error', 'Unknown error')
-                    task_status.pop(task_id, None)
+                    # Keep the error in status for a short time to ensure it shows in history
+                    if 'end_time' not in task_status[task_id]:
+                        task_status[task_id]['end_time'] = time.time()
+                    elif time.time() - task_status[task_id]['end_time'] > 5:
+                        task_status.pop(task_id, None)
                     return {"status": "Exception", "error": error}
         
         time.sleep(0.1)
@@ -654,13 +683,19 @@ with gr.Blocks(css=custom_css) as demo:
     with gr.Row(visible=False):
         api_input = gr.JSON()
         api_output = gr.JSON()
+        
+        async_api_input = gr.JSON()
+        async_api_output = gr.JSON()
+        
+        status_check_input = gr.Textbox()
+        status_check_output = gr.JSON()
     
     # Define update function
     def update_queue_info():
         return ui_get_queue_info()
     
     # Update queue info periodically
-    demo.load(update_queue_info, None, queue_info_html, every=3)
+    demo.load(update_queue_info, None, queue_info_html, every=1)
     
     # Refresh button event
     refresh_queue_btn.click(update_queue_info, None, queue_info_html)
@@ -668,6 +703,12 @@ with gr.Blocks(css=custom_css) as demo:
     # Add evaluation endpoint compatible with original interface
     demo.queue()
     evaluate_endpoint = demo.load(fn=synchronous_evaluate, inputs=api_input, outputs=api_output, api_name="evaluate")
+    
+    # Add async evaluation endpoint
+    enqueue_endpoint = demo.load(fn=enqueue_task, inputs=async_api_input, outputs=async_api_output, api_name="enqueue")
+    
+    # Add status check endpoint
+    status_endpoint = demo.load(fn=check_status, inputs=status_check_input, outputs=status_check_output, api_name="status")
 
 if __name__ == "__main__":
     try: