import streamlit as st
import yt_dlp
import os
import tempfile

# Set the title of the app
st.title("YouTube Video Downloader")

# Input field for YouTube video URL
video_url = st.text_input("Enter YouTube Video URL:")

# Function to download video
def download_video(url):
    # Create a temporary directory to store the downloaded video
    with tempfile.TemporaryDirectory() as tmpdirname:
        # Define download options
        ydl_opts = {
            'format': 'bestvideo+bestaudio/best',
            'outtmpl': os.path.join(tmpdirname, '%(title)s.%(ext)s'),
            'merge_output_format': 'webm',  # Download format as .webm
        }

        # Download the video using yt-dlp
        with yt_dlp.YoutubeDL(ydl_opts) as ydl:
            # Use a progress hook to update the progress bar
            def progress_hook(d):
                if d['status'] == 'downloading':
                    st.progress(d['downloaded_bytes'] / d['total_bytes'])
                elif d['status'] == 'finished':
                    st.success("Download completed!")

            ydl.params['progress_hooks'] = [progress_hook]
            ydl.download([url])
        
        # Return the path of the downloaded video
        return os.path.join(tmpdirname, f"{ydl.prepare_filename(ydl.extract_info(url))}")

# Download button
if st.button("Download Video"):
    if video_url:
        # Display progress bar
        st.write("Downloading video...")
        downloaded_file_path = download_video(video_url)

        # Provide a download link
        st.write("Download completed! Click below to download your video:")
        st.markdown(f"[Download Video]({downloaded_file_path})", unsafe_allow_html=True)
    else:
        st.error("Please enter a valid YouTube URL.")