Update app.py

app.py

@@ -27,8 +27,7 @@ class BasicAgent:
             model_name="deepseek-ai/DeepSeek-R1-0528",
             token=self.api_key,
             provider="auto",
-            max_iterations=10,
-            max_execution_time=60
+            max_iterations=10
         )
         self.agent = AgentWorkflow.from_tools_or_functions(
             [wikipedia_search_tool, youtube_transcript_tool, youtube_transcript_snippet_tool, 
@@ -56,8 +55,20 @@ class BasicAgent:
 
 # START MY CODE
 
-print("BasicAgent initialized.")
-api_key = os.getenv("HF_TOKEN")
+# print("BasicAgent initialized.")
+# api_key = os.getenv("HF_TOKEN")
+# llm = HuggingFaceInferenceAPI(
+#     model_name="deepseek-ai/DeepSeek-R1-0528",
+#     token=api_key,
+#     provider="auto",
+#     max_iterations=10,
+#     max_execution_time=60
+# )
+# agent = AgentWorkflow.from_tools_or_functions(
+#     [wikipedia_search_tool, youtube_transcript_tool, youtube_transcript_snippet_tool, 
+#      round_to_two_decimals_tool, text_inverter_tool, google_web_search_tool],
+#     llm=llm
+# )
 
 # END MY CODE
 
@@ -66,20 +77,6 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
     Fetches all questions, runs the BasicAgent on them, submits all answers,
     and displays the results.
     """
-
-    llm = HuggingFaceInferenceAPI(
-        model_name="deepseek-ai/DeepSeek-R1-0528",
-        token=api_key,
-        provider="auto",
-        max_iterations=10,
-        max_execution_time=60
-    )
-    agent = AgentWorkflow.from_tools_or_functions(
-        [wikipedia_search_tool, youtube_transcript_tool, youtube_transcript_snippet_tool, 
-         round_to_two_decimals_tool, text_inverter_tool, google_web_search_tool],
-        llm=llm
-    )
-    
     # --- Determine HF Space Runtime URL and Repo URL ---
     space_id = os.getenv("SPACE_ID") # Get the SPACE_ID for sending link to the code
 
@@ -95,11 +92,11 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
     submit_url = f"{api_url}/submit"
 
     # 1. Instantiate Agent ( modify this part to create your agent)
-    # try:
-    #     agent_OLD = BasicAgent()
-    # except Exception as e:
-    #     print(f"Error instantiating agent: {e}")
-    #     return f"Error initializing agent: {e}", None
+    try:
+        agent = BasicAgent()
+    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)