import os
import gradio as gr
import uvicorn
from sound_generator import generate_sound, generate_music
from fastapi import FastAPI, HTTPException
from fastapi.middleware.cors import CORSMiddleware
from fastapi.responses import FileResponse, RedirectResponse, HTMLResponse
from pydantic import BaseModel
# Create the FastAPI app with custom docs URL
app = FastAPI(docs_url="/api/docs")
# Configuraci贸n de CORS
app.add_middleware(
CORSMiddleware,
allow_origins=["*"],
allow_credentials=True,
allow_methods=["*"],
allow_headers=["*"],
)
# Define a Pydantic model to handle the input prompt
class AudioRequest(BaseModel):
prompt: str
# Prueba para verificar si la API funciona - la dejamos por ahora para debugging
@app.get("/status")
def status():
return {"status": "API running", "version": "1.0"}
@app.post("/generate-sound/")
async def generate_sound_endpoint(request: AudioRequest):
try:
# Llamada a la funci贸n para generar el sonido
audio_file_path = generate_sound(request.prompt)
# Verifica si el archivo se ha generado correctamente
if not os.path.exists(audio_file_path):
raise HTTPException(
status_code=404, detail="Archivo de audio no encontrado."
)
# Regresar el archivo generado como una respuesta de descarga
return FileResponse(
audio_file_path, media_type="audio/wav", filename="generated_audio.wav"
)
except Exception as e:
raise HTTPException(status_code=500, detail=str(e))
@app.post("/generate-music/")
async def generate_music_endpoint(request: AudioRequest):
try:
# Call the synchronous generate_music function
audio_file_path = generate_music(request.prompt)
# Verifies if the file has been generated correctly
if not os.path.exists(audio_file_path):
raise HTTPException(
status_code=404, detail="Archivo de audio no encontrado."
)
# Return the generated file as a download response
return FileResponse(
audio_file_path, media_type="audio/wav", filename="generated_audio.wav"
)
except Exception as e:
raise HTTPException(status_code=500, detail=str(e))
@app.get("/", response_class=HTMLResponse)
def home():
"""P谩gina de inicio con informaci贸n de la API y estilo inspirado en Gradio."""
return """
API de Sonidos Generativos
API de Sonidos Generativos
Esta API ofrece tres endpoints:
"""
if __name__ == "__main__":
uvicorn.run(app, host="0.0.0.0", port=7860)