Update app.py

app.py

@@ -7,12 +7,27 @@ from typing import TypedDict, Optional, Annotated
 from langchain_core.messages import AnyMessage
 from langchain_huggingface import HuggingFaceEndpoint, ChatHuggingFace
 from langgraph.graph.message import add_messages
+from langchain_hyperbrowser import HyperbrowserBrowserUseTool
+from langgraph.graph import START, StateGraph
+from langgraph.prebuilt import ToolNode, tools_condition
+from langchain_core.messages import HumanMessage
 
+browser_tool = HyperbrowserBrowserUseTool()
+tools = [browser_tool]
+llm = HuggingFaceEndpoint(repo_id="Qwen/Qwen2.5-Coder-32B-Instruct")
+chat = ChatHuggingFace(llm=llm)
+llm_with_tools = chat.bind_tools(tools, parallel_tool_calls=False)
 
 class State(TypedDict):
     input_file_path = Optional[str]
     messages = Annotated[list[AnyMessage], add_messages]
 
+def assistant(state:State):
+    system_message = "You are a helpful assistant. Your job is to answer the questions asked of you as accurately as possible. You have access to a browser search tool, which you may use when needed to answer a question."
+    return {
+        "messages": [llm_with_tools.invoke([system_message] + state["messages"])]
+    }
+    
 # (Keep Constants as is)
 # --- Constants ---
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
@@ -21,17 +36,22 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 # ----- THIS IS WHERE YOU CAN BUILD WHAT YOU WANT ------
 class BasicAgent:
     def __init__(self):
-        llm = HuggingFaceEndpoint(repo_id="Qwen/Qwen2.5-Coder-32B-Instruct")
-        chat = ChatHuggingFace(llm=llm, verbose=True)
-        messages = [("system", "You are a helpful assistant."),
-        ("human", "Hey how are you?")]
-        print(chat.invoke(messages))
+        workflow = StateGraph(state)
+        workflow.add_node("assistant", assistant)
+        workflow.add_node("tools", ToolNode([tools]))
+        workflow.add_edge(START, "assistant")
+        workflow.add_conditional_edge("assistant", tools_condition)
+        workflow.add_edge("tools", "assistant")
+        app = workflow.compile()
         print("BasicAgent initialized.")
     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}")
-        return fixed_answer
+        # fixed_answer = "This is a default answer."
+        messages = [HumanMessage(question)]
+        answer = app.invoke({"messages": messages})
+        # print(f"Agent returning fixed answer: {fixed_answer}")
+        print(f"Agent returning fixed answer: {answer}")
+        return answer
 
 def run_and_submit_all( profile: gr.OAuthProfile | None):
     """