Update main.py

main.py

@@ -429,9 +429,10 @@ async def process_image_generation(msg_data: str, session: requests.Session, hea
         # 发送结束消息
         elapsed_time = time.time() - start_time
         end_msg += f"\n🤔 Thinking for {elapsed_time:.1f}s...\n"
-        end_msg += "</think>\n\n"
+        end_msg += "</think>\n\n"  # 确保思考模块正确关闭
         
         if result:  # result 现在是上传后的图片URL
+            # 将图片放在思考模块外部
             end_msg += f"![Generated Image]({result})"
         else:
             end_msg += "*Image generation or upload failed.*\n"