Update app.py

app.py

@@ -18,18 +18,7 @@ SPECIAL_API_KEY = os.getenv("SPECIAL_API_KEY")
 ENDPOINT = os.getenv("ENDPOINT")
 SYSTEM_PROMPT = os.getenv("SYSTEM_PROMPT")
 EXCEPTION = os.getenv("EXCEPTION")
-AVAILABLE_MODELS = [
-    "deepseek-r1",
-    "gpt-4-turbo",
-    "gpt-4",
-    "gpt-3.5-turbo",
-    "mistral-7b",
-    "gemini-pro",
-    "claude-3-sonnet",
-    "claude-3-haiku",
-    "llama-3-8b",
-    "llama-3-70b"
-]
+AVAILABLE_MODELS = ["gpt-3.5-turbo","gpt-3.5-turbo-202201","gpt-4o","gpt-4o-2024-05-13","o1-preview","chatgpt-4o-latest","claude-3-5-sonnet","claude-sonnet-3.5","claude-3-5-sonnet-20240620","deepseek-r1","deepseek-llm-67b-chat","llama-3.1-405b","llama-3.1-70b","llama-3.1-8b","meta-llama/Llama-3.2-90B-Vision-Instruct","meta-llama/Meta-Llama-3.1-70B-Instruct-Turbo","meta-llama/Meta-Llama-3.1-8B-Instruct","meta-llama/Meta-Llama-3.1-8B-Instruct-Turbo","Meta-Llama-3.1-405B-Instruct-Turbo","Meta-Llama-3.3-70B-Instruct-Turbo","mistral","mistral-large","mistralai/Mixtral-8x22B-Instruct-v0.1","Qwen/Qwen2.5-72B-Instruc","Qwen/Qwen2.5-Coder-32B-Instruct","Qwen-QwQ-32B-Preview","gemini-pro","gemini-1.5-pro","gemini-1.5-pro-latest","gemini-1.5-flash","blackboxai","blackboxai-pro","openchat/openchat-3.6-8b","dbrx-instruct","Nous-Hermes-2-Mixtral-8x7B-DPO"]
 
 # Track API statistics
 api_usage = {}  # Stores {api_key: {"count": X, "reset_time": timestamp}}
@@ -193,6 +182,9 @@ def chat():
     top_p = data.get("top_p", 1)
     max_tokens = data.get("max_tokens", 4000)
 
+    if model not in AVAILABLE_MODELS:
+        return jsonify({"error": f"Invalid Model. Available Models are: {', '.join(AVAILABLE_MODELS)}"}), 404
+    
     api_payload = {
         "messages": [
             {'role': 'system', 'content': SYSTEM_PROMPT},