Update app.py

app.py

@@ -5,16 +5,15 @@ from flask import Flask, request, Response, stream_with_context, render_template
 
 app = Flask(__name__)
 
-# 🔐 हगिंग फेस 'Secrets' से आपका सुरक्षित डेटा उठाना
+# हगिंग फेस 'Secrets'
 API_KEY = os.environ.get("YOUR_VEDIKA_API_KEY") 
 BASE_URL = os.environ.get("BASE_URL")
-MODEL_ID = os.environ.get("MODEL_ID")
+MODEL_ID = os.environ.get("MODEL_ID", "google/gemma-3n-e2b-it")
 
-# URL को सुरक्षित बनाना (अगर BASE_URL में अंत में /chat/completions नहीं है तो उसे जोड़ना)
 if BASE_URL and not BASE_URL.endswith("/chat/completions"):
     INVOKE_URL = f"{BASE_URL.rstrip('/')}/chat/completions"
 else:
-    INVOKE_URL = BASE_URL
+    INVOKE_URL = BASE_URL or "https://integrate.api.nvidia.com/v1/chat/completions"
 
 @app.route('/')
 def home():
@@ -30,50 +29,51 @@ def chat():
     data = request.get_json() or {}
     user_message = data.get("message", "")
     attachments = data.get("attachments", []) 
+    system_prompt = data.get("system_prompt", "")
+    max_tokens = data.get("max_tokens", 4096)
+    temperature = data.get("temperature", 0.6)
 
-    # 🎨 मल्टीमोडल कंटेंट एरे तैयार करना
+    messages = []
+
+    # 1. सिस्टम प्रॉम्प्ट (System Instruction) सेट करना
+    if system_prompt.strip():
+        messages.append({"role": "system", "content": system_prompt})
+
+    # 2. मल्टीमोडल कंटेंट एरे तैयार करना
     content_payload = []
-    
-    # 1. टेक्स्ट इनपुट को जोड़ना
     if user_message.strip():
         content_payload.append({"type": "text", "text": user_message})
         
-    # 2. इमेज या ऑडियो (Base64) को जोड़ना
     for att in attachments:
         att_type = att.get("type")
         b64_data = att.get("data")
         
         if att_type == "image":
-            content_payload.append({
-                "type": "image_url", 
-                "image_url": {"url": f"data:image/jpeg;base64,{b64_data}"}
-            })
-        elif att_type == "audio":
-            content_payload.append({
-                "type": "input_audio", 
-                "input_audio": {"data": b64_data, "format": "wav"}
-            })
+            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 = {
         "Authorization": f"Bearer {API_KEY}",
         "Accept": "text/event-stream"
     }
 
-    # डायनामिक मॉडल आईडी का उपयोग
     payload = {
         "model": MODEL_ID,
-        "messages": [{"role": "user", "content": content_payload}],
-        "max_tokens": 2048,
-        "temperature": 0.20,
+        "messages": messages,
+        "max_tokens": int(max_tokens),
+        "temperature": float(temperature),
         "top_p": 0.70,
         "stream": True
     }
 
-    # API पर स्ट्रीमिंग रिक्वेस्ट भेजना
     response = requests.post(INVOKE_URL, headers=headers, json=payload, stream=True)
 
     def generate():