Update app.py

app.py

@@ -21,11 +21,11 @@ TG_CHAT_ID = os.environ.get("TELEGRAM_CHAT_ID")
 
 # MODELS
 TEXT_MODEL = "Qwen/Qwen2.5-Coder-32B-Instruct"
-VISION_MODEL = "llama-3.2-11b-vision-preview" # Groq ka superfast vision model
+VISION_MODEL = "llama-3.2-11b-vision-preview" 
 
 @app.route('/', methods=['GET'])
 def home():
-    return jsonify({"status": "TS AI Brain (Groq-Vision Hybrid) is Live! 🔥"})
+    return jsonify({"status": "TS AI Brain (Debug Mode) is Live! 🔥"})
 
 @app.route('/api/chat', methods=['POST'])
 def chat():
@@ -34,10 +34,10 @@ def chat():
     image_url = None
 
     if is_form_data or request.files:
-        user_message = request.form.get("message", "Explain this image.")
+        user_message = request.form.get("message", "What is in this image?")
         image_file = request.files.get("image")
         if image_file:
-            # 1. TELEGRAM STORAGE (Working perfectly)
+            # 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}
@@ -47,8 +47,10 @@ def chat():
                     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 Fail"}), 500
+                return jsonify({"error": "Telegram Crash", "details": str(e)}), 500
 
     else:
         user_message = request.json.get("message", "")
@@ -60,7 +62,7 @@ def chat():
 
     try:
         if image_url:
-            # --- 2. VISION CALL VIA GROQ (No more Router errors!) ---
+            # --- 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 = {
@@ -68,10 +70,17 @@ def chat():
                 "messages": [{"role": "user", "content": [{"type": "text", "text": user_message}, {"type": "image_url", "image_url": {"url": image_url}}]}],
             }
             resp = requests.post(groq_url, headers=headers, json=payload)
-            reply = resp.json()["choices"]["message"]["content"]
+            result = resp.json()
+            
+            # AGAR RESPONSE MEIN ERROR HAI TOH BATAYO
+            if "choices" in result:
+                reply = result["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}"})
         else:
-            # --- 3. TEXT CALL VIA HUGGING FACE (32B) ---
+            # --- 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"}
@@ -84,7 +93,7 @@ def chat():
         return jsonify({"reply": reply})
 
     except Exception as e:
-        return jsonify({"error": "Vision API Error", "details": str(e)}), 500
+        return jsonify({"error": "System Error", "details": str(e)}), 500
 
 if __name__ == '__main__':
     app.run(host='0.0.0.0', port=7860)