Update app.py

app.py

@@ -4,12 +4,28 @@ import requests
 import pytz
 import yaml
 from tools.final_answer import FinalAnswerTool
+from tools.ocr import ocr_tool
+
 
 from Gradio_UI import GradioUI
 
 # Below is an example of a tool that does nothing. Amaze us with your creativity !
 
 
+@tool 
+def ocr_tool(image_path: str) -> str:
+    """Extracts text from images or scanned documents using OCR.
+    Args:
+        image_path: Path to the image file
+    """
+    try:
+        ocr_engine = PaddleOCR(use_angle_cls=True, lang='en')
+        result = ocr_engine.ocr(image_path, cls=True)
+        texts = [line[1][0] for line in result[0]] if result else []
+        return "\n".join(texts)
+    except Exception as e:
+        return f"OCR Error: {str(e)}"
+
 
 @tool
 def get_current_time_in_timezone(timezone: str) -> str: