Update app.py

app.py

@@ -5,21 +5,16 @@ from typing import List, Tuple
 
 # Define available models
 AVAILABLE_MODELS = {
-    "DeepSeek V3": "deepseek-ai/DeepSeek-V3",
-    "Llama3.3-70b-Instruct": "meta-llama/Llama-3.3-70B-Instruct",
-    "Llama3.1-8b-Instruct": "meta-llama/Meta-Llama-3.1-8B-Instruct",
+    "Sonar Pro": "sonar-pro",
 }
 
-HYPERB_ENDPOINT_URL = "https://api.hyperbolic.xyz/v1"
-HF_ENDPOINT_URL = "https://huggingface.co/api/inference-proxy/together"
-HYPERB_API_KEY = os.getenv('HYPERBOLIC_XYZ_KEY')
-HF_API_KEY = os.getenv('HF_KEY')
+PX_ENDPOINT_URL = "https://api.perplexity.ai"
+PX_API_KEY = os.getenv('PX_KEY')
 PASSWORD = os.getenv("PASSWD")  # Store the password in an environment variable
 
 DEPLOY_TO_HF = ["deepseek-ai/DeepSeek-V3"]
 
-hyperb_client = OpenAI(base_url=HYPERB_ENDPOINT_URL, api_key=HYPERB_API_KEY)
-hf_client = OpenAI(base_url=HF_ENDPOINT_URL, api_key=HF_API_KEY)
+px_client = OpenAI(base_url=PX_ENDPOINT_URL, api_key=PX_API_KEY)
 
 def respond(
     message: str,
@@ -40,12 +35,7 @@ def respond(
 
     response = ""
     
-    if model_choice in DEPLOY_TO_HF:
-        this_client = hf_client
-    else:
-        this_client = hyperb_client
-
-    for chunk in this_client.chat.completions.create(
+    for chunk in px_client.chat.completions.create(
         model=AVAILABLE_MODELS[model_choice],  # Use the selected model
         messages=messages,
         max_tokens=max_tokens,