Update app.py

app.py

@@ -1,5 +1,6 @@
 import os
 import requests
+import time
 from flask import Flask, request, jsonify
 from flask_cors import CORS
 import firebase_admin
@@ -9,85 +10,84 @@ app = Flask(__name__)
 CORS(app) 
 
 # --- FIREBASE SETUP ---
-cred = credentials.Certificate("firebase_key.json")
-firebase_admin.initialize_app(cred)
-db = firestore.client()
+try:
+    if not firebase_admin._apps:
+        cred = credentials.Certificate("firebase_key.json")
+        firebase_admin.initialize_app(cred)
+    db = firestore.client()
+except Exception as e:
+    print(f"Firebase Error: {e}")
 
-# --- SECRETS ---
+# --- SECRETS (Tijori) ---
 HF_TOKEN = os.environ.get("HF_TOKEN")
 GROQ_API_KEY = os.environ.get("GROQ_API_KEY")
 TG_BOT_TOKEN = os.environ.get("TELEGRAM_BOT_TOKEN") 
 TG_CHAT_ID = os.environ.get("TELEGRAM_CHAT_ID")     
 
-# MODELS
+# 2026 CURRENT MODELS
 TEXT_MODEL = "Qwen/Qwen2.5-Coder-32B-Instruct"
-VISION_MODEL = "meta-llama/llama-4-scout-17b-16e-instruct" 
+VISION_MODEL = "meta-llama/llama-4-scout-17b-16e-instruct" # March 2026 Stable
 
 @app.route('/', methods=['GET'])
 def home():
-    return jsonify({"status": "TS AI Brain (Debug Mode) is Live! 🔥"})
+    return jsonify({"status": "TS AI Brain (Bulletproof 2026) is Active! 🚀"})
 
 @app.route('/api/chat', methods=['POST'])
 def chat():
-    is_form_data = 'multipart/form-data' in request.content_type if request.content_type else False
-    user_message = ""
-    image_url = None
+    try:
+        # 1. PATA LAGAO REQUEST KYA HAI
+        is_form_data = 'multipart/form-data' in request.content_type if request.content_type else False
+        user_message = ""
+        image_url = None
 
-    if is_form_data or request.files:
-        user_message = request.form.get("message", "What is in this image?")
-        image_file = request.files.get("image")
-        if image_file:
-            # TELEGRAM UPLOAD
-            tg_url = f"https://api.telegram.org/bot{TG_BOT_TOKEN}/sendPhoto"
-            files = {'photo': (image_file.filename, image_file.read(), image_file.mimetype)}
-            data = {'chat_id': TG_CHAT_ID}
-            try:
+        if is_form_data:
+            user_message = request.form.get("message", "Analyze this.")
+            if 'image' in request.files:
+                image_file = request.files['image']
+                # --- TELEGRAM GODOWN UPLOAD ---
+                tg_url = f"https://api.telegram.org/bot{TG_BOT_TOKEN}/sendPhoto"
+                files = {'photo': (image_file.filename, image_file.read(), image_file.mimetype)}
+                data = {'chat_id': TG_CHAT_ID}
                 tg_resp = requests.post(tg_url, data=data, files=files).json()
+                
                 if tg_resp.get('ok'):
                     file_id = tg_resp['result']['photo'][-1]['file_id']
                     file_info = requests.get(f"https://api.telegram.org/bot{TG_BOT_TOKEN}/getFile?file_id={file_id}").json()
                     image_url = f"https://api.telegram.org/file/bot{TG_BOT_TOKEN}/{file_info['result']['file_path']}"
-                else:
-                    return jsonify({"error": "Telegram error", "details": tg_resp}), 500
-            except Exception as e:
-                return jsonify({"error": "Telegram Crash", "details": str(e)}), 500
-
-    else:
-        user_message = request.json.get("message", "")
+        else:
+            data = request.get_json(silent=True) or {}
+            user_message = data.get("message", "")
 
-    # LOAD HISTORY
-    doc_ref = db.collection(u'chats').document(u'TS_Boss_Session')
-    doc = doc_ref.get()
-    history = doc.to_dict().get('messages', []) if doc.exists else [{"role": "system", "content": "Tum TS Boss ke AI ho. Hinglish mein baat karo."}]
+        # 2. LOAD HISTORY
+        doc_ref = db.collection(u'chats').document(u'TS_Boss_Session')
+        doc = doc_ref.get()
+        history = doc.to_dict().get('messages', []) if doc.exists else [{"role": "system", "content": "Tum TS Boss ke AI ho. Hinglish mein baat karo."}]
 
-    try:
+        # 3. VISION CALL (GROQ) OR TEXT CALL (HF)
         if image_url:
-            # --- VISION CALL VIA GROQ ---
             groq_url = "https://api.groq.com/openai/v1/chat/completions"
             headers = {"Authorization": f"Bearer {GROQ_API_KEY}", "Content-Type": "application/json"}
             payload = {
                 "model": VISION_MODEL,
                 "messages": [{"role": "user", "content": [{"type": "text", "text": user_message}, {"type": "image_url", "image_url": {"url": image_url}}]}],
+                "max_tokens": 1024
             }
-            resp = requests.post(groq_url, headers=headers, json=payload)
-            result = resp.json()
+            resp = requests.post(groq_url, headers=headers, json=payload).json()
             
-            # AGAR RESPONSE MEIN ERROR HAI TOH BATAYO
-            if "choices" in result:
-                reply = result["choices"]["message"]["content"]
+            # SAFE RESPONSE PARSING
+            if isinstance(resp, dict) and "choices" in resp:
+                reply = resp["choices"]["message"]["content"]
             else:
-                return jsonify({"error": "Groq ne mana kar diya!", "raw_response": result}), 500
-            
-            history.append({"role": "user", "content": f"[Screenshot] {user_message}"})
+                return jsonify({"error": "Groq API ne error diya", "raw": resp}), 500
         else:
-            # --- TEXT CALL VIA HUGGING FACE ---
             history.append({"role": "user", "content": user_message})
             hf_url = "https://router.huggingface.co/v1/chat/completions"
             headers = {"Authorization": f"Bearer {HF_TOKEN}", "Content-Type": "application/json"}
             payload = {"model": TEXT_MODEL, "messages": history}
-            resp = requests.post(hf_url, headers=headers, json=payload)
-            reply = resp.json()["choices"][0]["message"]["content"]
+            resp = requests.post(hf_url, headers=headers, json=payload).json()
+            reply = resp["choices"][0]["message"]["content"]
 
+        # 4. SAVE & REPLY
         history.append({"role": "assistant", "content": reply})
         doc_ref.set({'messages': history})
         return jsonify({"reply": reply})