Update app.py

app.py

@@ -219,8 +219,12 @@ def call_tools(state: AgentState) -> AgentState:
     elif label == "youtube":
         print("[TOOL] youtube_transcript")
         if matched_obj:
-            url = matched_obj[0]
-            state["context"] = get_youtube_transcript.invoke({"url": url})
+            url = re.sub(r'[.,;:!?")]+$', '', matched_obj.group(0))
+            print(f"[TOOL] fetching transcript for: {url}")
+            state["context"] = get_youtube_transcript.invoke({"video_url": url})
+        else:
+            print("[TOOL] youtube label but no URL found — falling back to web search")
+            state["context"] = web_search.invoke({"query": question})
     elif label == "research":
         print("[TOOL] web search")
         search_json = web_search.invoke({"query": question})
@@ -232,6 +236,10 @@ def call_tools(state: AgentState) -> AgentState:
     return state
 
 def synthesize_response(state: AgentState) -> AgentState:
+    # Skip if a tool already produced a direct answer (image / python / excel paths)
+    if state.get("answer"):
+        print(f"[SYNTHESIZE] skipped — answer already set by tool")
+        return state
     prompt = [
         SystemMessage(content=get_prompt("final_llm_system")),
         HumanMessage(
@@ -369,7 +377,7 @@ def _answer_question(item: dict) -> str:
             continue
         for attempt in range(2):
             try:
-                result = LGAgent(model_id=model_id)(augmented_question)
+                result = LGAgent(model_id=model_id)(augmented_question, task_id=task_id)
                 # Pause between questions to respect Groq's tokens/min limit
                 time.sleep(5)
                 return result