Update app.py

app.py

@@ -1,32 +1,38 @@
 import gradio as gr
 import easyocr
-from googletrans import Translator
+from deep_translator import GoogleTranslator
 
-# OCR reader with Hindi + English (Devanagari fix)
-reader = easyocr.Reader(['hi', 'en'])  # You can change 'hi' to 'mr' or 'ne' etc.
-translator = Translator()
+reader = easyocr.Reader(['en', 'hi'])  # Hindi aur English OCR ke liye
 
-# Function to extract text and translate
-def ocr_and_translate(image, target_lang):
-    result = reader.readtext(image)
-    extracted_text = " ".join([text[1] for text in result])
-    translated = translator.translate(extracted_text, dest=target_lang)
-    return translated.text
+def extract_and_translate(image, target_lang):
+    result = reader.readtext(image, detail=0)
+    text = ' '.join(result)
+    try:
+        translated_text = GoogleTranslator(source='auto', target=target_lang).translate(text)
+    except Exception as e:
+        translated_text = f"Translation Error: {str(e)}"
+    return translated_text
 
-# Gradio UI
-interface = gr.Interface(
-    fn=ocr_and_translate,
+lang_choices = {
+    "Hindi": "hi",
+    "English": "en",
+    "Punjabi": "pa",
+    "Gujarati": "gu",
+    "Bengali": "bn",
+    "Tamil": "ta",
+    "Telugu": "te",
+    "Marathi": "mr"
+}
+
+iface = gr.Interface(
+    fn=extract_and_translate,
     inputs=[
         gr.Image(type="filepath", label="Upload Image"),
-        gr.Dropdown(
-            ["en", "hi", "mr", "ne", "fr", "de", "es", "bn", "gu", "ta", "te"],
-            label="Translate To",
-            value="hi"
-        )
+        gr.Dropdown(choices=list(lang_choices.values()), label="Translate To")
     ],
     outputs="text",
     title="Image Text Translator",
-    description="Upload an image with text, and translate extracted text to your chosen language."
+    description="Upload an image with printed text. Extracts and translates text into selected language."
 )
 
-interface.launch()
+iface.launch()