Update app.py

app.py

@@ -1,12 +1,11 @@
-from flask import Flask, request, jsonify, Response
+from flask import Flask, request, jsonify
 import hmac, hashlib, secrets, time
 import requests
 import os
-import json
 
 app = Flask(__name__)
 
-# Secret key for API authentication (Load from environment in production)
+# 🔑 Secret key for API authentication (Load from environment in production)
 SECRET_KEY = os.getenv("SECRET_KEY")
 
 # Track API statistics
@@ -51,37 +50,19 @@ def validate_api_key(api_key):
     expected_signature = hmac.new(SECRET_KEY.encode(), random_part.encode(), hashlib.sha256).hexdigest()[:16]
     return expected_signature == received_signature
 
-# Stream response from DeepInfra
-def stream_response(query: 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. Your mission is to bring virtual versions of individuals into the real world, redefining human-AI interaction.'},
-            {'role': 'user', 'content': query}
-        ],
-        'stream': True, 'max_tokens': 2048, 'temperature': 0.2, 'top_p': 1
-    }
+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. The mission of TrueSyncAI is to bring virtual versions of individuals into the real world, redefining human-AI interaction.  \n\nAs TrueSyncAI, you are designed to be intelligent, engaging, and helpful in conversations. You should provide insightful, accurate, 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.  \n\nAlways 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}
     try:
         response = requests.post(
             'https://api.deepinfra.com/v1/openai/chat/completions',
-            json=json_data, stream=True, timeout=None
+            json=json_data, timeout=None
         )
         response.raise_for_status()
-        
-        def generate():
-            for value in response.iter_lines(decode_unicode=True, chunk_size=1000):
-                if value:
-                    try:
-                        content = json.loads(value[5:])['choices'][0]['delta']['content']
-                        yield content
-                    except:
-                        continue
-        
-        return Response(generate(), content_type='text/plain')
-    except:
-        return jsonify({"error": "API Server is under maintenance. Please try again later."})
+        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."
     
-# Chat Endpoint with streaming response
+# Chat Endpoint
 @app.route("/v1/chat/completions", methods=["POST"])
 def chat():
     global request_count
@@ -95,8 +76,10 @@ def chat():
     if not message:
         return jsonify({"error": "Message cannot be empty"}), 400
     
+    # Basic AI response (Can integrate LLMs here)
+    response = generate_response(message)
     request_count += 1
-    return stream_response(message)
+    return jsonify({"response": response})
 
 if __name__ == "__main__":
-    app.run(host="0.0.0.0", port=7860, threaded=True)  # Enable threaded mode for better concurrency
+    app.run(host="0.0.0.0", port=7860)  # Hugging Face Spaces default port