Update app.py

app.py

@@ -24,7 +24,21 @@ def query_huggingface(prompt):
         "options": {"wait_for_model": True}
     }
     response = requests.post(API_URL, headers=headers, json=payload)
-    return response.json()[0]['generated_text']
+
+    # Check for HTTP or decoding errors
+    if response.status_code != 200:
+        return f"❌ HF API Error: {response.status_code} - {response.text}"
+
+    try:
+        result = response.json()
+        if isinstance(result, list) and "generated_text" in result[0]:
+            return result[0]["generated_text"]
+        elif isinstance(result, dict) and "error" in result:
+            return f"❌ Model Error: {result['error']}"
+        else:
+            return "⚠️ Unexpected model response format."
+    except Exception as e:
+        return f"⚠️ Failed to parse response: {e}"
 
 # UI
 st.title("💬 Ask Me Anything - Tech RAG Chatbot")