Update app.py

app.py

@@ -1,68 +1,58 @@
+import gradio as gr
 import os
 import zipfile
-import csv
+import shutil
 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)
+def process_files(audio_file, video_zip, music_file, csv_file):
+    # Unzip the video files
+    video_folder = "videos"
+    with zipfile.ZipFile(video_zip.name, 'r') as zip_ref:
+        zip_ref.extractall(video_folder)
+
+    # Read the CSV file and get the order of videos
+    import csv
+    video_order = []
+    with open(csv_file.name, 'r') as csvfile:
+        reader = csv.DictReader(csvfile)
         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.")
+            video_order.append(os.path.join(video_folder, f"{row['word']}.mp4"))
 
-    final_video = concatenate_videoclips(clips)
+    # Concatenate the videos based on the CSV order
+    video_clips = [mp.VideoFileClip(video) for video in video_order]
+    final_video = mp.concatenate_videoclips(video_clips, method="compose")
 
-    # Add music if provided
-    if music_file:
+    # Add music to the final video if provided
+    if music_file is not None:
         music = mp.AudioFileClip(music_file.name)
         final_video = final_video.set_audio(music)
 
-    # Add narration audio if provided
-    if audio_file:
+    # Add narration audio to the final video if provided
+    if audio_file is not None:
         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)
+    # Save the final video
+    output_video_path = "final_video.mp4"
+    final_video.write_videofile(output_video_path)
 
-    return output_video
+    # Cleanup extracted video files
+    shutil.rmtree(video_folder)
 
-# 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)
+    return output_video_path
 
-    download_button.click(gradio_interface, [zip_file, csv_file, audio_file, music_file], output_video)
+# Define the Gradio interface
+interface = gr.Interface(
+    fn=process_files,
+    inputs=[
+        gr.File(label="Upload Narration Audio File (Optional)"),  # Updated
+        gr.File(label="Upload Video Zip File"),
+        gr.File(label="Upload Music File"),
+        gr.File(label="Upload CSV File")
+    ],
+    outputs=gr.File(label="Download Final Video"),
+    title="Video Concatenation Tool"
+)
 
 # Launch the Gradio interface
-demo.launch()
+interface.launch()