Update app.py

app.py

@@ -1,19 +1,21 @@
 import os
-import json
 import requests
 from flask import Flask, request, Response, stream_with_context, render_template_string
 
 app = Flask(__name__)
 
-# हगिंग फेस 'Secrets'
+# 🔐 100% सुरक्षित: सब कुछ केवल 'Secrets' से कॉल होगा, कोई हार्डकोडेड डेटा नहीं।
 API_KEY = os.environ.get("YOUR_VEDIKA_API_KEY") 
 BASE_URL = os.environ.get("BASE_URL")
-MODEL_ID = os.environ.get("MODEL_ID", "google/gemma-3n-e2b-it")
+MODEL_ID = os.environ.get("MODEL_ID")
 
-if BASE_URL and not BASE_URL.endswith("/chat/completions"):
-    INVOKE_URL = f"{BASE_URL.rstrip('/')}/chat/completions"
-else:
-    INVOKE_URL = BASE_URL or "https://integrate.api.nvidia.com/v1/chat/completions"
+# URL को सुरक्षित रूप से बनाना ताकि कोई डिफ़ॉल्ट लिंक न दिखे
+INVOKE_URL = ""
+if BASE_URL:
+    if not BASE_URL.endswith("/chat/completions"):
+        INVOKE_URL = f"{BASE_URL.rstrip('/')}/chat/completions"
+    else:
+        INVOKE_URL = BASE_URL
 
 @app.route('/')
 def home():
@@ -26,6 +28,10 @@ def home():
 
 @app.route('/api/chat', methods=['POST'])
 def chat():
+    # सुरक्षा जांच: अगर सीक्रेट्स सेट नहीं हैं, तो रिक्वेस्ट आगे नहीं जाएगी
+    if not API_KEY or not INVOKE_URL or not MODEL_ID:
+        return Response("Server Error: Configuration secrets are missing.", status=500)
+
     data = request.get_json() or {}
     user_message = data.get("message", "")
     attachments = data.get("attachments", []) 
@@ -35,11 +41,11 @@ def chat():
 
     messages = []
 
-    # 1. सिस्टम प्रॉम्प्ट (System Instruction) सेट करना
+    # 1. सिस्टम प्रॉम्प्ट
     if system_prompt.strip():
         messages.append({"role": "system", "content": system_prompt})
 
-    # 2. मल्टीमोडल कंटेंट एरे तैयार करना
+    # 2. मल्टीमोडल इनपुट
     content_payload = []
     if user_message.strip():
         content_payload.append({"type": "text", "text": user_message})
@@ -51,13 +57,11 @@ def chat():
         if att_type == "image":
             content_payload.append({"type": "image_url", "image_url": {"url": f"data:image/jpeg;base64,{b64_data}"}})
         elif att_type in ["audio", "file"]:
-            # ऑडियो या अन्य फ़ाइलों के लिए
             content_payload.append({"type": "input_audio", "input_audio": {"data": b64_data, "format": "wav"}})
 
     if not content_payload:
         content_payload.append({"type": "text", "text": "Hello"})
 
-    # यूज़र का मैसेज जोड़ना
     messages.append({"role": "user", "content": content_payload})
 
     headers = {
@@ -74,16 +78,20 @@ def chat():
         "stream": True
     }
 
-    response = requests.post(INVOKE_URL, headers=headers, json=payload, stream=True)
-
-    def generate():
-        for line in response.iter_lines():
-            if line:
-                decoded_line = line.decode("utf-8")
-                if decoded_line.startswith("data: "):
-                    yield decoded_line + "\n\n"
-                    
-    return Response(stream_with_context(generate()), mimetype='text/event-stream')
+    try:
+        # सुरक्षित स्ट्रीमिंग रिक्वेस्ट
+        response = requests.post(INVOKE_URL, headers=headers, json=payload, stream=True)
+        
+        def generate():
+            for line in response.iter_lines():
+                if line:
+                    decoded_line = line.decode("utf-8")
+                    if decoded_line.startswith("data: "):
+                        yield decoded_line + "\n\n"
+                        
+        return Response(stream_with_context(generate()), mimetype='text/event-stream')
+    except Exception as e:
+        return Response("Internal Error: Unable to process request securely.", status=500)
 
 if __name__ == '__main__':
     app.run(host='0.0.0.0', port=7860)