add more imports

.gitignore

@@ -5,3 +5,4 @@ test.py
 __pycache__/
 f918266a-b3e0-4914-865d-4faa564f1aef.py
 *.env
+answers_*.json

app.py

@@ -11,6 +11,7 @@ import time
 import datetime
 from utils import download_file
 from dotenv import load_dotenv
+import json
 
 load_dotenv()
 
@@ -60,8 +61,8 @@ class BasicAgent:
         #model = InferenceClientModel()
 
         response = litellm.completion(
-            #model="gemini/gemini-2.0-flash",
-            model="gemini/gemini-2.0-flash-lite",
+            model="gemini/gemini-2.0-flash",
+            #model="gemini/gemini-2.0-flash-lite",
             messages=messages,
         )
 
@@ -76,7 +77,7 @@ class BasicAgent:
             tools=[calculator_tool],
             managed_agents=[self.web_agent, describe_audio, describe_image, read_excel_file, read_python_file],
             model=self.model,
-            additional_authorized_imports=["pandas", "re", "requests", "json", "numpy"],
+            additional_authorized_imports=["pandas", "re", "requests", "json", "numpy", "bs4", "datetime", "os", "io", "csv"],
             max_steps=10,
             verbosity_level=2,
             planning_interval=5,
@@ -155,15 +156,26 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
             submitted_answer = agent(question_text)
             answers_payload.append({"task_id": task_id, "submitted_answer": submitted_answer})
             results_log.append({"Task ID": task_id, "Question": question_text, "Submitted Answer": submitted_answer})
+
         except Exception as e:
              print(f"Error running agent on task {task_id}: {e}")
              results_log.append({"Task ID": task_id, "Question": question_text, "Submitted Answer": f"AGENT ERROR: {e}"})
         
-        time.sleep(60)
+        time.sleep(100)
 
     if not answers_payload:
         print("Agent did not produce any answers to submit.")
         return "Agent did not produce any answers to submit.", pd.DataFrame(results_log)
+    
+    # 6. Save answers to json
+    try:
+        timestamp = datetime.datetime.now().strftime("%Y%m%d_%H%M%S")
+        filename = f"answers_{timestamp}.json"
+        with open(filename, "w") as f:
+            json.dump(answers_payload, f)
+        print(f"Answers saved to {filename}")
+    except Exception as e:
+        print(f"Error saving answers to file: {e}")
 
     # 4. Prepare Submission 
     submission_data = {"username": username.strip(), "agent_code": agent_code, "answers": answers_payload}
@@ -211,7 +223,9 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
         status_message = f"An unexpected error occurred during submission: {e}"
         print(status_message)
         results_df = pd.DataFrame(results_log)
-        return status_message, results_df
+        return status_message, results_df    
+
+
 
 
 # --- Build Gradio Interface using Blocks ---