Update app.py

app.py

@@ -25,6 +25,7 @@ API_ENDPOINT = "https://api-st.siliconflow.cn/v1/user/info"
 TEST_MODEL_ENDPOINT = "https://api-st.siliconflow.cn/v1/chat/completions"
 MODELS_ENDPOINT = "https://api-st.siliconflow.cn/v1/models"
 EMBEDDINGS_ENDPOINT = "https://api-st.siliconflow.cn/v1/embeddings"
+IMAGE_ENDPOINT = "https://api-st.siliconflow.cn/v1/images/generations"
 
 app = Flask(__name__)
 app.wsgi_app = ProxyFix(app.wsgi_app, x_for=1)
@@ -534,6 +535,8 @@ def handsome_embeddings():
     data = request.get_json()
     if not data or 'model' not in data:
         return jsonify({"error": "Invalid request data"}), 400
+    if data['model'] not in embedding_models:
+        return jsonify({"error": "Invalid model"}), 400
 
     model_name = data['model']
     request_type = determine_request_type(
@@ -613,6 +616,8 @@ def handsome_images_generations():
     data = request.get_json()
     if not data or 'model' not in data:
         return jsonify({"error": "Invalid request data"}), 400
+    if data['model'] not in image_models:
+        return jsonify({"error": "Invalid model"}), 400
 
     model_name = data.get('model')
     
@@ -701,7 +706,7 @@ def handsome_images_generations():
         try:
             start_time = time.time()
             response = requests.post(
-                "https://api-st.siliconflow.cn/v1/images/generations",
+                IMAGE_ENDPOINT,
                 headers=headers,
                 json=siliconflow_data,
                 timeout=120
@@ -904,7 +909,7 @@ def handsome_chat_completions():
         try:
             start_time = time.time()
             response = requests.post(
-                "https://api.siliconflow.cn/v1/images/generations",
+                IMAGE_ENDPOINT,
                 headers=headers,
                 json=siliconflow_data,
                 timeout=120,