Update app.py

app.py

@@ -57,7 +57,30 @@ def get_default_font(size=20):
 
 def translate_text(text, target_lang):
     """Translate text with error handling and caching"""
-    # ... keep existing code (translate_text function)
+    if not text or not text.strip():
+        return ""
+    
+    # Use cache if available
+    cache_key = f"{text}|{target_lang}"
+    if cache_key in translation_cache:
+        return translation_cache[cache_key]
+    
+    # Handle translation with retries
+    max_retries = 3
+    for attempt in range(max_retries):
+        try:
+            translated = GoogleTranslator(source='auto', target=target_lang).translate(text)
+            if translated:
+                translation_cache[cache_key] = translated
+                return translated
+            time.sleep(1)  # Short delay before retry
+        except Exception as e:
+            print(f"Translation error (attempt {attempt+1}): {e}")
+            if attempt == max_retries - 1:
+                return f"[Translation Error: {text}]"
+            time.sleep(1)  # Wait before retry
+    
+    return f"[Unable to translate: {text}]"
 
 def get_dominant_color(image, bbox, padding=4):
     """Extract the dominant color around text for better background matching"""
@@ -224,7 +247,53 @@ def process_image(image, target_lang, overlay_opacity=0.7, progress=gr.Progress(
 
 # Create Gradio interface
 with gr.Blocks(title="Enhanced Image Translator") as iface:
-    # ... keep existing code (Gradio interface setup)
+    gr.Markdown("# Enhanced Image Translator")
+    gr.Markdown("Extract & translate text from images with improved overlay visualization")
+    
+    with gr.Row():
+        with gr.Column():
+            input_image = gr.Image(type="pil", label="Upload Image")
+            
+            with gr.Row():
+                target_lang = gr.Dropdown(
+                    choices=list(SUPPORTED_LANGUAGES.values()),
+                    value="Hindi",
+                    label="Translate To"
+                )
+                overlay_opacity = gr.Slider(
+                    minimum=0.1, 
+                    maximum=1.0, 
+                    value=0.7, 
+                    step=0.1, 
+                    label="Overlay Opacity"
+                )
+            
+            translate_btn = gr.Button("Translate", variant="primary")
+            
+        with gr.Column():
+            output_image = gr.Image(type="pil", label="Image with Translated Overlay")
+            output_text = gr.Textbox(label="Translated Text Output", lines=10)
+    
+    # Connect the button to the processing function
+    translate_btn.click(
+        fn=process_image,
+        inputs=[input_image, target_lang, overlay_opacity],
+        outputs=[output_image, output_text]
+    )
+    
+    gr.Markdown("""
+    ## Features
+    - Supports multiple languages for translation
+    - Semi-transparent overlays for better readability
+    - Simple and efficient text extraction and translation
+    """)
 
 if __name__ == "__main__":
-    # ... keep existing code (initialization and app launch)
+    # Initialize OCR model at startup to avoid delay on first request
+    try:
+        initialize_reader()
+    except:
+        pass
+    
+    # Launch the app
+    iface.launch()