Refactor API integration and update model details in app.py

app.py

@@ -2,20 +2,30 @@
 import os
 import requests
 import gradio as gr
+from dotenv import load_dotenv
 
-API_URL = "https://router.huggingface.co/together/v1/chat/completions"
+load_dotenv()
+token = os.getenv("TOGETHER_API_KEY")
+
+API_URL = "https://api.together.xyz/v1/chat/completions"
 headers = {
-    "Authorization": f"Bearer {os.environ['huggingface_token']}",
+    "Authorization": f"Bearer {token}",
 }
 
 def query_api(message):
     payload = {
         "messages": [{"role": "user", "content": message}],
-        "model": "mistralai/Mistral-7B-Instruct-v0.3"
+        "model": "mistralai/Mixtral-8x7B-Instruct-v0.1"
     }
     response = requests.post(API_URL, headers=headers, json=payload)
     try:
-        return response.json()["choices"][0]["message"]["content"]
+        result = response.json()
+        if "choices" in result:
+            return result["choices"][0]["message"]["content"]
+        elif "error" in result:
+            return f"Error: {result['error']}"
+        else:
+            return str(result)
     except Exception as e:
         return f"Error: {e}\n\nFull response:\n{response.text}"
 
@@ -23,7 +33,7 @@ demo = gr.Interface(
     fn=query_api,
     inputs=gr.Textbox(label="Ask something"),
     outputs="text",
-    title="Chat with AI Co-Pilot (Mistral 7B Instruct)",
+    title="Chat with AI Co-Pilot (Llama 3.1)",
 )
 
 if __name__ == "__main__":