audio

app.py

@@ -21,7 +21,6 @@ from state import AgentState
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
 from tools import ocr_image_tool, parse_excel_tool, web_search_tool, run_tools, audio_transcriber_tool
-tool_node = ToolNode([ocr_image_tool, parse_excel_tool, web_search_tool, audio_transcriber_tool])
 
 llm = ChatOpenAI(model_name="gpt-4.1-mini", temperature=0.0)
 
@@ -121,7 +120,12 @@ def finalize_node(state: AgentState) -> AgentState:
     return {"final_answer": llm_response.content.strip()}
 
 # ─── 4) Wrap tools in a ToolNode ───
-tool_node = ToolNode([web_search_tool, ocr_image_tool, parse_excel_tool])
+tool_node = ToolNode([
+    web_search_tool,
+    ocr_image_tool,
+    parse_excel_tool,
+    audio_transcriber_tool
+])
 
 # Add a node to store the previous state
 
@@ -149,8 +153,15 @@ graph.add_edge(START, "plan")
 
 # 5.c) plan → conditional: if any tool key was set, go to "tools"; otherwise "finalize"
 def route_plan(plan_out: AgentState) -> str:
-    if plan_out.get("web_search_query") or plan_out.get("ocr_path") or plan_out.get("excel_path"):
+    if (
+        plan_out.get("web_search_query")
+        or plan_out.get("ocr_path")
+        or plan_out.get("excel_path")
+        or plan_out.get("audio_path")     # newly added
+    ):
+        print("Going to tools")
         return "tools"
+    print("Going to finalize")
     return "finalize"
 
 graph.add_conditional_edges(

tools.py

@@ -12,6 +12,7 @@ def web_search_tool(state: AgentState) -> AgentState:
     Returns: {"web_search_query": None, "web_search_result": <string>}
     We also clear web_search_query so we don’t loop forever.
     """
+    print("reached web search tool")
     query = state.get("web_search_query", "")
     if not query:
         return {}  # nothing to do
@@ -30,6 +31,7 @@ def ocr_image_tool(state: AgentState) -> AgentState:
     Expects: state["ocr_path"] is a path to an image file.
     Returns: {"ocr_path": None, "ocr_result": <string>}.
     """
+    print("reached ocr image tool")
     path = state.get("ocr_path", "")
     if not path:
         return {}
@@ -51,6 +53,7 @@ def parse_excel_tool(state: AgentState) -> AgentState:
              and state["excel_sheet_name"] optionally names a sheet.
     Returns: {"excel_path": None, "excel_sheet_name": None, "excel_result": <string>}.
     """
+    print("reached parse excel tool")
     path = state.get("excel_path", "")
     sheet = state.get("excel_sheet_name", "")
     if not path:
@@ -106,6 +109,7 @@ def audio_transcriber_tool(state: AgentState) -> AgentState:
       }
     If no valid audio_path is provided, returns {}.
     """
+    print("reached audio transcriber tool")
     path = state.get("audio_path", "")
     if not path or not os.path.exists(path):
         return {}