Update app.py

app.py

@@ -1,40 +1,42 @@
 import gradio as gr
-import yt_dlp as youtube_dl
+import yt_dlp
 import os
-import re
-
-def sanitize_filename(filename):
-    """Remove or replace characters that are unsafe for filenames."""
-    unsafe_chars = r'[<>:"/\\|?*\u0000-\u001F]'
-    return re.sub(unsafe_chars, '_', filename)
+import io
 
 def download_youtube_video(youtube_url):
-    """Downloads a YouTube video using yt_dlp and returns the downloaded file path."""
+    """Downloads a YouTube video using yt_dlp and returns the video data and filename."""
     try:
         ydl_opts = {
             'format': 'bestvideo[ext=mp4]+bestaudio[ext=m4a]/best[ext=mp4]/best',
-            'outtmpl': '%(title)s.%(ext)s',
-            'postprocessors': [{
-                'key': 'FFmpegVideoConvertor',
-                'preferedformat': 'mp4',
-            }],
+            'outtmpl': 'video.%(ext)s',
         }
 
-        with youtube_dl.YoutubeDL(ydl_opts) as ydl:
-            result = ydl.extract_info(youtube_url, download=True)
-            video_title = sanitize_filename(result.get('title', 'video'))
-            video_ext = 'mp4'  # We're forcing mp4 format
-            downloaded_file = f"{video_title}.{video_ext}"
-            return downloaded_file
+        with yt_dlp.YoutubeDL(ydl_opts) as ydl:
+            info = ydl.extract_info(youtube_url, download=False)
+            video_title = info.get('title', 'video')
+            video_ext = info.get('ext', 'mp4')
+            
+            # Use a simple filename
+            filename = f"youtube_video.{video_ext}"
+            
+            # Download to memory
+            ydl_opts['outtmpl'] = '-'
+            ydl_opts['logtostderr'] = True
+
+            with io.BytesIO() as video_buffer:
+                ydl.download([youtube_url])
+                video_data = video_buffer.getvalue()
+
+            return video_data, filename, video_title
 
     except Exception as e:
         print("An error occurred:", e)
-        return None
+        return None, None, None
 
 def app(video_link):
-    video_path = download_youtube_video(video_link)
-    if video_path:
-        return video_path, "Download successful! Click the file name to download."
+    video_data, filename, video_title = download_youtube_video(video_link)
+    if video_data and filename:
+        return (filename, video_data), f"Download successful! Video: {video_title}"
     else:
         return None, "Download failed. Please check the URL and try again."