Update main.py

main.py

@@ -407,8 +407,10 @@ 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"
@@ -416,8 +418,10 @@ async def process_image_generation(msg_data: str, session: requests.Session, hea
         # 检查图片状态和上传
         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>"