Update app.py

app.py

@@ -126,12 +126,18 @@ def translate_local(text_to_translate, target_lang='ta-IN', device=None):
 
     translated_text = " ".join(translated_chunks)
 
-def main_run(video_path):
+def main_run(video_path,target_lang):
     original_audio_file = extract_audio_from_video(video_path)
     original_text = transcribe_audio(original_audio_file)
-    translated_text = translate_local(original_text)
+    translated_text = translate_local(original_text,target_lang)
     return translated_text
-    
+
+def audio_pipeline_run(audio_path,target_lang):
+    original_text = transcribe_audio(audio_path)
+    translated_text = translate_local(original_text,target_lang)
+    return translated_text
+
+
 # --- Gradio Interface Definition ---
 # Title and description for the new Space
 title = "Custom Whisper Transcription App"
@@ -147,16 +153,51 @@ article = "<p style='text-align: center'><a href='https://huggingface.co/openai/
 # We define the input as an Audio component and the output as a Textbox
 app_interface = gr.Interface(
     fn=main_run,
-    inputs=gr.Video(
-        label="Upload Video"
-    ),
-    outputs=gr.Textbox(label="Translation Result"),
+    inputs=gr.Video(label="Upload Video"),gr.
+    outputs=gr.Textbox(label="Translation Result"),gr.Radio(choices=["Tamil", "Telugu", "Hindi"], label="Target Language", value="Tamil")
     title=title,
     description=description,
     article=article,
     allow_flagging="never"
 )
 
+with gr.Blocks(theme=Soft(), title="Audio/Video Translation Toolkit") as ui:
+    gr.Markdown("# 🚀 Audio/Video Translation Toolkit")
+    with gr.Tabs():
+        with gr.Tab("🎬 Translate Video"):
+            with gr.Row():
+                with gr.Column():
+                    video_in = gr.Video(label="Input Video", height=500)
+                    lang_radio_vid = gr.Radio(choices=["Tamil", "Telugu", "Hindi"], label="Target Language", value="Tamil")
+                    submit_btn_vid = gr.Button("Translate Video", variant="primary")
+
+                with gr.Column():
+                    video_out = gr.Textbox(label="Translation")
+
+                '''
+                with gr.Column():
+                    video_out = gr.Video(label="Output Video", interactive=False, height=500)
+                '''
+                    
+            submit_btn_vid.click(fn=main_run, inputs=[video_in, lang_radio_vid], outputs=[video_out])
+
+        with gr.Tab("🎵 Translate Audio"):
+            with gr.Row():
+                with gr.Column():
+                    audio_in_pipe = gr.Audio(type="filepath", label="Input Audio")
+                    lang_radio_aud = gr.Radio(choices=["Tamil", "Telugu", "Hindi"], label="Target Language", value="Tamil")
+                    submit_btn_aud = gr.Button("Translate Audio", variant="primary")
+
+                with gr.Column():
+                    audio_out_pipe = gr.Textbox(label="Translation")
+                
+                '''
+                with gr.Column():
+                    audio_out_pipe = gr.Audio(label="Output Audio", interactive=False)
+                '''
+                
+            submit_btn_aud.click(fn=audio_pipeline_run, inputs=[audio_in_pipe, lang_radio_aud], outputs=[audio_out_pipe])
+
 # --- Launch the App ---
 if __name__ == "__main__":
     # The launch() method creates a web server and makes the interface accessible.