app.py

import streamlit as st
from moviepy.editor import *
from pytube import YouTube
import os
from collections import defaultdict


# output folder
output_folder = "output"

# create the output folder if it does not exist
if not os.path.exists(output_folder):
    os.makedirs(output_folder)


def progress_callback(stream, chunk, bytes_remaining):
    size = stream.filesize
    progress = (size - bytes_remaining) / size * 100
    print(f"{progress:.2f}% downloaded")


resolutions = ["2160p", "1440p", "1080p", "720p", "480p"]

st.title("YouTube Downloader")
mp4_url = st.text_input("Enter MP4 URL:")
if "youtube.com" in mp4_url or "youtu.be" in mp4_url:
    try:
        # Create a YouTube object using the video URL
        yt = YouTube(mp4_url, on_progress_callback=progress_callback, use_oauth=True)

        # Display the video details to the user
        st.markdown(f"## {yt.title} - {yt.author}")

        # show the picture of the video
        st.image(yt.thumbnail_url)

        yt_streams = yt.streams

        # pre-select the resolution in yt_streams: 144p, 240p, 360p, 480p, 720p, 1080p, 1440p, 2160p
        avialable_stream = defaultdict(str)
        for res in resolutions:
            yt_stream = yt_streams.filter(res=res, type="video").first()
            if yt_stream is not None:
                print(f"Found {res} stream")
                print(yt_stream.is_progressive)
                avialable_stream[res] = yt_stream

        # show a list of available streams
        st.write("Available streams:")
        for res in avialable_stream:
            st.write(f"{res}: {avialable_stream[res]}")

        # dropdown to select the resolution
        resolution = st.selectbox("Select resolution", resolutions)

        # Add a download button for the video in MP3 format
        if st.button("Download"):
            try:
                with st.spinner("Downloading..."):
                    # get the video stream with certain resolution
                    video_stream = yt.streams.filter(res="1080p").first()
                    # if the video stream is not available, get the highest resolution video stream
                    if video_stream is None:
                        video_stream = yt.streams.get_highest_resolution()
                    # download the video
                    video_stream.download(filename="Video.mp4")

                    # get the audio stream
                    audio_stream = yt.streams.get_audio_only()
                    # download the audio
                    audio_stream.download(filename="Audio.mp3")

                    # Merge the video and audio files into a single video file
                    video = VideoFileClip("Video.mp4")
                    audio = AudioFileClip("Audio.mp3")
                    # set the audio of the video as the audio file
                    merge_video = video.set_audio(audio)
                    # specify the output video format codec as H.264 and audio codec as AAC

                    merge_video.write_videofile(
                        f"{output_folder}/{yt.title}.mp4",
                        codec="libx264",
                        audio_codec="aac",
                        fps=30,
                    )

                    # Display a success message to the user
                st.success("Download complete!")
            except:
                st.error("Download failed.")
            with open(f"{output_folder}/{yt.title}.mp4", "rb") as f:
                byte_content = f.read()
                st.download_button(
                    label="Download mp4",
                    data=byte_content,
                    file_name=f"{yt.title}.mp4",
                    mime="audio/mpeg",
                )
    except:
        # Display an error message if the video cannot be downloaded or converted
        st.error("Download failed!")