Create app.py

app.py

@@ -0,0 +1,68 @@
+import os
+import zipfile
+import csv
+import moviepy.editor as mp
+from moviepy.editor import concatenate_videoclips
+import gradio as gr
+
+# Function to unzip the downloaded videos
+def unzip_videos(zip_file, extract_dir):
+    with zipfile.ZipFile(zip_file.name, 'r') as zip_ref:
+        zip_ref.extractall(extract_dir)
+    return extract_dir
+
+# Function to concatenate videos based on the CSV file
+def concatenate_videos_from_csv(video_dir, csv_file, audio_file, music_file):
+    clips = []
+    with open(csv_file.name, 'r', encoding='utf-8') as file:
+        reader = csv.DictReader(file)
+        for row in reader:
+            word = row['word']
+            video_path = os.path.join(video_dir, f"{word}.mp4")
+            if os.path.exists(video_path):
+                clip = mp.VideoFileClip(video_path)
+                clips.append(clip)
+
+    if not clips:
+        raise ValueError("No videos were found that match the CSV file.")
+
+    final_video = concatenate_videoclips(clips)
+
+    # Add music if provided
+    if music_file:
+        music = mp.AudioFileClip(music_file.name)
+        final_video = final_video.set_audio(music)
+
+    # Add narration audio if provided
+    if audio_file:
+        narration = mp.AudioFileClip(audio_file.name)
+        final_video = final_video.set_audio(narration)
+
+    output_file = "final_output_video.mp4"
+    final_video.write_videofile(output_file)
+
+    return output_file
+
+# Gradio Interface
+def gradio_interface(zip_file, csv_file, audio_file, music_file):
+    extract_dir = "extracted_videos"
+    os.makedirs(extract_dir, exist_ok=True)
+
+    video_dir = unzip_videos(zip_file, extract_dir)
+    output_video = concatenate_videos_from_csv(video_dir, csv_file, audio_file, music_file)
+
+    return output_video
+
+# Create Gradio UI
+with gr.Blocks() as demo:
+    zip_file = gr.File(label="Upload Video ZIP File")
+    csv_file = gr.File(label="Upload CSV File")
+    audio_file = gr.File(label="Upload Narration Audio File (Optional)", optional=True)
+    music_file = gr.File(label="Upload Background Music File (Optional)", optional=True)
+    download_button = gr.Button("Concatenate Videos and Add Audio")
+    output_video = gr.File(label="Final Video File", interactive=False)
+
+    download_button.click(gradio_interface, [zip_file, csv_file, audio_file, music_file], output_video)
+
+# Launch the Gradio interface
+demo.launch()