Update app.py

app.py

@@ -1,52 +1,97 @@
 import os
-import logging
-import time
-from flask import Flask, request, Response, stream_with_context, jsonify, send_from_directory
+import requests
+from flask import Flask, request, Response, stream_with_context, render_template_string
 
-# --- 1. एन्टरप्राइज लॉगिंग कॉन्फ़िगरेशन ---
-logging.basicConfig(level=logging.INFO, format='[%(asctime)s] %(levelname)s: %(message)s')
-logger = logging.getLogger("ArjunCore")
+app = Flask(__name__)
 
-app = Flask(__name__, static_folder='.', static_url_path='')
+# 🔐 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")
 
-# --- 2. कॉन्फ़िगरेशन क्लास (Environment variables से डेटा लेने के लिए) ---
-class Config:
-    API_KEY = os.environ.get("YOUR_VEDIKA_API_KEY")
-    ENDPOINT = os.environ.get("BASE_URL")
-    MODEL_ID = os.environ.get("MODEL_ID")
+# 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
 
-# --- 3. रूट: सर्वर रन होते ही ट्रिगर होना ---
 @app.route('/')
-def index():
-    """यह फंक्शन सुनिश्चित करता है कि जैसे ही यूज़र सर्वर पर आए, index.html लोड हो जाए।"""
-    logger.info("Serving index.html interface...")
-    return send_from_directory('.', 'index.html')
+def home():
+    try:
+        with open('index.html', 'r', encoding='utf-8') as f:
+            html_content = f.read()
+        return render_template_string(html_content)
+    except Exception as e:
+        return f"index.html file missing! Error: {str(e)}"
 
-# --- 4. चैट एपीआई (LLM Proxy) ---
 @app.route('/api/chat', methods=['POST'])
 def chat():
-    """फ्रंटएंड (Gradio/JS) से आने वाली रिक्वेस्ट को LLM तक पहुँचाना।"""
+    # सुरक्षा जांच: अगर सीक्रेट्स सेट नहीं हैं, तो रिक्वेस्ट आगे नहीं जाएगी
+    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", []) 
+    system_prompt = data.get("system_prompt", "")
+    max_tokens = data.get("max_tokens", 4096)
+    temperature = data.get("temperature", 0.6)
+
+    messages = []
+
+    # 1. सिस्टम प्रॉम्प्ट
+    if system_prompt.strip():
+        messages.append({"role": "system", "content": system_prompt})
+
+    # 2. मल्टीमोडल इनपुट
+    content_payload = []
+    if user_message.strip():
+        content_payload.append({"type": "text", "text": user_message})
+        
+    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 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": messages,
+        "max_tokens": int(max_tokens),
+        "temperature": float(temperature),
+        "top_p": 0.70,
+        "stream": True
+    }
+
     try:
-        data = request.json
-        # यहाँ आप PayloadBuilder का उपयोग करके अपना लॉजिक बढ़ा सकते हैं
-        logger.info("Processing chat request...")
+        # सुरक्षित स्ट्रीमिंग रिक्वेस्ट
+        response = requests.post(INVOKE_URL, headers=headers, json=payload, stream=True)
         
-        # प्रॉक्सी लॉजिक यहाँ डालें (जैसे पिछला कोड था)
-        return jsonify({"status": "processing", "message": "Backend connected successfully"})
+        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 jsonify({"error": str(e)}), 500
-
-# --- 5. हेल्थ चेक ---
-@app.route('/health')
-def health():
-    return jsonify({"status": "Arjun Core Active", "uptime": time.time()})
+        return Response("Internal Error: Unable to process request securely.", status=500)
 
-# --- 6. मुख्य एक्जीक्यूशन ---
 if __name__ == '__main__':
-    port = int(os.environ.get("PORT", 7860))
-    print("\n==============================================")
-    print("      ARJUN CORE V3 - SYSTEM ONLINE           ")
-    print("==============================================\n")
-    
-    # यह Flask का सर्वर सीधे index.html को सर्व करेगा
-    app.run(host='0.0.0.0', port=port, threaded=True)
+    app.run(host='0.0.0.0', port=7860)