Update tools.py

tools.py

@@ -3,7 +3,7 @@
 import pandas as pd
 
 from pathlib import Path
-
+import requests
 import regex as re
 import time
 import os
@@ -32,6 +32,7 @@ def _download_file_for_task(task_id: str, ext: str) -> str:
                 f.write(resp.content)
             return local_path
     except Exception:
+        print(f"Error downloading file from {url} to {local_path}")
         pass
 
     # If we get here, either 404 or download error
@@ -66,18 +67,14 @@ def ocr_image_tool(args: dict) -> str:
         with open(local_img, "rb") as f:
             image_bytes = f.read()
     except Exception as e:
-        return {
-            "ocr_path": None,
-            "ocr_result": f"Error reading image file: {e}"
-        }
+        return  f"Error reading image file: {e}"
+        
 
     # 3) Prepare HF Inference headers
     hf_token = os.getenv("HF_TOKEN")
     if not hf_token:
-        return {
-            "ocr_path": None,
-            "ocr_result": "Error: HUGGINGFACE_API_KEY not set in environment."
-        }
+        return "Error: HUGGINGFACE_API_KEY not set in environment."
+        
 
     headers = {"Authorization": f"Bearer {hf_token}"}