Update app.py

app.py

@@ -12,12 +12,12 @@ 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):
+    def __init__(self, HF_API_TOKEN):
         print("BasicAgent initialized.")
         # Initialize the model
         #model = HfApiModel()
-        api_token = os.getenv("HF_API_TOKEN")  # get your token
-        model = HfApiModel(model_id="Alibaba-NLP/gte-Qwen1.5-7B-instruct", api_token=api_token)
+        self.HF_API_TOKEN = HF_API_TOKEN
+        model = HfApiModel(model_id="Alibaba-NLP/gte-Qwen1.5-7B-instruct", api_token=self.HF_API_TOKEN)
 
         # Initialize the search tool
         search_tool = DuckDuckGoSearchTool()
@@ -32,7 +32,7 @@ class BasicAgent:
         print(f"Agent returning fixed answer: {fixed_answer}")
         return fixed_answer
 
-def run_and_submit_all( profile: gr.OAuthProfile | None):
+def run_and_submit_all( profile: gr.OAuthProfile | None, HF_API_TOKEN : str):
     """
     Fetches all questions, runs the BasicAgent on them, submits all answers,
     and displays the results.
@@ -53,7 +53,7 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
 
     # 1. Instantiate Agent ( modify this part to create your agent)
     try:
-        agent = BasicAgent()
+        agent = BasicAgent(HF_API_TOKEN)
     except Exception as e:
         print(f"Error instantiating agent: {e}")
         return f"Error initializing agent: {e}", None
@@ -170,7 +170,9 @@ with gr.Blocks() as demo:
     )
 
     gr.LoginButton()
-
+    
+    HF_API_TOKEN_box = gr.Textbox(label="HF_API_TOKEN", type="password", placeholder="sk-...", lines=1)
+    
     run_button = gr.Button("Run Evaluation & Submit All Answers")
 
     status_output = gr.Textbox(label="Run Status / Submission Result", lines=5, interactive=False)
@@ -179,6 +181,9 @@ with gr.Blocks() as demo:
 
     run_button.click(
         fn=run_and_submit_all,
+        inputs=[HF_API_TOKEN_box],
+
+
         outputs=[status_output, results_table]
     )