Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,5 +1,6 @@
 import streamlit as st
 import yt_dlp
+import tempfile
 import os
 
 st.title("📥 Multi-Platform Video Downloader")
@@ -13,28 +14,32 @@ if st.button("Download"):
     else:
         with st.spinner("Downloading video..."):
             try:
-                # Setup output directory
-                os.makedirs("downloads", exist_ok=True)
-
-                # yt-dlp options with spoofed User-Agent
-                ydl_opts = {
-                    'outtmpl': 'downloads/%(title)s.%(ext)s',
-                    'format': 'bestvideo+bestaudio/best',
-                    'merge_output_format': 'mp4',
-                    'noplaylist': True,
-                    'quiet': True,
-                    'http_headers': {
-                        'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64)'
+                # Use a temporary directory
+                with tempfile.TemporaryDirectory() as tmp_dir:
+                    ydl_opts = {
+                        'outtmpl': os.path.join(tmp_dir, '%(title)s.%(ext)s'),
+                        'format': 'bestvideo+bestaudio/best',
+                        'merge_output_format': 'mp4',
+                        'noplaylist': True,
+                        'quiet': True,
+                        'http_headers': {
+                            'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64)'
+                        }
                     }
-                }
 
-                with yt_dlp.YoutubeDL(ydl_opts) as ydl:
-                    info = ydl.extract_info(url, download=True)
-                    filename = ydl.prepare_filename(info)
+                    with yt_dlp.YoutubeDL(ydl_opts) as ydl:
+                        info = ydl.extract_info(url, download=True)
+                        filename = ydl.prepare_filename(info)
+
+                    st.success("✅ Download completed!")
 
-                st.success("✅ Download completed!")
-                with open(filename, 'rb') as f:
-                    st.download_button(label="📥 Click to Download", data=f, file_name=os.path.basename(filename), mime="video/mp4")
+                    with open(filename, 'rb') as f:
+                        st.download_button(
+                            label="📥 Click to Download",
+                            data=f,
+                            file_name=os.path.basename(filename),
+                            mime="video/mp4"
+                        )
 
             except yt_dlp.utils.DownloadError as e:
                 error_msg = str(e)
@@ -42,4 +47,3 @@ if st.button("Download"):
                     st.error("⚠️ This video requires login. Only public videos can be downloaded.")
                 else:
                     st.error(f"❌ Download failed: {error_msg}")
-