Update app.py

app.py

@@ -1,6 +1,5 @@
 import os
 import requests
-import time
 from flask import Flask, request, jsonify
 from flask_cors import CORS
 import firebase_admin
@@ -18,52 +17,58 @@ try:
 except Exception as e:
     print(f"Firebase Error: {e}")
 
-# --- SECRETS (Tijori) ---
+# --- SECRETS ---
 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")     
 
-# 2026 CURRENT MODELS
+# MODELS
 TEXT_MODEL = "Qwen/Qwen2.5-Coder-32B-Instruct"
-VISION_MODEL = "meta-llama/llama-4-scout-17b-16e-instruct" # March 2026 Stable
+VISION_MODEL = "llama-3.2-90b-vision-preview" # Groq ka sabse active vision model
 
 @app.route('/', methods=['GET'])
 def home():
-    return jsonify({"status": "TS AI Brain (Bulletproof 2026) is Active! 🚀"})
+    return jsonify({"status": "TS AI Brain (Crash-Proof Mode) is Active! 🚀"})
 
 @app.route('/api/chat', methods=['POST'])
 def chat():
     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
 
+        # 1. READ REQUEST
         if is_form_data:
-            user_message = request.form.get("message", "Analyze this.")
+            user_message = request.form.get("message", "Is photo mein kya error hai?")
             if 'image' in request.files:
                 image_file = request.files['image']
-                # --- TELEGRAM GODOWN UPLOAD ---
+                
+                # 2. TELEGRAM UPLOAD WITH STRICT CHECK
                 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']}"
+                if not tg_resp.get('ok'):
+                    # Agar Telegram ne photo block ki, toh yahi error aayega
+                    return jsonify({"error": "Telegram upload fail", "details": tg_resp}), 500
+                
+                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:
             data = request.get_json(silent=True) or {}
             user_message = data.get("message", "")
 
-        # 2. LOAD HISTORY
+        # 3. HISTORY LOAD
         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."}]
 
-        # 3. VISION CALL (GROQ) OR TEXT CALL (HF)
+        reply = ""
+
+        # 4. VISION CALL (Agar Image Hai)
         if image_url:
             groq_url = "https://api.groq.com/openai/v1/chat/completions"
             headers = {"Authorization": f"Bearer {GROQ_API_KEY}", "Content-Type": "application/json"}
@@ -74,26 +79,33 @@ def chat():
             }
             resp = requests.post(groq_url, headers=headers, json=payload).json()
             
-            # SAFE RESPONSE PARSING
+            # STRICT RESPONSE PARSING (Ab crash nahi hoga)
             if isinstance(resp, dict) and "choices" in resp:
                 reply = resp["choices"]["message"]["content"]
             else:
-                return jsonify({"error": "Groq API ne error diya", "raw": resp}), 500
+                return jsonify({"error": "Groq API Error", "raw_response": resp}), 500
+        
+        # 5. TEXT CALL (Sirf Text Hai Toh)
         else:
             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).json()
-            reply = resp["choices"][0]["message"]["content"]
+            
+            # STRICT RESPONSE PARSING (Yehi tha asli culprit pichli baar)
+            if isinstance(resp, dict) and "choices" in resp:
+                reply = resp["choices"][0]["message"]["content"]
+            else:
+                return jsonify({"error": "Hugging Face Text API Error", "raw_response": resp}), 500
 
-        # 4. SAVE & REPLY
+        # 6. SAVE AND RETURN
         history.append({"role": "assistant", "content": reply})
         doc_ref.set({'messages': history})
         return jsonify({"reply": reply})
 
     except Exception as e:
-        return jsonify({"error": "System Error", "details": str(e)}), 500
+        return jsonify({"error": "Python Code Crash", "details": str(e)}), 500
 
 if __name__ == '__main__':
     app.run(host='0.0.0.0', port=7860)