PodMagic / app.py
Ribot's picture
Create app.py
11fd592 verified
raw
history blame
3.01 kB
import os
import re
import sys
import subprocess
import importlib.util
import shutil
import tempfile
# === INSTALLATION AUTOMATIQUE DES DÉPENDANCES ===
def install_if_missing(package_name, import_name=None):
import_name = import_name or package_name
if importlib.util.find_spec(import_name) is None:
print(f"Installation de {package_name}...")
subprocess.check_call([sys.executable, "-m", "pip", "install", package_name])
for package in [
("requests",),
("bs4", "bs4"),
("gradio",),
]:
install_if_missing(*package)
# === IMPORTS ===
import requests
import gradio as gr
from bs4 import BeautifulSoup
# === UTILITAIRES ===
def slugify(text, max_length=50):
text = text.lower()
text = re.sub(r'[^\w\s-]', '', text)
text = re.sub(r'[-\s]+', '_', text)
return text[:max_length].strip('_')
def extract_mp3_links_and_title(url):
response = requests.get(url)
response.raise_for_status()
soup = BeautifulSoup(response.text, 'html.parser')
# Titre du podcast
title_tag = soup.find('h1') or soup.find('title')
podcast_title = slugify(title_tag.get_text()) if title_tag else "podcast"
# Liens MP3
mp3_links = [a['href'] for a in soup.find_all('a', href=True) if a['href'].endswith('.mp3')]
return podcast_title, mp3_links
def download_and_zip_podcast(url):
try:
podcast_title, mp3_links = extract_mp3_links_and_title(url)
if not mp3_links:
return "Aucun fichier MP3 trouvé.", None
temp_dir = tempfile.mkdtemp()
for i, mp3_url in enumerate(mp3_links, start=1):
filename = f"{podcast_title}_{i:02}.mp3"
filepath = os.path.join(temp_dir, filename)
with requests.get(mp3_url, stream=True) as r:
r.raise_for_status()
with open(filepath, 'wb') as f:
for chunk in r.iter_content(chunk_size=8192):
f.write(chunk)
zip_path = os.path.join(temp_dir, f"{podcast_title}.zip")
shutil.make_archive(zip_path.replace('.zip', ''), 'zip', temp_dir)
return f"{len(mp3_links)} fichiers téléchargés avec succès.", zip_path
except Exception as e:
return f"Erreur : {str(e)}", None
# === INTERFACE GRADIO ===
with gr.Blocks() as app:
gr.Markdown("# Téléchargeur de Podcasts MP3")
with gr.Row():
url_input = gr.Textbox(label="URL de la page série", placeholder="https://www.radiofrance.fr/...")
download_button = gr.Button("Télécharger et compresser")
output_text = gr.Textbox(label="Message")
file_output = gr.File(label="Fichier ZIP", file_types=[".zip"])
def process(url):
message, zip_file = download_and_zip_podcast(url)
return message, zip_file
download_button.click(fn=process, inputs=[url_input], outputs=[output_text, file_output])
# === LANCEMENT LOCAL OU SUR HUGGINGFACE ===
if __name__ == "__main__":
app.launch(share=True) # `share=True` utile pour Hugging Face