INVIDEO_BASIC / app.py
gnosticdev's picture
Update app.py
1672ed1 verified
raw
history blame
3.06 kB
import os
import subprocess
import gradio as gr
from moviepy.editor import *
import requests
from datetime import datetime
# 1. Configuración inicial
PEXELS_API_KEY = os.getenv("PEXELS_API_KEY") # Configura esto en Hugging Face
# 2. Lista de voces CORRECTA (formato válido para edge-tts)
VOICES = [
"es-MX-DaliaNeural", # Mujer México
"es-ES-ElviraNeural", # Mujer España
"es-AR-ElenaNeural", # Mujer Argentina
"es-CO-SalomeNeural", # Mujer Colombia
"en-US-JennyNeural", # Mujer EE.UU. (inglés)
"fr-FR-DeniseNeural", # Mujer Francia (francés)
"de-DE-KatjaNeural", # Mujer Alemania (alemán)
"pt-BR-FranciscaNeural" # Mujer Brasil (portugués)
# ¡Añade más si necesitas!
]
# 3. Función principal (a prueba de errores)
def generar_video(prompt, voz_seleccionada, musica=None):
try:
# A. Generar voz
voz_archivo = "voz.mp3"
subprocess.run([
'edge-tts',
'--voice', voz_seleccionada,
'--text', prompt,
'--write-media', voz_archivo
], check=True)
# B. Buscar videos en Pexels
headers = {"Authorization": PEXELS_API_KEY}
response = requests.get(
f"https://api.pexels.com/videos/search?query={prompt[:50]}&per_page=2",
headers=headers,
timeout=15
)
videos = response.json().get("videos", [])
# C. Procesar música
audio = AudioFileClip(voz_archivo)
if musica:
musica_clip = AudioFileClip(musica.name)
if musica_clip.duration < audio.duration:
musica_clip = musica_clip.loop(duration=audio.duration)
audio = CompositeAudioClip([audio, musica_clip.volumex(0.3)])
# D. Crear video
clips = [VideoFileClip(v["video_files"][0]["link"]).subclip(0, 5) for v in videos[:2]]
final_clip = concatenate_videoclips(clips).set_audio(audio)
output_path = f"video_{datetime.now().strftime('%Y%m%d_%H%M%S')}.mp4"
final_clip.write_videofile(output_path, codec="libx264", audio_codec="aac", threads=2)
return output_path
except Exception as e:
print(f"ERROR: {str(e)}")
return None
# 4. Interfaz Gradio (simple pero efectiva)
with gr.Blocks() as app:
gr.Markdown("# 🎬 GENERADOR DE VIDEOS (ESTABLE)")
with gr.Row():
with gr.Column():
prompt = gr.Textbox(label="Tema del video", placeholder="Ej: 'Lugares misteriosos de España'")
voz = gr.Dropdown(label="Selecciona voz", choices=VOICES, value="es-ES-ElviraNeural")
musica = gr.File(label="Música de fondo (opcional)", file_types=[".mp3"])
btn = gr.Button("Generar Video", variant="primary")
with gr.Column():
output = gr.Video(label="Resultado", autoplay=True)
btn.click(
fn=generar_video,
inputs=[prompt, voz, musica],
outputs=output
)
app.launch(server_name="0.0.0.0", server_port=7860)