Update app.py

app.py

@@ -1,17 +1,21 @@
 import gradio as gr
 from paddleocr import PaddleOCR
-from PIL import Image
+from PIL import Image, ImageOps
 import numpy as np
 
 # Initialize PaddleOCR
 ocr = PaddleOCR(use_angle_cls=True, lang='en')
 
-# Function to process uploaded image
-def process_uploaded_image(image):
+# Function to process uploaded/captured images
+def process_uploaded_image(image, camera_type="rear"):
     if image is None:
         return "No image uploaded."
     
-    # Perform OCR on the uploaded image
+    # Mirror the image for rear camera
+    if camera_type == "rear":
+        image = ImageOps.mirror(image)
+    
+    # Perform OCR on the corrected image
     result = ocr.ocr(np.array(image))
     extracted_text = []
     for line in result[0]:
@@ -19,12 +23,31 @@ def process_uploaded_image(image):
     return "\n".join(extracted_text)
 
 # Gradio Interface
-app = gr.Interface(
-    fn=process_uploaded_image,
-    inputs=gr.Image(type="pil", label="Upload or Capture Image"),
-    outputs=gr.Textbox(label="Extracted Text"),
-    title="OCR Processor"
-)
+def create_app():
+    with gr.Blocks() as app:
+        gr.Markdown("## OCR Processor with Rear Camera Fix")
+        
+        # Input for image capture and camera type selection
+        with gr.Row():
+            image_input = gr.Image(source="webcam", type="pil", label="Capture Image from Webcam")
+            camera_type = gr.Dropdown(
+                choices=["rear", "front"],
+                value="rear",
+                label="Camera Type"
+            )
+        
+        # Output for extracted text
+        extracted_text = gr.Textbox(label="Extracted Text")
+        
+        # Button to process the image
+        process_button = gr.Button("Process Image")
+        process_button.click(
+            fn=process_uploaded_image,
+            inputs=[image_input, camera_type],
+            outputs=extracted_text
+        )
+        
+    return app
 
 if __name__ == "__main__":
-    app.launch()
+    create_app().launch(share=True)