Upload app.py

app.py

@@ -167,10 +167,6 @@ def stream_ocr(image):
         return
 
     try:
-        # Load image if it's a filepath (compatibility for type="filepath")
-        if isinstance(image, str):
-            image = Image.open(image).convert("RGB")
-            
         # Resize - Crucial for CPU speed
         valid_image = resize_for_ocr(image, max_dim=896)
 
@@ -243,12 +239,11 @@ data_dir = os.path.join(base_dir, 'data')
 if os.path.exists(data_dir):
     valid_exts = {".jpg", ".jpeg", ".png", ".bmp", ".webp"}
     
-    # Use ABSOLUTE paths. Gradio's `allowed_paths` will handle serving them.
-    # Relative paths often cause issues with component population in Spaces.
+    # Found files list
     found_files = [f for f in os.listdir(data_dir) if os.path.splitext(f)[1].lower() in valid_exts]
     print(f"DEBUG: Found {len(found_files)} images in {data_dir}")
     
-    # Use full paths. Since we allow `base_dir` in launch(), these will be servable.
+    # Use ABSOLUTE paths for robustness (matches app.py logic which works)
     example_images = [[os.path.join(data_dir, f)] for f in found_files]
     
     # Limit to 5 examples to prevent UI clutter if many files exist
@@ -351,9 +346,10 @@ with gr.Blocks(title="Ultra OCR", theme=theme, css=custom_css) as demo:
         with gr.Row(equal_height=False, variant="panel"):
             with gr.Column(scale=4):
                 input_img = gr.Image(
-                    type="filepath", 
+                    type="pil", 
                     label="📄 Document Source", 
                     height=500,
+                    sources=['upload', 'clipboard'],
                     format="png"
                 )
                 run_btn = gr.Button("⚡ Start Transcription", variant="primary", size="lg")
@@ -390,12 +386,7 @@ with gr.Blocks(title="Ultra OCR", theme=theme, css=custom_css) as demo:
 
 if __name__ == "__main__":
     # Fix for spaces visual/SSR errors: Disable SSR
-    # explicitly allow the data directory and the current directory
-    allowed = [os.path.dirname(os.path.abspath(__file__))]
-    if os.path.exists(data_dir):
-        allowed.append(data_dir)
-        
     demo.launch(
-        allowed_paths=allowed, 
+        allowed_paths=[os.path.dirname(os.path.abspath(__file__))], 
         ssr_mode=False
     )