Update app.py

app.py

@@ -625,15 +625,7 @@ def handsome_images_generations():
     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
+        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}",
@@ -790,6 +782,8 @@ def handsome_chat_completions():
     data = request.get_json()
     if not data or 'model' not in data:
         return jsonify({"error": "Invalid request data"}), 400
+    if data['model'] not in text_models and data['model'] not in image_models:
+        return jsonify({"error": "Invalid model"}), 400
 
     model_name = data['model']