Update app.py

app.py

@@ -41,16 +41,47 @@ LANG_CODE_MAP = {
 }
 
 def initialize_reader():
-    """Initialize EasyOCR reader with optimized settings"""
+    """Initialize EasyOCR reader with fallback options"""
     global reader
     if reader is None:
-        try:
-            # Initialize with multiple languages for better detection
-            reader = easyocr.Reader(['en', 'hi', 'es', 'fr', 'de'], gpu=False, verbose=False)
-            print("EasyOCR initialized successfully")
-        except Exception as e:
-            print(f"Error initializing EasyOCR: {e}")
-            return None
+        # Try different initialization strategies
+        init_strategies = [
+            (['en', 'hi'], "English and Hindi"),
+            (['en'], "English only"),
+            (['en', 'hi'], "English and Hindi with verbose"),
+        ]
+        
+        for i, (languages, description) in enumerate(init_strategies):
+            try:
+                print(f"Attempting OCR initialization: {description}")
+                verbose_setting = True if i == 2 else False
+                
+                reader = easyocr.Reader(
+                    languages, 
+                    gpu=False, 
+                    verbose=verbose_setting,
+                    download_enabled=True,
+                    detector=True,
+                    recognizer=True
+                )
+                print(f"✅ EasyOCR initialized successfully with {description}")
+                return reader
+                
+            except ImportError as e:
+                print(f"❌ Import error: {e}")
+                continue
+            except Exception as e:
+                print(f"❌ Initialization attempt {i+1} failed: {e}")
+                if i < len(init_strategies) - 1:
+                    print("Trying alternative approach...")
+                    continue
+                else:
+                    print("All initialization strategies failed")
+        
+        # If all strategies fail, return None
+        reader = None
+        print("❌ Could not initialize EasyOCR with any strategy")
+    
     return reader
 
 def calculate_distance(box1, box2):
@@ -397,22 +428,70 @@ def process_image_enhanced(image: Image.Image, target_language: str, progress=gr
     
     progress(0.1, "🔧 Initializing OCR engine...")
     
-    # Initialize OCR
-    ocr = initialize_reader()
-    if ocr is None:
-        return image, "❌ Failed to initialize OCR. Please try again."
+    # Initialize OCR with better error handling
+    try:
+        ocr = initialize_reader()
+        if ocr is None:
+            return image, """❌ OCR initialization failed. This might be due to:
+• Missing system dependencies
+• Network issues downloading models
+• Insufficient memory
+
+Please try refreshing the page or contact support."""
+        
+        # Test OCR with a simple operation
+        test_array = np.array(image.convert('RGB'))
+        if test_array.size == 0:
+            return image, "❌ Invalid image format. Please upload a valid image file."
+            
+    except Exception as e:
+        error_details = str(e)
+        return image, f"""❌ OCR Setup Error: {error_details}
+
+Possible solutions:
+• Refresh the browser and try again
+• Upload a different image format (JPG/PNG)
+• Check if the image is not corrupted
+
+Technical details: {type(e).__name__}"""
     
     progress(0.3, "🔍 Extracting and grouping text regions...")
     
     try:
-        # Convert PIL image to numpy array
-        img_array = np.array(image)
+        # Convert PIL image to numpy array with error handling
+        img_array = np.array(image.convert('RGB'))
+        
+        if img_array is None or img_array.size == 0:
+            return image, "❌ Error processing image. Please try a different image."
         
-        # Perform OCR with higher confidence threshold
-        results = ocr.readtext(img_array, detail=1, paragraph=False)
+        print(f"Image shape: {img_array.shape}")
+        
+        # Perform OCR with error handling and fallback options
+        try:
+            results = ocr.readtext(img_array, detail=1, paragraph=False, width_ths=0.7, height_ths=0.7)
+        except Exception as ocr_error:
+            print(f"Primary OCR failed: {ocr_error}")
+            # Fallback: try with different parameters
+            try:
+                results = ocr.readtext(img_array, detail=1)
+            except Exception as fallback_error:
+                print(f"Fallback OCR failed: {fallback_error}")
+                return image, f"""❌ OCR Processing Failed: {str(ocr_error)}
+
+Troubleshooting:
+• Image might be too complex or low quality
+• Try uploading a clearer image
+• Ensure text is clearly visible
+
+Fallback error: {str(fallback_error)}"""
         
         if not results:
-            return image, "ℹ️ No readable text found in the image."
+            return image, """ℹ️ No readable text found in the image.
+
+Tips for better results:
+• Ensure text is clearly visible and well-lit
+• Upload higher resolution images
+• Make sure text is not too small or blurry"""
         
         # Filter by confidence
         filtered_results = [(bbox, text, conf) for bbox, text, conf in results 
@@ -619,14 +698,32 @@ with gr.Blocks(css=custom_css, title="Enhanced Multilingual Signboard Translator
 
 if __name__ == "__main__":
     print("🔧 Initializing Enhanced OCR Translator...")
+    print("System Information:")
+    print(f"Python version: {os.sys.version}")
+    print(f"NumPy version: {np.__version__}")
+    
+    # Pre-initialize with detailed logging
     try:
-        initialize_reader()
-        print("✅ System ready!")
+        print("Starting OCR initialization...")
+        ocr_reader = initialize_reader()
+        if ocr_reader:
+            print("✅ OCR System ready!")
+        else:
+            print("⚠️ OCR initialization failed - will retry when needed")
     except Exception as e:
-        print(f"⚠️ Initialization warning: {e}")
+        print(f"⚠️ Pre-initialization error: {e}")
+        print("OCR will be initialized on first use")
     
-    demo.launch(
-        share=True,
-        show_error=True,
-        server_name="0.0.0.0"
-    )
+    # Launch with better error handling
+    try:
+        demo.launch(
+            share=True,
+            show_error=True,
+            server_name="0.0.0.0",
+            server_port=7860,
+            enable_queue=True
+        )
+    except Exception as e:
+        print(f"Launch error: {e}")
+        # Fallback launch
+        demo.launch()