Create app.py

app.py

@@ -0,0 +1,61 @@
+import gradio as gr
+import subprocess
+import os
+import shutil
+import uuid
+from transformers import pipeline
+from gtts import gTTS
+
+
+def translate_video(file_path):
+    
+    try:
+        audio_path = os.path.join(file_path, "audio.wav")
+
+        if not os.path.exists(audio_path):
+             raise FileNotFoundError("Audio extraction failed. yt-dlp did not produce a .wav file.")
+
+        # 3. Translate the audio using the whisper-tiny model
+        translator = pipeline(
+            "automatic-speech-recognition",
+            model="openai/whisper-tiny",
+            device="cpu"
+        )
+        
+        translation = translator(audio_path, return_timestamps=True, generate_kwargs={"task": "translate"})
+        
+        translated_text = translation["text"]
+
+        if not translated_text:
+            return "No speech was detected in the video.", None, video_path
+
+        # 4. Convert translated text to speech using gTTS
+        tts = gTTS(translated_text.strip(), lang='en')
+        translated_audio_path = os.path.join(temp_dir, "translated_audio.mp3")
+        tts.save(translated_audio_path)
+
+        return translated_text, translated_audio_path, video_path
+
+    except Exception as e:
+        gr.Warning(f"An unexpected error occurred: {str(e)}")
+        return f"An error occurred: {str(e)}", None, None
+
+# Create the Gradio interface
+iface = gr.Interface(
+    fn=translate_video,
+    inputs=gr.Video(label="Upload your video to translate"),
+    outputs=[
+        gr.Textbox(label="Translated Text", interactive=False),
+        gr.Audio(label="Translated Audio"),
+        gr.Video(label="Original Video"),
+    ],
+    title="Twitter/X Video Translator",
+    description="Enter a link to a Twitter/X video to translate its audio to English. Handles videos longer than 30 seconds.",
+    allow_flagging="never",
+    examples=[["https://x.com/OpenAI/status/1790145460393734444"]]
+)
+
+if __name__ == "__main__":
+    if not os.path.exists("downloads"):
+        os.makedirs("downloads")
+    iface.launch()