Update app.py

app.py

@@ -27,6 +27,7 @@ def assistant(state: AgentState):
     }
     
 class BasicAgent:
+    
     def __init__(self, llm_checkpoint):
         llm = HuggingFaceEndpoint(repo_id=llm_checkpoint)
         chat = ChatHuggingFace(llm=llm, verbose=True)
@@ -35,14 +36,15 @@ class BasicAgent:
         ]
         chat_with_tools = chat.bind_tools(self.tools)
         self.graph = self._build_graph()
+        
         print("BasicAgent initialized.")
     def __call__(self, question: str) -> str:
         print(f"Agent received question (first 50 chars): {question[:50]}...")
         messages = [HumanMessage(content="Tell me about our guest named 'Lady Ada Lovelace'.")]
         response = self.invoke({"messages": messages})
-        response['messages'][-1].content
-        fixed_answer = "This is a default answer."
+        response = response['messages'][-1].content
         print(f"Agent returning answer: {response}")
+        return response
         
     def _build_graph(self):
         self.builder = StateGraph(AgentState)
@@ -53,8 +55,9 @@ class BasicAgent:
         self.builder.add_edge("tools", "assistant")
         self.builder.compile()
         
-        return fixed_answer
+        
 
+    
 def run_and_submit_all( profile: gr.OAuthProfile | None):
     """
     Fetches all questions, runs the BasicAgent on them, submits all answers,