Update app.py

app.py

@@ -9,10 +9,10 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
 # Basic Agent Definition using HF Inference API
 class BasicAgent:
-    def __init__(self):
+    def __init__(self, token: str):
         print("BasicAgent initialized.")
-        token = os.getenv("HUGGINGFACEHUB_API_TOKEN")
         self.client = InferenceClient("mistralai/Mistral-7B-Instruct-v0.1", token=token)
+
     def __call__(self, question: str) -> str:
         print(f"Agent received question: {question[:50]}...")
         try:
@@ -25,7 +25,7 @@ class BasicAgent:
             print(f"Error while querying HF model: {e}")
             return f"Error: {e}"
 
-def run_and_submit_all(profile: gr.OAuthProfile | None):
+def run_and_submit_all(profile: gr.OAuthProfile | None, token_input: str):
     space_id = os.getenv("SPACE_ID")
 
     if profile:
@@ -41,7 +41,7 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
 
     # 1. Instantiate Agent
     try:
-        agent = BasicAgent()
+        agent = BasicAgent(token=token_input)
     except Exception as e:
         print(f"Error instantiating agent: {e}")
         return f"Error initializing agent: {e}", None
@@ -132,16 +132,22 @@ with gr.Blocks() as demo:
         **Instructions:**
         1. Please clone this space, then modify the code to define your agent's logic, the tools, the necessary packages, etc ...
         2. Log in to your Hugging Face account using the button below. This uses your HF username for submission.
-        3. Click 'Run Evaluation & Submit All Answers' to fetch questions, run your agent, submit answers, and see the score.
+        3. Paste your Hugging Face token below.
+        4. Click 'Run Evaluation & Submit All Answers' to fetch questions, run your agent, submit answers, and see the score.
         """
     )
 
     gr.LoginButton()
+    token_box = gr.Textbox(label="Enter your Hugging Face token", type="password", placeholder="hf_...")
     run_button = gr.Button("Run Evaluation & Submit All Answers")
     status_output = gr.Textbox(label="Run Status / Submission Result", lines=5, interactive=False)
     results_table = gr.DataFrame(label="Questions and Agent Answers", wrap=True)
 
-    run_button.click(fn=run_and_submit_all, outputs=[status_output, results_table])
+    run_button.click(
+        fn=run_and_submit_all,
+        inputs=[gr.OAuthProfile(), token_box],
+        outputs=[status_output, results_table]
+    )
 
 if __name__ == "__main__":
     demo.launch(debug=True, share=False)