Spaces:
Runtime error
Runtime error
import gradio as gr | |
import os | |
import yt_dlp | |
import markdown | |
def downloader(video_url, audio_format, audio_name=None): | |
# Ensure the directory exists | |
os.makedirs('audios', exist_ok=True) | |
# Use a temporary placeholder for the output file | |
if audio_name: | |
temp_output_path = f"audios/{audio_name}.%(ext)s" | |
else: | |
temp_output_path = f"audios/%(title)s.%(ext)s" | |
ydl_opts = { | |
'format': 'bestaudio/best', | |
'postprocessors': [{ | |
'key': 'FFmpegExtractAudio', | |
'preferredcodec': audio_format, | |
}], | |
'outtmpl': temp_output_path, | |
} | |
with yt_dlp.YoutubeDL(ydl_opts) as ydl: | |
ydl.download([video_url]) | |
# Find the downloaded file and rename it if audio_name is provided | |
if audio_name: | |
temp_file = temp_output_path.replace('%(ext)s', audio_format) | |
final_output_path = f"audios/{audio_name}.{audio_format}" | |
os.rename(temp_file, final_output_path) | |
else: | |
final_output_path = None | |
# Since the file is already named by title, find it by the pattern | |
for file in os.listdir('audios'): | |
if file.endswith(f".{audio_format}"): | |
final_output_path = os.path.join('audios', file) | |
break | |
return final_output_path | |
# Example usage: | |
# downloader('https://www.youtube.com/watch?v=dQw4w9WgXcQ', 'mp3', 'my_audio') | |
# downloader('https://www.youtube.com/watch?v=dQw4w9WgXcQ', 'mp3') | |
# Function to read the markdown file and convert it to HTML | |
def read_markdown_file(file_path): | |
with open(file_path, 'r', encoding='utf-8') as file: | |
md_content = file.read() | |
html_content = markdown.markdown(md_content) | |
return html_content | |
# Path to your markdown file | |
markdown_file_path = 'notread.md' | |
# Read and convert the markdown content | |
html_content = read_markdown_file(markdown_file_path) | |
# Read and convert the markdown content | |
html_content = read_markdown_file(markdown_file_path) | |
logschart = """ | |
### Changelog: | |
#### 2.0 - 2024-05-16 | |
##### Added | |
- **change webui**: change the old webui to 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)` | |
##### Changed | |
- **Output Template Placeholder**: Updated `outtmpl` to use a temporary placeholder for the file extension. | |
- `outtmpl: f"audios/{audio_name}.%(ext)s"` | |
- **File Renaming**: Added logic to rename the temporary output file to the final desired name and extension. | |
- `temp_file = temp_output_path.replace('%(ext)s', audio_format)` | |
- `final_output_path = f"audios/{audio_name}.{audio_format}` | |
- `os.rename(temp_file, final_output_path)` | |
- **Return Correct Path**: The `downloader` function now returns the final output path to ensure Gradio can find and load the audio file correctly. | |
- `return final_output_path` | |
##### 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_name = gr.Textbox(label="Audio name of YouTube audio", info="this is optional") | |
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, audio_name], outputs=output) | |
with gr.Group(): | |
with gr.Row(): | |
gr.Markdown(logschart) | |
with gr.Row(): | |
gr.HTML(html_content) | |
demo.launch() | |