Update app.py

app.py

@@ -585,6 +585,8 @@ def handsome_chat_completions():
     }
     
     if model_name in image_models:
+        # Handle image generation
+        # Map OpenAI-style parameters to SiliconFlow's parameters
         siliconflow_data = {
             "model": model_name,
             "prompt": data.get("messages", [{}])[0].get("content") if isinstance(data.get("messages"), list) else "",
@@ -675,6 +677,7 @@ def handsome_chat_completions():
 
                             for chunk in image_response.iter_content(chunk_size=1024):
                                 if chunk:
+                                    base64_chunk = base64.b64encode(chunk).decode('utf-8')
                                     chunk_data = {
                                         "id": f"chatcmpl-{uuid.uuid4()}",
                                         "object": "chat.completion.chunk",
@@ -685,7 +688,7 @@ def handsome_chat_completions():
                                                 "index": 0,
                                                 "delta": {
                                                     "role": "assistant",
-                                                    "content": chunk.decode('latin-1',errors='ignore')
+                                                    "content": base64_chunk
                                                 },
                                                 "finish_reason": None
                                             }
@@ -723,7 +726,7 @@ def handsome_chat_completions():
 
                             with data_lock:
                                 request_timestamps.append(time.time())
-                                token_counts.append(0)
+                                token_counts.append(0)  # Image generation doesn't use tokens
 
                         else:
                             chunk_data = {
@@ -803,6 +806,7 @@ def handsome_chat_completions():
                 try:
                     images = response_json.get("images", [])
                     
+                    # Extract the first URL if available
                     image_url = ""
                     if images and isinstance(images[0], dict) and "url" in images[0]:
                         image_url = images[0]["url"]
@@ -859,7 +863,7 @@ def handsome_chat_completions():
 
                 with data_lock:
                     request_timestamps.append(time.time())
-                    token_counts.append(0)
+                    token_counts.append(0)  # Image generation doesn't use tokens
 
                 return jsonify(response_data)
         except requests.exceptions.RequestException as e: