Update app.py

app.py

@@ -1,68 +1,41 @@
 import gradio as gr
 import easyocr
-from deep_translator import GoogleTranslator
-from PIL import Image, ImageDraw, ImageFont
+from libretranslatepy import LibreTranslateAPI
+from PIL import Image
 import numpy as np
 import cv2
 
-# Load OCR reader
+# Initialize OCR reader and translator
 reader = easyocr.Reader(['en', 'hi'])
+lt = LibreTranslateAPI("https://libretranslate.de")
 
+def process_image(img):
+    # Convert PIL image to NumPy array
+    img_array = np.array(img)
+    img_gray = cv2.cvtColor(img_array, cv2.COLOR_RGB2GRAY)
 
+    result = reader.readtext(img_gray)
 
-def translate_image(image, src_lang, tgt_lang):
-    # Convert PIL Image to OpenCV format
-    img_np = np.array(image)
-    img_cv = cv2.cvtColor(img_np, cv2.COLOR_RGB2BGR)
+    extracted_text = ""
+    translated_text = ""
 
-    # OCR
-    results = reader.readtext(img_cv)
+    for detection in result:
+        text = detection[1]
+        extracted_text += text + " "
 
-    if not results:
-        return "No text detected.", image
+    # Translate extracted text to Hindi
+    if extracted_text.strip():
+        translated_text = lt.translate(extracted_text, "en", "hi")
 
-    draw = ImageDraw.Draw(image)
-    font = ImageFont.truetype("arial.ttf", 22)
-
-    for (bbox, text, prob) in results:
-        # Translate text
-        try:
-            translated = GoogleTranslator(source='auto', target='hi').translate("Hello")
-        except:
-            translated = text  # fallback
-
-        # Coordinates
-        top_left = tuple(map(int, bbox[0]))
-        bottom_right = tuple(map(int, bbox[2]))
-
-        # Draw rectangle over original text (optional blur it)
-        draw.rectangle([top_left, bottom_right], fill="white")
-
-        # Write translated text
-        draw.text(top_left, translated, fill="black", font=font)
-
-    return image
-
-# Language options for dropdown (EasyOCR short codes → Googletrans short codes)
-lang_map = {
-    "English": "en",
-    "Hindi": "hi",
-    "French": "fr",
-    "Spanish": "es",
-    "German": "de",
-    "Chinese": "zh-cn"
-}
+    return extracted_text.strip(), translated_text.strip()
 
+# Gradio interface
 iface = gr.Interface(
-    fn=translate_image,
-    inputs=[
-        gr.Image(type="pil", label="Upload Image"),
-        gr.Dropdown(list(lang_map.keys()), label="From Language", value="Hindi"),
-        gr.Dropdown(list(lang_map.keys()), label="To Language", value="English"),
-    ],
-    outputs=gr.Image(type="pil", label="Translated Image"),
-    title="Photo Translate",
-    description="Upload an image with text. It will detect the text, translate it, and regenerate the image with the translated text."
+    fn=process_image,
+    inputs=gr.Image(type="pil"),
+    outputs=["text", "text"],
+    title="Image Text Translator (English to Hindi)",
+    description="Upload an image with English text, and get Hindi translation using LibreTranslate (no API key needed)."
 )
 
 iface.launch()