Update app.py

app.py

@@ -29,6 +29,14 @@ def assistant(state:State):
     return {
         "messages": [llm_with_tools.invoke([system_message] + state["messages"])]
     }
+
+workflow = StateGraph(State)
+workflow.add_node("assistant", assistant)
+workflow.add_node("tools", ToolNode(tools))
+workflow.add_edge(START, "assistant")
+workflow.add_conditional_edges("assistant", tools_condition)
+workflow.add_edge("tools", "assistant")
+app = workflow.compile()
     
 # (Keep Constants as is)
 # --- Constants ---
@@ -38,13 +46,6 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 # ----- THIS IS WHERE YOU CAN BUILD WHAT YOU WANT ------
 class BasicAgent:
     def __init__(self):
-        workflow = StateGraph(State)
-        workflow.add_node("assistant", assistant)
-        workflow.add_node("tools", ToolNode(tools))
-        workflow.add_edge(START, "assistant")
-        workflow.add_conditional_edges("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]}...")