Update run.py

run.py

@@ -1,74 +1,33 @@
 import gradio as gr
+import subprocess
 import os
-import yt_dlp
-import markdown
 
-def downloader(video_url, audio_format):
-    # Ensure the directory exists
-    os.makedirs('audios', exist_ok=True)
-    
-    # Use a temporary placeholder for the output file
-    final_output_path = f"audios/%(title)s.%(ext)s"
-    
+def download_media(url):
+    # Determine if the URL is for audio or video
     ydl_opts = {
-        'format': 'bestaudio/best',
-        'postprocessors': [{
-            'key': 'FFmpegExtractAudio',
-            'preferredcodec': audio_format,
-        }],
-        'outtmpl': final_output_path 
+        'format': 'bestvideo[ext=mp4]+bestaudio[ext=m4a]/best[ext=mp4]/best',  # Prefer mp4 format
+        'outtmpl': 'downloaded_media.%(ext)s',  # Output file name
     }
 
-    with yt_dlp.YoutubeDL(ydl_opts) as ydl:
-        result = ydl.download([video_url])
-    
-    # Replace %(ext)s with the actual format
-    final_output_path = final_output_path.replace('%(ext)s', audio_format)
-    
-    return final_output_path
-
-
-
-logschart = """
-    ### Changelog:
-
-    #### 2.0 - 2024-05-16
-
-    ##### Added
-    - **Change WebUI**: Updated the web interface to a new version.
-    - **Directory Check**: Added a check to ensure the `audios` directory exists before attempting to save files to it.
-      - `os.makedirs('audios', exist_ok=True)`
-
-
-
-    ##### Fixed
-    - **File Not Found Error**: Resolved the issue where the file could not be found due to incorrect output path handling.
-
-    ### Summary of Changes
-
-    1. **Ensured Directory Exists**: Added code to create the `audios` directory if it doesn't exist to prevent file saving errors.
-    2. **Output Path Handling**: Modified the output template to use a placeholder for the file extension, which yt-dlp will replace with the actual extension upon downloading.
-    3. **File Renaming Logic**: After downloading the audio, the code now renames the file to match the desired audio name and format.
-    4. **Correct File Path Return**: The correct file path is now returned from the `downloader` function, ensuring that the Gradio `gr.Audio` component can properly display and play the downloaded audio file.
-
-    These changes collectively ensure that the audio file is downloaded correctly, renamed appropriately, and made accessible to the Gradio interface for playback.
-    """
-
-with gr.Blocks() as demo:
-    gr.Markdown("# YouTube Downloader 2.0")
-    with gr.Row():
-        video_url = gr.Textbox(label="YouTube video link")
-    with gr.Row():
-        audio_format = gr.Radio(["wav", "flac", "mp3"], label="Select the output format", value="wav")
-    with gr.Row():
-        output = gr.Audio(label="Output")
-    with gr.Row():
-        download_button = gr.Button("Download")
-        download_button.click(downloader, inputs=[video_url, audio_format], outputs=[output])
-        
-    with gr.Group():
-        with gr.Row():
-            gr.Markdown(logschart)
-
-
-demo.launch()
+    # Download using yt-dlp
+    subprocess.run(['yt-dlp', url, '--merge-output-format', 'mp4', '--recode-video', 'mp4'])
+
+    # Check if downloaded file exists
+    if os.path.exists('downloaded_media.mp4'):
+        return gr.outputs.Video("downloaded_media.mp4")
+    elif os.path.exists('downloaded_media.m4a'):
+        return gr.outputs.Audio("downloaded_media.m4a")
+    else:
+        return "Media could not be downloaded or unsupported format."
+
+# Create Gradio interface
+iface = gr.Interface(
+    fn=download_media,
+    inputs="text",
+    outputs="auto",
+    title="YouTube Downloader",
+    description="Enter a YouTube video URL to download its audio or video."
+)
+
+# Launch the interface
+iface.launch()