webm2mp3 / app.py
mobenta's picture
Update app.py
2401333 verified
import gradio as gr
import tempfile
import subprocess
import os
import datetime
def convert_webm_to_mp3(webm_file, output_filename=None):
# Get current date
current_date = datetime.datetime.now().strftime("%Y-%m-%d")
if not output_filename:
output_filename = f"Morning_and_{current_date}.mp3"
else:
# Ensure the filename ends with .mp3
if not output_filename.endswith(".mp3"):
output_filename += ".mp3"
# Create a temporary directory to store the output file
temp_dir = tempfile.mkdtemp()
output_path = os.path.join(temp_dir, output_filename)
# Run ffmpeg command directly for conversion with error handling
ffmpeg_command = [
"ffmpeg", "-y", # Overwrite output file if exists
"-i", webm_file, # Input file
"-vn", # No video
"-acodec", "libmp3lame", # Use the mp3 codec
"-q:a", "2", # Set quality to ensure good compression
output_path # Output file path
]
# Run the ffmpeg command and capture any errors
process = subprocess.run(ffmpeg_command, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
# Check if there was an error during the conversion
if process.returncode != 0:
# Log or print the error for debugging
print("Error in conversion:", process.stderr.decode())
return "Error in conversion. Please check the file format and try again."
# Return the path to the .mp3 file if successful
return output_path
# Gradio interface
iface = gr.Interface(
fn=convert_webm_to_mp3,
inputs=[
gr.File(type="filepath", label="Upload .webm file"),
gr.Textbox(lines=1, placeholder="Enter output filename (optional)", label="Output Filename")
],
outputs=gr.File(label="Converted .mp3 file"),
title="WebM to MP3 Converter",
description="Upload a .webm audio file, and this tool will convert it to .mp3 format. You can set a custom filename for the output."
)
# Launch the interface
if __name__ == "__main__":
iface.launch()