Update app.py

app.py

@@ -581,6 +581,46 @@ def handsome_chat_completions():
         "Content-Type": "application/json"
     }
     
+    if model_name in image_models:
+        # Handle image generation
+        # Map OpenAI-style parameters to SiliconFlow's parameters
+        siliconflow_data = {
+            "model": model_name,
+        @app.route('/handsome/v1/chat/completions', methods=['POST'])
+def handsome_chat_completions():
+    if not check_authorization(request):
+        return jsonify({"error": "Unauthorized"}), 401
+
+    data = request.get_json()
+    if not data or 'model' not in data:
+        return jsonify({"error": "Invalid request data"}), 400
+
+    model_name = data['model']
+    
+    request_type = determine_request_type(
+        model_name,
+        text_models + image_models,
+        free_text_models + free_image_models
+    )
+    
+    api_key = select_key(request_type, model_name)
+
+    if not api_key:
+        return jsonify(
+            {
+                "error": (
+                    "No available API key for this "
+                    "request type or all keys have "
+                    "reached their limits"
+                )
+            }
+        ), 429
+
+    headers = {
+        "Authorization": f"Bearer {api_key}",
+        "Content-Type": "application/json"
+    }
+    
     if model_name in image_models:
         # Handle image generation
         # Map OpenAI-style parameters to SiliconFlow's parameters