Update app.py

app.py

@@ -661,7 +661,25 @@ def handsome_chat_completions():
                      }
                    ],
                 }
+                
+                if data.get("stream", False):
+                    def generate():
+                        yield f"data: {json.dumps(response_data)}\n\n"
+                        yield "data: [DONE]\n\n"
+
+                    logging.info(
+                        f"使用的key: {api_key}, "
+                        f"总共用时: {total_time:.4f}秒, "
+                        f"使用的模型: {model_name}"
+                    )
 
+                    with data_lock:
+                        request_timestamps.append(time.time())
+                        token_counts.append(0)  # Image generation doesn't use tokens
+                    return Response(
+                        stream_with_context(generate()),
+                        content_type='text/event-stream'
+                    )
             except (KeyError, ValueError, IndexError) as e:
                 logging.error(
                     f"解析响应 JSON 失败: {e}, "
@@ -683,7 +701,6 @@ def handsome_chat_completions():
                       }
                    ],
                 }
-
             logging.info(
                 f"使用的key: {api_key}, "
                 f"总共用时: {total_time:.4f}秒, "