Update app.py

app.py

@@ -3,10 +3,11 @@ import cv2
 import numpy as np
 import tempfile
 import os
-import easyocr
+# import easyocr
 from PIL import Image, ImageDraw, ImageFont
 from deep_translator import GoogleTranslator
 import base64
+from paddleocr import PaddleOCR
 
 from langchain.prompts import PromptTemplate
 from langchain.chains import LLMChain
@@ -110,10 +111,10 @@ def save_text_as_image(text, file_path):
 
 set_background("background_img.jpg")
 
-# OCR
-@st.cache_resource
-def load_easyocr_reader():
-    return easyocr.Reader(['en'])
+# # OCR
+# @st.cache_resource
+# def load_easyocr_reader():
+#     return easyocr.Reader(['en'])
 
 st.sidebar.title("💊 MediAssist")
 st.sidebar.markdown("Analyze prescriptions with ease using AI")
@@ -155,10 +156,17 @@ if uploaded_file:
     dilated_path = orig_path.replace(".png", "_dilated.png")
     cv2.imwrite(dilated_path, dilated)
 
-    reader = load_easyocr_reader()
-    text_list = reader.readtext(dilated, detail=0)
+    
+    ocr = PaddleOCR(use_angle_cls=True, lang='en')  # use_angle_cls for better orientation handling
+    result = ocr.ocr(dilated_path, cls=True)
+    text_list = [line[1][0] for line in result[0]]  # Extract only text
+    
     text = "\n".join(text_list)
 
+    # reader = easyocr.Reader(['en'])
+    # text_list = reader.readtext(dilated, detail=0)
+    # text = "\n".join(text_list)
+
     col1, col2 = st.columns([1, 2])
     with col1:
         st.image(dilated, caption="🧾 Preprocessed Prescription", channels="GRAY", use_container_width=True)