Update app.py

app.py

@@ -21,10 +21,10 @@ 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. . . .")
-        self.agent = smart_agent()
+# class BasicAgent:
+#     def __init__(self):
+#         print("BasicAgent initialized. . . .")
+#         self.agent = smart_agent()
         # self.api_key = os.getenv("HF_TOKEN")
         # self.llm = HuggingFaceInferenceAPI(
         #     model_name="deepseek-ai/DeepSeek-R1-0528",
@@ -38,17 +38,33 @@ class BasicAgent:
         #     llm=self.llm
         # )
 
+    # async def run(self, question: str) -> str:
+    #     print(f"Agent received question (first 50 chars): {question[:50]}...")
+    #     answer = await self.agent.run(question)
+    #     print(f"Agent returning answer: {answer}")
+    #     return answer
+        
+    # def __call__(self, question: str) -> str:
+    #     #return asyncio.run(self.run(question))
+    #     return self.run(question)
+    #     #return self.agent.run(question)
+
+class BasicAgent:
+    def __init__(self, agent):
+        self.agent = agent
+        print("✅ BasicAgent initialized successfully.")
 
     async def run(self, question: str) -> str:
         print(f"Agent received question (first 50 chars): {question[:50]}...")
         answer = await self.agent.run(question)
-        print(f"Agent returning answer: {answer}")
+        print(f"Agent returning answer: {answer[:100]}")
         return answer
-        
-    def __call__(self, question: str) -> str:
-        #return asyncio.run(self.run(question))
-        return self.run(question)
-        #return self.agent.run(question)
+
+    @classmethod
+    async def create(cls):
+        """Factory method to asynchronously initialise BasicAgent."""
+        agent = await smart_agent()
+        return cls(agent)
 
 def run_and_submit_all( profile: gr.OAuthProfile | None):
     """
@@ -71,7 +87,8 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
 
     # 1. Instantiate Agent ( modify this part to create your agent)
     try:
-        agent = BasicAgent()
+        #agent = BasicAgent()
+        agent = BasicAgent.create()
     except Exception as e:
         print(f"Error instantiating agent: {e}")
         return f"Error initializing agent: {e}", None