Update main.py

main.py

@@ -419,20 +419,22 @@ async def process_image_generation(msg_data: str, session: requests.Session, hea
         
         # 发送思考开始的消息
         start_time = time.time()
-        end_msg = "<think>\n"
-        end_msg += "🎨 Generating image...\n\n"
-        end_msg += f"Prompt: {prompt}\n"
+        think_msg = "<think>\n"
+        think_msg += "🎨 Generating image...\n\n"
+        think_msg += f"Prompt: {prompt}\n"
         
         # 检查图片状态和上传
         result = await check_image_status(session, job_id, headers)
         
-        # 发送结束消息
+        # 完成思考部分
         elapsed_time = time.time() - start_time
-        end_msg += f"\n🤔 Thinking for {elapsed_time:.1f}s...\n"
-        end_msg += "</think>\n\n"  # 确保思考模块正确关闭
+        think_msg += f"\n🤔 Thinking for {elapsed_time:.1f}s...\n"
+        think_msg += "</think>"  # 关闭思考标签
+        
+        # 构建完整消息，确保图片在思考模块外
+        end_msg = think_msg + "\n\n"  # 添加空行分隔
         
         if result:  # result 现在是上传后的图片URL
-            # 将图片放在思考模块外部
             end_msg += f"![Generated Image]({result})"
         else:
             end_msg += "*Image generation or upload failed.*\n"