Hugging Face's logo

Spaces:
Curinha
/
ContentCreatorsLab
Running

App Files Community
ContentCreatorsLab
File size: 2,953 Bytes 
7814ee2
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
 
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
from pydantic import BaseModel

# Create the FastAPI app
app = FastAPI()


# Configuraci贸n de CORS para permitir solicitudes desde cualquier origen
app.add_middleware(
    CORSMiddleware,
    allow_origins=[
        "*"
    ],  # Permitir todos los or铆genes (puedes restringirlo a tus or铆genes espec铆ficos)
    allow_credentials=True,
    allow_methods=["*"],  # Permitir todos los m茅todos HTTP (GET, POST, OPTIONS, etc.)
    allow_headers=["*"],  # Permitir todos los encabezados
)


# Define a Pydantic model to handle the input prompt
class AudioRequest(BaseModel):
    prompt: str


@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))


# Crear interfaz de Gradio separadamente
with gr.Blocks() as demo:
    gr.Markdown("# API de Sonidos Generativos")
    gr.Markdown("Bienvenido al servicio de generaci贸n de sonidos mediante IA.")
    gr.Markdown(
        "Esta API est谩 disponible para uso a trav茅s de los siguientes endpoints:"
    )
    gr.Markdown("- Generar sonido: `/generate-sound/`")
    gr.Markdown("- Generar m煤sica: `/generate-music/`")
    gr.Markdown("- Documentaci贸n de la API: `/docs`")

# Montar Gradio en FastAPI - IMPORTANTE: esto debe hacerse antes de iniciar el servidor
app = gr.mount_gradio_app(app, demo, path="/")

if __name__ == "__main__":
    uvicorn.run(app, host="0.0.0.0", port=7860)