update55

app.py

@@ -374,13 +374,25 @@ def check_status(task_id):
 def get_queue_status():
     """Get queue status"""
     with lock:
+        # Get all tasks that are either queued or processing
         queued_tasks = [t for t in task_status.values() if t['status'] == 'queued']
         processing_tasks = [t for t in task_status.values() if t['status'] == 'processing']
         
-        queue_size = len(queued_tasks)  # Use actual count of queued tasks
+        # Get the actual queue size from the queue object
+        actual_queue_size = task_queue.qsize()
+        
+        # Use the larger of actual queue size or queued tasks count
+        queue_size = max(actual_queue_size, len(queued_tasks))
         active_tasks = len(processing_tasks)
         waiting_tasks = queue_size
         
+        # Clean up completed or error tasks that are older than 1 hour
+        current_time = time.time()
+        for task_id in list(task_status.keys()):
+            if task_status[task_id]['status'] in ['completed', 'error']:
+                if current_time - task_status[task_id].get('end_time', 0) > 3600:
+                    task_status.pop(task_id, None)
+        
         remaining_processing_time = 0
         for task in processing_tasks:
             if 'start_time' in task and 'estimated_time' in task: