Enhance FastAPI app with custom documentation URL and add status endpoint

app.py

@@ -5,29 +5,29 @@ 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 fastapi.responses import FileResponse, JSONResponse
 from pydantic import BaseModel
 
-# Create the FastAPI app
-app = FastAPI()
+# Create the FastAPI app with custom docs URL
+app = FastAPI(docs_url="/api/docs")
 
-
-# Configuración de CORS para permitir solicitudes desde cualquier origen
+# Configuración de CORS
 app.add_middleware(
     CORSMiddleware,
-    allow_origins=[
-        "*"
-    ],  # Permitir todos los orígenes (puedes restringirlo a tus orígenes específicos)
+    allow_origins=["*"],
     allow_credentials=True,
-    allow_methods=["*"],  # Permitir todos los métodos HTTP (GET, POST, OPTIONS, etc.)
-    allow_headers=["*"],  # Permitir todos los encabezados
+    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):
@@ -49,7 +49,6 @@ async def generate_sound_endpoint(request: AudioRequest):
     except Exception as e:
         raise HTTPException(status_code=500, detail=str(e))
 
-
 @app.post("/generate-music/")
 async def generate_music_endpoint(request: AudioRequest):
     try:
@@ -70,20 +69,24 @@ async def generate_music_endpoint(request: AudioRequest):
     except Exception as e:
         raise HTTPException(status_code=500, detail=str(e))
 
+# Interfaz de Gradio
+demo = gr.Blocks()
 
-# Crear interfaz de Gradio separadamente
-with gr.Blocks() as demo:
+with 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
+    gr.Markdown("Esta API está disponible para uso a través de los siguientes endpoints:")
+    gr.Markdown("- Generar sonido: `/api/generate-sound/`")
+    gr.Markdown("- Generar música: `/api/generate-music/`")
+    gr.Markdown("- Documentación de la API: `/api/docs`")
+
+# Montar la app de Gradio en FastAPI
 app = gr.mount_gradio_app(app, demo, path="/")
 
+# Para confirmar que el código está funcionando, podemos agregar un print
+print("✅ Servidor inicializado. Gradio montado en la ruta principal '/'")
+print("📝 Endpoints de la API disponibles en '/api/*'")
+print("📄 Documentación disponible en '/api/docs'")
+
 if __name__ == "__main__":
     uvicorn.run(app, host="0.0.0.0", port=7860)