Update app.py

app.py

@@ -228,6 +228,12 @@ def handle_request():
         if not model or not messages or len(messages) == 0:
             return jsonify({"error": "Bad Request: Missing required fields"}), 400
         
+        # 映射 model
+        if model in MODEL_MAPPING:
+            mapped_model = MODEL_MAPPING[model]['mapping']
+        else:
+            return jsonify({"error": f"Model '{model}' not found"}), 400
+        
         prompt = messages[-1]['content']
         image_size, clean_prompt, use_original, size_param = extract_params_from_prompt(prompt)
         
@@ -241,7 +247,7 @@ def handle_request():
         else:
             enhanced_prompt = translate_and_enhance_prompt(clean_prompt, random_token)
         
-        new_url = f'https://api.siliconflow.cn/v1/{model}/text-to-image'
+        new_url = f'https://api.siliconflow.cn/v1/{mapped_model}/text-to-image'
         new_request_body = {
             "prompt": enhanced_prompt,
             "image_size": image_size,
@@ -279,7 +285,7 @@ def handle_request():
             params.append("-o")
         params_str = " ".join(params) if params else "no params"
         
-        app.logger.info(f"Status: 200 - Token: {random_token} - Model: {model} - Params: {params_str} - Image URL: {image_url}")
+        app.logger.info(f"Status: 200 - Token: {random_token} - Model: {mapped_model} - Params: {params_str} - Image URL: {image_url}")
         
         if stream:
             return stream_response(unique_id, image_data, clean_prompt, enhanced_prompt, image_size, current_timestamp, model, system_fingerprint, use_original)