Update app.py

app.py

@@ -7,6 +7,7 @@ import pandas as pd
 from langchain_tavily import TavilySearch
 from langgraph.prebuilt import create_react_agent
 from langchain_huggingface import HuggingFaceEndpoint, ChatHuggingFace
+from langchain_core.messages import HumanMessage, SystemMessage
 
 # (Keep Constants as is)
 # --- Constants ---
@@ -36,6 +37,9 @@ class BasicAgent:
             huggingfacehub_api_token=HUGGINGFACEHUB_API_TOKEN,
         )
 
+        print("ChatHuggingFace")
+        chat = ChatHuggingFace(llm=llm, verbose=True)
+
         # Initialize Tavily Search Tool
         print("Initialize Tavily Search Tool")
         tavily_search_tool = TavilySearch(
@@ -43,8 +47,8 @@ class BasicAgent:
             topic="general",
         )
 
-        print("create_react_agent")
-        agent = create_react_agent(llm, [tavily_search_tool])
+ #       print("create_react_agent")
+ #       agent = create_react_agent(llm, [tavily_search_tool])
 
 #        chat = ChatHuggingFace(llm=llm, verbose=True)
 #        tools = [search_web_tool]
@@ -52,7 +56,7 @@ class BasicAgent:
         
     def __call__(self, question: str) -> str:
         print(f"Agent received question (first 50 chars): {question[:50]}...")
-        answer = agent.invoke({"messages": [{"role": "user", "content": question}]})
+        answer = chat.invoke({"messages": [{"role": "user", "content": question}]})
         print(f"Agent returning answer: {answer}")
         return answer