Update app.py

app.py

@@ -34,14 +34,22 @@ class GAIAAgent:
         filepath = os.path.join("./", filename)
         if filename.endswith(".txt") and os.path.exists(filepath):
             with open(filepath, "r") as file:
-                return file.read()[:1000]
+                return file.read()[:1000]  # limit to 1000 chars
         return ""
 
     def __call__(self, question: str, file_name: str = None) -> str:
         file_content = self.read_file(file_name) if file_name else None
         prompt = self.format_prompt(question, file_content)
-        result = self.model(prompt)
-        return result.strip().split("Answer:")[-1].strip()
+        try:
+            print("Prompt sent to model:", prompt)
+            result = self.model(prompt)
+            print("Model raw result:", result)
+            if not result or not isinstance(result, str):
+                return "AGENT ERROR: Empty or invalid response"
+            return result.strip().split("Answer:")[-1].strip()
+        except Exception as e:
+            print(f"Model inference failed: {e}")
+            return f"AGENT ERROR: {e}"
 
 
 def run_and_submit_all(profile: gr.OAuthProfile | None):
@@ -127,4 +135,4 @@ with gr.Blocks() as demo:
 
 if __name__ == "__main__":
     print("Launching GAIA agent app...")
-    demo.launch(debug=True, share=False)
+    demo.launch(debug=True, share=False)