update app

app.py

@@ -3,7 +3,7 @@ import gradio as gr
 import requests
 import inspect
 import pandas as pd
-from tools import intelligent_agent, get_file_type, write_bytes_to_temp_dir
+from tools import intelligent_agent, get_file_type, write_bytes_to_temp_dir, State
 
 # (Keep Constants as is)
 # --- Constants ---
@@ -11,14 +11,14 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
 # --- Basic Agent Definition ---
 # ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
-class BasicAgent:
-    def __init__(self):
-        print("BasicAgent initialized.")
-    def __call__(self, question: str) -> str:
-        print(f"Agent received question (first 50 chars): {question[:50]}...")
-        fixed_answer = "This is a default answer."
-        print(f"Agent returning fixed answer: {fixed_answer}")
-        return fixed_answer
+#class BasicAgent:
+#    def __init__(self):
+#        print("BasicAgent initialized.")
+#    def __call__(self, question: str) -> str:
+#        print(f"Agent received question (first 50 chars): {question[:50]}...")
+#        fixed_answer = "This is a default answer."
+#        print(f"Agent returning fixed answer: {fixed_answer}")
+#        return fixed_answer
 
 def run_and_submit_all( profile: gr.OAuthProfile | None):
     """
@@ -39,9 +39,37 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
     questions_url = f"{api_url}/questions"
     submit_url = f"{api_url}/submit"
 
+    # 1. 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)
+        else:
+            item['input_file'] = None  # Or handle error as needed
+
     # 1. Instantiate Agent ( modify this part to create your agent)
     try:
-        agent = BasicAgent()
+        ans = []
+        for r in range(len(hf_questions)):
+            s = State(question = hf_questions[r]['question'],
+                        input_file = hf_questions[r]['input_file'],
+                        file_type = hf_questions[r]['file_type'],
+                        file_path = hf_questions[r]['file_path'])
+            print(s['question'])
+        temp_ans = intelligent_agent(s)
+        print(temp_ans)
+        print('\n')
+        ans.append(temp_ans)
     except Exception as e:
         print(f"Error instantiating agent: {e}")
         return f"Error initializing agent: {e}", None
@@ -70,23 +98,6 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
         print(f"An unexpected error occurred fetching questions: {e}")
         return f"An unexpected error occurred fetching questions: {e}", None
     
-    # 3. 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)
-        else:
-            item['input_file'] = None  # Or handle error as needed
 
 """
     # 3. Run your Agent