Update app.py

app.py

@@ -1,11 +1,7 @@
 import gradio as gr
 import subprocess
 import traceback
-import os
-from transformers import MarianMTModel, MarianTokenizer
-import torch
-import numpy as np
-import wave
+from deep_translator import GoogleTranslator
 
 # -------------------------------
 # 1. Audio extraction
@@ -17,16 +13,7 @@ def extract_audio(video_path):
     return audio_path
 
 # -------------------------------
-# 2. Read WAV
-# -------------------------------
-def read_wav(path):
-    with wave.open(path, "rb") as wf:
-        frames = wf.readframes(wf.getnframes())
-        audio = np.frombuffer(frames, dtype=np.int16)
-    return audio
-
-# -------------------------------
-# 3. Simple STT using Silero
+# 2. STT with Silero
 # -------------------------------
 def stt(audio_path):
     try:
@@ -46,31 +33,19 @@ def stt(audio_path):
         return f"STT Error: {traceback.format_exc()}"
 
 # -------------------------------
-# 4. Translation using MarianMT
+# 3. Translation
 # -------------------------------
 def translate_text(text, target_lang):
     if target_lang == "original":
         return text
-
-    lang_map = {
-        "en": "en",
-        "ur": "ur",
-        "hi": "hi",
-        "ps": "ps",
-        "ar": "ar"
-    }
-    tgt = lang_map.get(target_lang, "en")
-    model_name = f"Helsinki-NLP/opus-mt-en-{tgt}" if tgt != "en" else "Helsinki-NLP/opus-mt-en-en"
-    tokenizer = MarianTokenizer.from_pretrained(model_name)
-    model = MarianMTModel.from_pretrained(model_name)
-
-    batch = tokenizer([text], return_tensors="pt", padding=True)
-    translated = model.generate(**batch)
-    translated_text = tokenizer.decode(translated[0], skip_special_tokens=True)
-    return translated_text
+    try:
+        translated = GoogleTranslator(source='auto', target=target_lang).translate(text)
+        return translated
+    except Exception as e:
+        return f"Translation Error: {traceback.format_exc()}"
 
 # -------------------------------
-# 5. Main processing
+# 4. Main processing
 # -------------------------------
 def process_video(video, lang):
     try:
@@ -88,7 +63,7 @@ def process_video(video, lang):
         return "", f"❌ ERROR:\n{traceback.format_exc()}"
 
 # -------------------------------
-# 6. Gradio UI
+# 5. Gradio UI
 # -------------------------------
 languages = {
     "original": "Original",
@@ -100,7 +75,7 @@ languages = {
 }
 
 with gr.Blocks() as demo:
-    gr.Markdown("## 🎬 Video → Text → Translation (No API, Offline, CPU-friendly)")
+    gr.Markdown("## 🎬 Video → Text → Translation (Token-free, CPU, HF Spaces Free)")
 
     video_input = gr.Video(label="Upload Video")
     lang_dropdown = gr.Dropdown(list(languages.keys()), value="original", label="Translate To")