error checking

app.py

@@ -3,6 +3,7 @@ import gradio as gr
 import requests
 import inspect
 import pandas as pd
+import traceback
 from tools import intelligent_agent, get_file_type, write_bytes_to_temp_dir, State
 
 # (Keep Constants as is)
@@ -61,24 +62,29 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
         return f"An unexpected error occurred fetching questions: {e}", None
     
     # 2. Create states
-    for item in hf_questions:
-        file_name = item.get('file_name', '')
-        if file_name == '':
-            item['input_file'] = None
-            item['file_type'] = None
-            item['file_path'] = None
-        else:
-            # Call the API to retrieve the file; adjust params as needed
-            task_id = item['task_id']
-            api_response = requests.get(f"{api_url}/{task_id}")
-            if api_response.status_code == 200:
-                item['input_file'] = api_response.content  # Store file as bytes
-                item['file_type'] = get_file_type(file_name)
-                item['file_path'] = write_bytes_to_temp_dir(item['input_file'], file_name)
+    try:
+        for item in hf_questions:
+            file_name = item.get('file_name', '')
+            if file_name == '':
+                item['input_file'] = None
+                item['file_type'] = None
+                item['file_path'] = None
             else:
-                item['input_file'] = None  # Or handle error as needed
+                # Call the API to retrieve the file; adjust params as needed
+                task_id = item['task_id']
+                api_response = requests.get(f"{api_url}/{task_id}")
+                if api_response.status_code == 200:
+                    item['input_file'] = api_response.content  # Store file as bytes
+                    item['file_type'] = get_file_type(file_name)
+                    item['file_path'] = write_bytes_to_temp_dir(item['input_file'], file_name)
+                else:
+                    item['input_file'] = None  # Or handle error as needed
+    except Exception as e:
+        tb_str = traceback.format_exc()
+        print(f"Error creating new states: {tb_str}")
+        return f"Error creating new states: {tb_str}", None
 
-    # 1. Instantiate Agent ( modify this part to create your agent)
+    # 3. Setup states for questions
     try:
         ans = []
         for r in range(len(hf_questions)):
@@ -92,8 +98,9 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
         print('\n')
         #ans.append(temp_ans)
     except Exception as e:
-        print(f"Error instantiating agent: {e}")
-        return f"Error initializing agent: {e}", None
+        tb_str = traceback.format_exc()
+        print(f"Error setting up states: {tb_str}")
+        return f"Error setting up states: {tb_str}", None
     # In the case of an app running as a hugging Face space, this link points toward your codebase ( usefull for others so please keep it public)
     agent_code = f"https://huggingface.co/spaces/{space_id}/tree/main"
     print(agent_code)