Update app.py

app.py

@@ -1,64 +1,32 @@
-# app.py
 import gradio as gr
 import easyocr
-from transformers import MarianMTModel, MarianTokenizer
-
-# OCR Reader Initialization
-reader = easyocr.Reader(['en', 'hi', 'fr', 'de', 'es', 'ru'], gpu=False)  # Add more if needed
-
-# Supported Languages for Translation
-LANGUAGE_CODES = {
-    "English": "en",
-    "Hindi": "hi",
-    "French": "fr",
-    "German": "de",
-    "Spanish": "es",
-    "Russian": "ru"
-}
-
-# Function to load MarianMT model
-model_cache = {}
-def get_model(target_lang):
-    model_name = f"Helsinki-NLP/opus-mt-ROMANCE-{target_lang}" if target_lang in ['fr', 'es', 'ro', 'pt'] else f"Helsinki-NLP/opus-mt-en-{target_lang}"
-    if model_name not in model_cache:
-        tokenizer = MarianTokenizer.from_pretrained(model_name)
-        model = MarianMTModel.from_pretrained(model_name)
-        model_cache[model_name] = (tokenizer, model)
-    return model_cache[model_name]
-
-# Main function
-def translate_image_text(image, target_lang):
-    try:
-        # OCR
-        result = reader.readtext(image, detail=0, paragraph=True)
-        extracted_text = " ".join(result)
-
-        if not extracted_text.strip():
-            return "No text found in the image."
-
-        # Get model
-        code = LANGUAGE_CODES[target_lang]
-        tokenizer, model = get_model(code)
-
-        # Translation
-        batch = tokenizer([extracted_text], return_tensors="pt", padding=True)
-        gen = model.generate(**batch)
-        translated = tokenizer.batch_decode(gen, skip_special_tokens=True)[0]
-
-        return translated
-    except Exception as e:
-        return f"Error: {str(e)}"
-
-# Gradio Interface
-iface = gr.Interface(
-    fn=translate_image_text,
+from googletrans import Translator
+
+# OCR reader with Hindi + English (Devanagari fix)
+reader = easyocr.Reader(['hi', 'en'])  # You can change 'hi' to 'mr' or 'ne' etc.
+translator = Translator()
+
+# 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
+
+# Gradio UI
+interface = gr.Interface(
+    fn=ocr_and_translate,
     inputs=[
         gr.Image(type="filepath", label="Upload Image"),
-        gr.Dropdown(choices=list(LANGUAGE_CODES.keys()), label="Translate To")
+        gr.Dropdown(
+            ["en", "hi", "mr", "ne", "fr", "de", "es", "bn", "gu", "ta", "te"],
+            label="Translate To",
+            value="hi"
+        )
     ],
-    outputs=gr.Textbox(label="Translated Text"),
+    outputs="text",
     title="Image Text Translator",
-    description="Upload an image containing text, and choose a language to translate the extracted text."
+    description="Upload an image with text, and translate extracted text to your chosen language."
 )
 
-iface.launch()
+interface.launch()