Update tools.py

tools.py

@@ -1,4 +1,6 @@
 import cmath
+import pytesseract
+from PIL import Image
 from langchain_core.tools import tool
 from langchain_community.document_loaders import WikipediaLoader
 
@@ -98,8 +100,28 @@ def wiki_search(query: str) -> str:
     )
     return {"wiki_results": formatted_search_docs}
 
+## OCR
+@tool
+def get_text_from_image(image_path: str) -> str:
+    """
+    Extract text from an image using OCR pytesseract.
+    Args:
+        image_path (str): the path to the image file.
+    """
+    try:
+        # Open the image
+        image = Image.open(image_path)
+        # Extract text from the image
+        text = pytesseract.image_to_string(image)
+        return f"Extracted text from image:\n\n{text}"
+    except ModuleNotFoundError as e:
+            return f'There is not module installed: {str(e)}'
+    except Exception as e:
+        return f"Error extracting text from image: {str(e)}"
+
 alltools = [
     wiki_search,
+    get_text_from_image,
     multiply,
     add,
     subtract,