Add application files and dependencies

README.md

@@ -10,4 +10,9 @@ pinned: false
 license: apache-2.0
 ---
 
+
+# YouTube Video Processor
+
+This Streamlit app allows you to process YouTube videos or playlists. It downloads videos, trims them using auto-editor, and re-encodes them with ffmpeg.
+
 Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -0,0 +1,106 @@
+import streamlit as st
+import subprocess
+import os
+import glob
+
+def run_command(command):
+    process = subprocess.Popen(command, stdout=subprocess.PIPE, stderr=subprocess.PIPE, shell=True, universal_newlines=True)
+    output, error = process.communicate()
+    return output, error
+
+def process_video(url):
+    output_dir = "./output_folder"
+    os.makedirs(output_dir, exist_ok=True)
+
+    # Determine if it's a playlist or single video
+    content_type = "playlist" if "list=" in url else "video"
+
+    # Download video(s)
+    format = "bestvideo[height<=1080][ext=mp4]+bestaudio[ext=m4a]/best[height<=1080][ext=mp4]"
+    download_command = f'yt-dlp -f "{format}" -N 64 -o "{output_dir}/%(title)s.%(ext)s" "{url}"'
+    output, error = run_command(download_command)
+    st.text(output)
+    if error:
+        st.error(error)
+        return
+
+    # Process videos
+    if content_type == "video":
+        video_file = glob.glob(f"{output_dir}/*.mp4")[0]
+        trimmed_file = video_file.replace(".mp4", "_trimmed.mp4")
+        auto_editor_command = f'auto-editor "{video_file}" --margin 0.5sec --output "{trimmed_file}"'
+        output, error = run_command(auto_editor_command)
+        st.text(output)
+        if error:
+            st.error(error)
+            return
+
+        final_file = trimmed_file.replace("_trimmed.mp4", "_f.mp4")
+        ffmpeg_command = f'ffmpeg -i "{trimmed_file}" -map 0:v:0 -map 0:a:0 -c:v libx264 -preset ultrafast -crf 23 -maxrate 25M -vf "scale=-1:1080" -c:a aac -b:a 192k -ac 2 -strict -2 "{final_file}"'
+        output, error = run_command(ffmpeg_command)
+        st.text(output)
+        if error:
+            st.error(error)
+            return
+
+        return final_file
+
+    elif content_type == "playlist":
+        trimmed_dir = f"{output_dir}/trimmed"
+        os.makedirs(trimmed_dir, exist_ok=True)
+
+        for video_file in glob.glob(f"{output_dir}/*.mp4"):
+            video_name = os.path.basename(video_file).replace(".mp4", "")
+            trimmed_file = f"{trimmed_dir}/{video_name}_trimmed.mp4"
+            auto_editor_command = f'auto-editor "{video_file}" --margin 0.5sec --output "{trimmed_file}"'
+            output, error = run_command(auto_editor_command)
+            st.text(output)
+            if error:
+                st.error(error)
+                continue
+
+            final_file = f"{trimmed_dir}/{video_name}_f.mp4"
+            ffmpeg_command = f'ffmpeg -i "{trimmed_file}" -map 0:v:0 -map 0:a:0 -c:v libx264 -preset ultrafast -crf 23 -maxrate 25M -vf "scale=-1:1080" -c:a aac -b:a 192k -ac 2 -strict -2 "{final_file}"'
+            output, error = run_command(ffmpeg_command)
+            st.text(output)
+            if error:
+                st.error(error)
+                continue
+
+        return trimmed_dir
+
+def main():
+    st.title("YouTube Video Processor")
+
+    url = st.text_input("Enter the YouTube URL:")
+
+    if st.button("Process Video"):
+        if url:
+            with st.spinner("Processing video..."):
+                result = process_video(url)
+            
+            if result:
+                if os.path.isfile(result):
+                    st.success("Video processing completed!")
+                    st.download_button(
+                        label="Download Processed Video",
+                        data=open(result, 'rb'),
+                        file_name=os.path.basename(result),
+                        mime="video/mp4"
+                    )
+                elif os.path.isdir(result):
+                    st.success("Playlist processing completed!")
+                    for video in glob.glob(f"{result}/*_f.mp4"):
+                        st.download_button(
+                            label=f"Download {os.path.basename(video)}",
+                            data=open(video, 'rb'),
+                            file_name=os.path.basename(video),
+                            mime="video/mp4"
+                        )
+            else:
+                st.error("An error occurred during processing.")
+        else:
+            st.warning("Please enter a YouTube URL.")
+
+if __name__ == "__main__":
+    main()

packages.txt

@@ -0,0 +1 @@
+ffmpeg

requirements.txt

@@ -0,0 +1,3 @@
+streamlit
+yt-dlp
+auto-editor