update12

app.py

@@ -38,7 +38,7 @@ def queue_processor():
     """处理队列中的任务"""
     while running:
         try:
-            # 从队列中获取任务，如果队列为空等待0.1秒
+            # 从队列中获取任务，如果队列为空等待0.1
             task_id, input_data, request_time = task_queue.get(timeout=0.1)
             with lock:
                 task_status[task_id]['status'] = 'processing'
@@ -276,16 +276,35 @@ def get_queue_status():
         waiting_tasks = sum(1 for status in task_status.values() if status['status'] == 'queued')
         
         recent_tasks = task_history[-5:] if task_history else []
-        avg_time = 0
+        
+        # 计算平均处理时间，如果没有历史数据，使用默认值
         if recent_tasks:
             avg_time = sum(task['process_time'] for task in recent_tasks) / len(recent_tasks)
+            # 确保平均时间至少为1秒
+            avg_time = max(avg_time, 1.0)
+        else:
+            # 没有历史数据时的默认平均处理时间（秒）
+            avg_time = 5.0
+        
+        # 估算等待时间
+        # 如果有活跃任务，假设它们还需要完成一半的平均时间
+        active_time = (active_tasks * avg_time) / 2 if active_tasks > 0 else 0
+        # 排队任务的总时间
+        queue_time = waiting_tasks * avg_time
+        # 总预计等待时间
+        estimated_wait = active_time + queue_time
+        
+        # 确保即使没有任务也返回有效值
+        if estimated_wait == 0 and (waiting_tasks > 0 or active_tasks > 0):
+            estimated_wait = 1.0
             
         return {
             'queue_size': queue_size,
             'active_tasks': active_tasks,
             'waiting_tasks': waiting_tasks,
             'worker_threads': worker_threads,
-            'estimated_wait': waiting_tasks * avg_time if avg_time > 0 else waiting_tasks * 5,
+            'estimated_wait': estimated_wait,
+            'avg_process_time': avg_time,
             'recent_tasks': recent_tasks
         }
 
@@ -356,7 +375,16 @@ def ui_check_status(task_id):
     
     if status['status'] == 'queued':
         queue_info = get_queue_status()
-        est_wait = queue_info['estimated_wait'] / queue_info['waiting_tasks'] * status['queue_position'] if queue_info['waiting_tasks'] > 0 else 0
+        
+        # 计算这个特定任务的等待时间
+        avg_time = queue_info.get('avg_process_time', 5.0)
+        active_time = (queue_info['active_tasks'] * avg_time) / 2 if queue_info['active_tasks'] > 0 else 0
+        
+        # 计算此任务前面的任务数
+        tasks_ahead = status['queue_position'] - 1
+        
+        # 此任务的估计等待时间 = 活跃任务完成时间 + 前面排队任务时间
+        est_wait = active_time + (tasks_ahead * avg_time)
         
         return f"""
         <div class="status-card">
@@ -425,6 +453,9 @@ def ui_get_queue_info():
         </tr>
         """
     
+    # 添加平均处理时间显示
+    avg_time_html = f"""<p><b>平均处理时间:</b> {format_time(queue_info.get('avg_process_time', 0))}</p>"""
+    
     return f"""
     <div class="queue-info-card">
         <h3>队列状态</h3>
@@ -444,6 +475,7 @@ def ui_get_queue_info():
         </div>
         
         <div class="wait-time">
+            {avg_time_html}
             <p><b>当前预计等待时间:</b> {format_time(queue_info['estimated_wait'])}</p>
         </div>