Update app.py

app.py

@@ -1,33 +1,33 @@
 import streamlit as st
-import yt_dlp
+import pafy
+import os
 
 # Function to download video or audio
 def download_media(url, media_type, resolution=None):
-    ydl_opts = {
-        'format': 'bestvideo+bestaudio/best' if media_type == 'video' else 'bestaudio',
-        'postprocessors': [{'key': 'FFmpegExtractAudio', 'preferredcodec': 'mp3', 'preferredquality': '192'}] if media_type == 'audio' else [],
-    }
-
-    if resolution and media_type == 'video':
-        ydl_opts['format'] = f'bestvideo[height<={resolution}]+bestaudio/best[height<={resolution}]'
-
-    with yt_dlp.YoutubeDL(ydl_opts) as ydl:
-        result = ydl.download([url])
+    video = pafy.new(url)
     
-    return result
+    if media_type == 'video':
+        stream = video.getbest(preftype="mp4") if resolution is None else video.getbest(preftype="mp4", ftypestrict=False, ftype='mp4', res=resolution)
+    elif media_type == 'audio':
+        stream = video.getbestaudio(preftype="m4a")
+
+    download_path = os.path.join("downloads", stream.filename)
+    stream.download(filepath=download_path)
+    return download_path
 
 # Streamlit UI
 st.title("All in One YouTube Downloader")
 url = st.text_input("Enter YouTube URL:")
 media_type = st.selectbox("Select type:", ("video", "audio"))
-resolution = st.selectbox("Select resolution (for video):", ("144", "240", "360", "480", "720", "1080"), index=4)
+resolution = st.selectbox("Select resolution (for video):", ("360p", "480p", "720p", "1080p"), index=2)
 
 if st.button("Download"):
     if url:
         with st.spinner('Downloading...'):
             try:
-                download_media(url, media_type, resolution if media_type == 'video' else None)
+                file_path = download_media(url, media_type, resolution if media_type == 'video' else None)
                 st.success("Download completed!")
+                st.write(f"File saved as: {file_path}")
             except Exception as e:
                 st.error(f"Error: {e}")
     else: