Update app.py

app.py

@@ -9,7 +9,7 @@ import pandas as pd
 import requests
 from langchain_community.document_loaders import UnstructuredExcelLoader
 
-from agents import supervisor_agent
+from agents import build_supervisor_agent
 
 # (Keep Constants as is)
 # --- Constants ---
@@ -91,8 +91,8 @@ def filter_supported_content_blocks(messages):
     return filtered
 
 class Agent:
-    def __init__(self):
-        self.main_agent = supervisor_agent
+    def __init__(self, main_agent):
+        self.main_agent = main_agent
         print("Agent initialized.")
     def __call__(self, question: str, file_name: str = "", task_id: str = "") -> str:
         print(f"Agent received question (first 50 chars): {question[:50]}...")
@@ -110,7 +110,7 @@ class Agent:
         else:
             return str(content)
 
-def run_and_submit_all( profile: gr.OAuthProfile | None):
+def run_and_submit_all(profile: gr.OAuthProfile | None, openai_key: str, google_key: str):
     """
     Fetches all questions, runs the BasicAgent on them, submits all answers,
     and displays the results.
@@ -131,11 +131,11 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
 
     # 1. Instantiate Agent ( modify this part to create your agent)
     try:
-        agent = Agent()
+        agent_instance = build_supervisor_agent(openai_key, google_key)
+        agent = Agent(agent_instance)
     except Exception as e:
         print(f"Error instantiating agent: {e}")
         return f"Error initializing agent: {e}", 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)
 
@@ -246,8 +246,8 @@ with gr.Blocks() as demo:
 
         ---
         **Disclaimers:**
-        Once clicking on the "submit" button, it can take quite some time (this is the time for the agent to go through all the questions).
-        This space provides a basic setup and is intentionally sub-optimal to encourage you to develop your own, more robust solution. For instance for the delay process of the submit button, a solution could be to cache the answers and submit in a separate action or even to answer the questions in async.
+        Once clicking on the "submit button, it can take quite some time ( this is the time for the agent to go through all the questions).
+        This space provides a basic setup and is intentionally sub-optimal to encourage you to develop your own, more robust solution. For instance for the delay process of the submit button, a solution could be to cache the answers and submit in a seperate action or even to answer the questions in async.
         """
     )
 
@@ -281,6 +281,7 @@ with gr.Blocks() as demo:
 
     run_button.click(
         fn=run_and_submit_all,
+        inputs=[gr.OAuthProfile(), openai_key_box, google_key_box],
         outputs=[status_output, results_table]
     )