Update app.py

app.py

@@ -1,24 +1,45 @@
-import gradio as gr, pytesseract, cv2, os
+import gradio as gr
+import pytesseract
+import cv2
+import os
 
-def process(image: str, lang: str = 'eng') -> str:
-    try:
-        img = cv2.imread(image) 
-        gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY) 
-        _,threshold_img = cv2.threshold(gray, 127, 255, cv2.THRESH_TOZERO)
-        result = pytesseract.image_to_string(threshold_img, lang=lang)
-        os.remove(image)
-        return result
-    except Exception as e:
-        return str(e)
+def process(images, lang: str = 'eng') -> str:
+    results = []
+    for image in images:
+        try:
+            img = cv2.imread(image)
+            gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
+            _, threshold_img = cv2.threshold(gray, 127, 255, cv2.THRESH_TOZERO)
+            result = pytesseract.image_to_string(threshold_img, lang=lang)
+            results.append(result)
+        except Exception as e:
+            results.append(f"Error processing {image}: {str(e)}")
+        finally:
+            os.remove(image)
+    return "\n\n".join(results)
 
+# Get the available languages for pytesseract
 langs = pytesseract.get_languages()
 
+# Define the Gradio interface
 interface = gr.Interface(
-    process,
-    [gr.Image(type="filepath"), gr.Dropdown(label="Select Language", choices=langs, type="value")],
+    fn=process,
+    inputs=[
+        gr.Image(type="filepath", multiple=True),
+        gr.Dropdown(label="Select Language", choices=langs, type="value")
+    ],
     outputs="text",
     css="footer {visibility: hidden}",
-    title="Optical Character Recognition | Image To Text",
-    article = """<p style='text-align: center;'>Hello, thanks for coming, visit AI tools: <a href="https://www.genelify.com" target="_blank">Genelify</a>, visit Social Media tools: <a href="https://www.tubtic.com" target="_blank">Tubtic</a></p>"""
+    title="Optical Character Recognition | Batch Image To Text",
+    article="""
+    <p style='text-align: center;'>
+        Hello, thanks for coming! Visit AI tools: 
+        <a href="https://www.genelify.com" target="_blank">Genelify</a>, 
+        and check out Social Media tools: 
+        <a href="https://www.tubtic.com" target="_blank">Tubtic</a>
+    </p>
+    """
 )
-interface.launch(show_api=False)
+
+# Launch the interface
+interface.launch(show_api=False)