Update app.py

app.py

@@ -12,11 +12,21 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 # ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
 class BasicAgent:
     def __init__(self):
-        print("BasicAgent initialized.")
+        try:
+            self.llm = init_chat_model(model="openai:gpt-4.1")
+            self.agent = create_react_agent(
+                model=self.llm,
+                tools=[websearch_tool, code_interpreter_tool],
+                prompt="You are a helpful assistant. You can use tools to answer questions."
+            )
+BasicAgentprint("BasicAgent initialized.")
+        except Exception as e:
+            print(f"Failed to initialize BasicAgent: {e}")
+            
     def __call__(self, question: str) -> str:
-        print(f"Agent received question (first 50 chars): {question[:50]}...")
-        fixed_answer = "This is a default answer."
-        print(f"Agent returning fixed answer: {fixed_answer}")
+        # print(f"Agent received question (first 50 chars): {question[:50]}...")
+        fixed_answer = self.agent.invoke({"messages": [{"role": "user", "content": question}]}).get("messages", [])[-1].content
+        # print(f"Agent returning fixed answer: {fixed_answer}")
         return fixed_answer
 
 def run_and_submit_all( profile: gr.OAuthProfile | None):