Update app.py

app.py

@@ -3,6 +3,12 @@ import gradio as gr
 from transformers import pipeline
 import torch
 import ffmpeg
+import nltk
+import re
+from deep_translator import MyMemoryTranslator
+
+# You only need to run this download command once
+nltk.download('punkt')
 
 # --- Model Loading ---
 # We load the model once when the app starts, not on every function call.
@@ -121,7 +127,8 @@ def translate_local(text_to_translate, target_lang='ta-IN', device=None):
 def main_run(video_path):
     original_audio_file = extract_audio_from_video(video_path)
     original_text = transcribe_audio(original_audio_file)
-    return original_text
+    translated_text = translate_local(original_text)
+    return translated_text
     
 # --- Gradio Interface Definition ---
 # Title and description for the new Space
@@ -141,7 +148,7 @@ app_interface = gr.Interface(
     inputs=gr.Video(
         label="Upload Video"
     ),
-    outputs=gr.Textbox(label="Transcription Result"),
+    outputs=gr.Textbox(label="Translation Result"),
     title=title,
     description=description,
     article=article,