Update main.py

main.py

@@ -410,17 +410,40 @@ async def process_image_generation(msg_data: str, session: requests.Session, hea
         # 返回多个消息块
         messages = []
         
-        # 第一个消息块：开始思考
-        start_msg = "<think>\n🎨 Generating image...\n\n"
-        start_msg += f"Prompt: {prompt}\n"
+        # 发送思考开始标记
+        messages.append({
+            "id": f"chatcmpl-{chat_id}-1",
+            "object": "chat.completion.chunk",
+            "created": int(time.time()),
+            "model": "AkashGen",
+            "choices": [{
+                "delta": {"content": "<think>\n"},
+                "index": 0,
+                "finish_reason": None
+            }]
+        })
+        
+        # 发送生成图片的提示
+        messages.append({
+            "id": f"chatcmpl-{chat_id}-2",
+            "object": "chat.completion.chunk",
+            "created": int(time.time()),
+            "model": "AkashGen",
+            "choices": [{
+                "delta": {"content": "🎨 Generating image...\n\n"},
+                "index": 0,
+                "finish_reason": None
+            }]
+        })
         
+        # 发送提示词
         messages.append({
-            "id": f"chatcmpl-{chat_id}-start",
+            "id": f"chatcmpl-{chat_id}-3",
             "object": "chat.completion.chunk",
             "created": int(time.time()),
             "model": "AkashGen",
             "choices": [{
-                "delta": {"content": start_msg},
+                "delta": {"content": f"Prompt: {prompt}\n\n"},
                 "index": 0,
                 "finish_reason": None
             }]
@@ -435,44 +458,53 @@ async def process_image_generation(msg_data: str, session: requests.Session, hea
         # 计算实际经过的时间
         elapsed_time = time.time() - start_time
         
-        # 第二个消息块：结束思考
-        end_think_msg = f"\n🤔 Thinking for {elapsed_time:.1f}s...\n</think>"
+        # 发送思考时间
+        messages.append({
+            "id": f"chatcmpl-{chat_id}-4",
+            "object": "chat.completion.chunk",
+            "created": int(time.time()),
+            "model": "AkashGen",
+            "choices": [{
+                "delta": {"content": f"🤔 Thinking for {elapsed_time:.1f}s...\n"},
+                "index": 0,
+                "finish_reason": None
+            }]
+        })
         
+        # 发送思考结束标记
         messages.append({
-            "id": f"chatcmpl-{chat_id}-end-think",
+            "id": f"chatcmpl-{chat_id}-5",
             "object": "chat.completion.chunk",
             "created": int(time.time()),
             "model": "AkashGen",
             "choices": [{
-                "delta": {"content": end_think_msg},
+                "delta": {"content": "</think>\n\n"},
                 "index": 0,
                 "finish_reason": None
             }]
         })
         
-        # 第三个消息块：图片结果
+        # 发送图片结果
         if result:
-            image_msg = f"\n\n![Generated Image]({result})"
             messages.append({
-                "id": f"chatcmpl-{chat_id}-image",
+                "id": f"chatcmpl-{chat_id}-6",
                 "object": "chat.completion.chunk",
                 "created": int(time.time()),
                 "model": "AkashGen",
                 "choices": [{
-                    "delta": {"content": image_msg},
+                    "delta": {"content": f"![Generated Image]({result})"},
                     "index": 0,
                     "finish_reason": None
                 }]
             })
         else:
-            fail_msg = "\n\n*Image generation or upload failed.*"
             messages.append({
-                "id": f"chatcmpl-{chat_id}-fail",
+                "id": f"chatcmpl-{chat_id}-6",
                 "object": "chat.completion.chunk",
                 "created": int(time.time()),
                 "model": "AkashGen",
                 "choices": [{
-                    "delta": {"content": fail_msg},
+                    "delta": {"content": "*Image generation or upload failed.*"},
                     "index": 0,
                     "finish_reason": None
                 }]