Update tools.py

tools.py

@@ -104,16 +104,16 @@ def ocr_image_tool(state: AgentState) -> AgentState:
     """
     print("reached ocr_image_tool")
     path_or_id = state.get("ocr_path", "")
-    if not path_or_id:
-        return {}
+    # if not path_or_id:
+    #     return {}
 
     # 1) Determine local_img: either existing path_or_id or download by Task ID
-    local_img = ""
-    if os.path.exists(path_or_id):
-        local_img = path_or_id
-    else:
-        for ext in ("png", "jpg", "jpeg"):
-            candidate = _download_file_for_task(path_or_id, ext)
+    # local_img = ""
+    # if os.path.exists(path_or_id):
+    #     local_img = path_or_id
+    # else:
+    for ext in ("png", "jpg", "jpeg"):
+            candidate = _download_file_for_task(state.get("task_id"), ext)
             if candidate:
                 local_img = candidate
                 break
@@ -185,7 +185,7 @@ def ocr_image_tool(state: AgentState) -> AgentState:
 
     # 6) Combine OCR + caption
     combined = f"OCR text:\n{ocr_text}\n\nImage caption:\n{caption}"
-
+    print("combined: ")
     return {
         "ocr_path": None,
         "ocr_result": combined