Update app.py

app.py

@@ -26,9 +26,9 @@ import os
 # Function to perform OCR
 def ocr(input_file, lang='fas'):  # 'fas': Persian language (Farsi)
     extracted_text = ""
-    if isinstance(input_file, Image.Image):  # If the input is an image
-        text = pytesseract.image_to_string(input_file, lang=lang)
-        extracted_text = text
+    # if isinstance(input_file, Image.Image):  # If the input is an image
+    text = pytesseract.image_to_string(input_file, lang=lang)
+    extracted_text = text
 
     return extracted_text
 
@@ -82,9 +82,16 @@ def ocr_with_easy(img):
 Generate OCR
 """
 
-def process(input_type, file, lang):
+def process(input_type, gr_img, lang):
 
-    # image = Image.open(file.name)
+    if isinstance(gr_img, np.ndarray):
+    # Create a temporary file to save the image
+    with tempfile.NamedTemporaryFile(suffix=".png", delete=False) as temp_file:
+        temp_path = temp_file.name
+        # Save the NumPy array as an image file
+        Image.fromarray(gr_img).save(temp_path)
+    
+    image = Image.open(temp_path)
     image = file
     extracted_text = ocr(image, lang)
     return extracted_text