Update main.py

main.py

@@ -407,17 +407,21 @@ async def process_image_generation(msg_data: str, session: requests.Session, hea
         job_id, prompt, negative = match.groups()
         print(f"Starting image generation process for job_id: {job_id}")
         
-        # 发送思考开始的消息
-        start_time = time.time()
+        # 构建思考消息的开始部分
         think_msg = "<think>\n"
         think_msg += "🎨 Generating image...\n\n"
         think_msg += f"Prompt: {prompt}\n"
         
+        # 记录开始时间（在检查图片状态前）
+        start_time = time.time()
+        
         # 检查图片状态和上传
         result = await check_image_status(session, job_id, headers)
         
-        # 完成思考部分
+        # 计算实际经过的时间
         elapsed_time = time.time() - start_time
+        
+        # 完成思考部分
         think_msg += f"\n🤔 Thinking for {elapsed_time:.1f}s...\n"
         think_msg += "</think>"