url fixced

app.py

@@ -11,6 +11,22 @@ import httpx
 BASE_URL = os.getenv("API_BASE_URL", "http://127.0.0.1:7860")
 
 
+def _public_base_url(request: gr.Request | None) -> str:
+    if request is None:
+        return os.getenv("PUBLIC_BASE_URL", "").rstrip("/")
+    headers = {k.lower(): v for k, v in dict(request.headers).items()}
+    origin = headers.get("origin")
+    if origin:
+        return origin.rstrip("/")
+    proto = headers.get("x-forwarded-proto")
+    host = headers.get("x-forwarded-host") or headers.get("host")
+    if proto and host:
+        return f"{proto}://{host}".rstrip("/")
+    if host:
+        return f"https://{host}".rstrip("/")
+    return os.getenv("PUBLIC_BASE_URL", "").rstrip("/")
+
+
 def create_gradio_app():
     """Create and return the Gradio Blocks app for Music Memories."""
 
@@ -160,15 +176,15 @@ def create_gradio_app():
         except Exception as e:
             return "Error", {"error": str(e)}
 
-    def song_stream_url_fn(song_id):
+    def song_stream_url_fn(song_id, request: gr.Request | None = None):
         try:
             song_id_i = _maybe_int(song_id)
             if song_id_i is None:
                 raise ValueError("song_id is required")
-            location = _redirect_location(f"/songs/{song_id_i}/stream")
-            if not location:
-                return "No redirect returned", ""
-            return "Success!", location
+            public_base = _public_base_url(request)
+            if public_base:
+                return "Success!", f"{public_base}/songs/{song_id_i}/stream"
+            return "Success!", f"/songs/{song_id_i}/stream"
         except Exception as e:
             return "Error", str(e)
 

main.py

@@ -1,5 +1,5 @@
 from contextlib import asynccontextmanager
-from fastapi import FastAPI, Query, HTTPException, UploadFile, File, Form
+from fastapi import FastAPI, Query, HTTPException, UploadFile, File, Form, Request
 from fastapi.responses import JSONResponse, RedirectResponse, StreamingResponse
 import gradio as gr
 from app import gradio_app
@@ -201,6 +201,7 @@ async def get_song(song_id: int):
 
 @app.post("/songs")
 async def create_song(
+    request: Request,
     title: str = Form(...),
     artist: str = Form(...),
     album: str = Form(None),
@@ -241,10 +242,14 @@ async def create_song(
     
     response = {"status": "success", "song": song}
     if minio_info:
+        public_stream_url = f"{request.base_url}songs/{song['id']}/stream"
+        public_download_url = f"{request.base_url}songs/{song['id']}/download"
         response["audio"] = {
             "uploaded": True,
             "size": minio_info["size"],
-            "stream_url": minio_info["presigned_url"],
+            "stream_url": public_stream_url,
+            "download_url": public_download_url,
+            "minio_presigned_url": minio_info["presigned_url"],
         }
     return response
 
@@ -271,14 +276,23 @@ async def delete_song_endpoint(song_id: int):
 
 @app.get("/songs/{song_id}/stream")
 async def stream_song(song_id: int):
-    """Stream an MP3 file from MinIO (redirects to presigned URL)."""
-    stream_info = stream_mp3_file(song_id)
-    if not stream_info:
-        # Fallback: try to get from local storage
+    """Stream an MP3 file through the FastAPI app domain.
+
+    NOTE: Redirecting to a MinIO presigned URL may point at an internal hostname (e.g. 127.0.0.1)
+    in hosted environments like Hugging Face Spaces, which browsers cannot reach.
+    """
+    file_data = download_mp3_file(song_id)
+    if not file_data:
         raise HTTPException(status_code=404, detail="Audio file not found")
-    
-    # Redirect to MinIO presigned URL for direct streaming
-    return RedirectResponse(url=stream_info["presigned_url"])
+
+    song = get_song_by_id(song_id)
+    filename = f"{song.get('title', 'song')}_{song_id}.mp3" if song else f"song_{song_id}.mp3"
+
+    return StreamingResponse(
+        io.BytesIO(file_data),
+        media_type="audio/mpeg",
+        headers={"Content-Disposition": f"inline; filename={filename}"},
+    )
 
 
 @app.get("/songs/{song_id}/download")
@@ -299,7 +313,7 @@ async def download_song(song_id: int):
 
 
 @app.post("/songs/{song_id}/upload-audio")
-async def upload_audio(song_id: int, audio_file: UploadFile = File(...)):
+async def upload_audio(request: Request, song_id: int, audio_file: UploadFile = File(...)):
     """Upload/update MP3 file for an existing song."""
     # Verify song exists
     song = get_song_by_id(song_id)
@@ -336,7 +350,9 @@ async def upload_audio(song_id: int, audio_file: UploadFile = File(...)):
         "audio": {
             "uploaded": True,
             "size": minio_info["size"],
-            "stream_url": minio_info["presigned_url"],
+            "stream_url": f"{request.base_url}songs/{song_id}/stream",
+            "download_url": f"{request.base_url}songs/{song_id}/download",
+            "minio_presigned_url": minio_info["presigned_url"],
         },
     }