Update app.py

app.py

@@ -19,8 +19,18 @@ headers = {
 def query_mistral(question):
     payload = {"inputs": question, "parameters": {"max_length": 256}}
     response = requests.post(API_URL, headers=headers, json=payload)
-    output = response.json()
-    return output[0]["generated_text"]
+    
+    try:
+        output = response.json()
+        # Check for standard output format
+        if isinstance(output, list) and "generated_text" in output[0]:
+            return output[0]["generated_text"]
+        else:
+            # Return error message or full object for debugging
+            return f"[Error from Mistral API]: {output}"
+    except Exception as e:
+        return f"[Exception in query_mistral]: {str(e)}"
+
 
 POLYGON_API_KEY = os.getenv("POLYGON_API_KEY")