Update app.py

app.py

@@ -6,6 +6,7 @@ from huggingface_hub import InferenceClient
 from dataclasses import dataclass
 import pytesseract
 from PIL import Image
+import easyocr
 
 @dataclass
 class ChatMessage:
@@ -40,8 +41,7 @@ class XylariaChat:
 
         # System prompt with more detailed instructions
         self.system_prompt = """You are a helpful and harmless assistant. You are Xylaria developed by Sk Md Saad Amin. You should think step-by-step. You should respond to image questions"""
-
-    def store_information(self, key, value):
+        self.reader = easyocr.Reader(['en']) 
         """Store important information in persistent memory"""
         self.persistent_memory[key] = value
         return f"Stored: {key} = {value}"
@@ -112,7 +112,7 @@ class XylariaChat:
 
     def perform_math_ocr(self, image_path):
         """
-        Perform OCR on an image and return the extracted text.
+        Perform OCR on an image using EasyOCR and return the extracted text.
         Args:
             image_path (str): Path to the image file.
         Returns:
@@ -122,11 +122,14 @@ class XylariaChat:
             # Open the image using Pillow library
             img = Image.open(image_path)
 
-            # Use Tesseract to do OCR on the image
-            text = pytesseract.image_to_string(img)
+            # Use EasyOCR to do OCR on the image
+            results = self.reader.readtext(image_path)
+
+            # Extract text from results (combining text from multiple detections)
+            extracted_text = ' '.join([result[1] for result in results])
 
             # Remove leading/trailing whitespace and return
-            return text.strip()
+            return extracted_text.strip()
 
         except Exception as e:
             return f"Error during Math OCR: {e}"
@@ -222,7 +225,7 @@ class XylariaChat:
     
     def create_interface(self):
         def streaming_response(message, chat_history, image_filepath, math_ocr_image_path):
-            
+
             ocr_text = ""
             if math_ocr_image_path:
                 ocr_text = self.perform_math_ocr(math_ocr_image_path)
@@ -232,7 +235,7 @@ class XylariaChat:
                     yield "", updated_history, None, None
                     return
                 else:
-                    message = f"Math OCR Result: {ocr_text}\n\nUser's message: {message}"
+                    message = f"Math OCR Result: {ocr_text}\n\nUser's message: {message}
 
             # Check if an image was actually uploaded
             if image_filepath: