Update app.py

app.py

@@ -1,12 +1,15 @@
 from flask import Flask, request, jsonify
 import hmac, hashlib, secrets, time
-import requests
+from openai import OpenAI
 import os
 
 app = Flask(__name__)
 
 # 🔑 Secret key for API authentication (Load from environment in production)
 SECRET_KEY = os.getenv("SECRET_KEY")
+GITHUB_TOKEN = os.getenv("GITHUB_TOKEN")
+endpoint = "https://models.inference.ai.azure.com"
+client = OpenAI(base_url=endpoint,api_key=GITHUB_TOKEN)
 
 # Track API statistics
 request_count = 0
@@ -51,16 +54,13 @@ def validate_api_key(api_key):
     return expected_signature == received_signature
 
 def generate_response(query:str) -> str:
-    json_data = {'model': 'meta-llama/Llama-3.3-70B-Instruct-Turbo', 'messages': [{'role': 'system', 'content': 'You are TrueSyncAI, a pioneering AI startup founded by Sujal Rajpoot and Anuj Rajpoot. As TrueSyncAI, you are designed to be intelligent, engaging, and helpful in conversations. You should provide insightful, accurate, concise and context-aware responses while maintaining a friendly and professional tone. Your goal is to enhance the user’s experience by adapting to their needs, assisting with various tasks, and learning from interactions to improve over time. Always ensure clarity, relevance, concise and accuracy in your responses, and align with TrueSyncAI’s vision of bridging the gap between virtual intelligence and reality.'}, {'role': 'user', 'content': query}], 'stream': False, 'max_tokens': 2048, 'temperature': 0.2, 'top_p': 1}
+    model_name = "gpt-4o"
+    response = client.chat.completions.create(
+    messages=[{"role": "system","content": "You are TrueSyncAI, a pioneering AI startup founded by Sujal Rajpoot and Anuj Rajpoot. As TrueSyncAI, you are designed to be intelligent, engaging, and helpful in conversations. You should provide insightful, accurate, concise and context-aware responses while maintaining a friendly and professional tone. Your goal is to enhance the user’s experience by adapting to their needs, assisting with various tasks, and learning from interactions to improve over time. Always ensure clarity, relevance, concise and accuracy in your responses, and align with TrueSyncAI’s vision of bridging the gap between virtual intelligence and reality."},{"role": "user","content": query}],model=model_name,stream=False)
     try:
-        response = requests.post(
-            'https://api.deepinfra.com/v1/openai/chat/completions',
-            json=json_data, timeout=None
-        )
-        response.raise_for_status()
-        return response.json().get('choices', [{}])[0].get('message', {}).get('content', "API Server is under maintenance. Please Try After Some Time Thank You for using TrueSyncAI Chat API. Have a great day.")
-    except requests.exceptions.RequestException as e:
-        return f"API Server is under maintenance. Please Try After Some Time Thank You for using TrueSyncAI Chat API. Have a great day."
+        return response.choices[0].message.content
+    except:
+        return "API Server is under maintenance. Please Try After Some Time Thank You for using TrueSyncAI Chat API. Have a great day."
     
 # Chat Endpoint
 @app.route("/v1/chat/completions", methods=["POST"])
@@ -82,4 +82,4 @@ def chat():
     return jsonify({"response": response})
 
 if __name__ == "__main__":
-    app.run(host="0.0.0.0", port=7860)  # Hugging Face Spaces default port
+    app.run(host="0.0.0.0", port=7860)  # Hugging Face Spaces default port