qwen url fix

main.py

@@ -3,6 +3,7 @@ from fastapi import FastAPI, Query, HTTPException, UploadFile, File, Form, Reque
 from fastapi.responses import JSONResponse, RedirectResponse, StreamingResponse
 import gradio as gr
 from app import gradio_app
+import os
 from database import (
     init_database,
     # Songs
@@ -213,7 +214,7 @@ async def create_song(
 ):
     """Add a new song with optional MP3 file upload to MinIO."""
     song = add_song(title, artist, album, duration, bpm, energy_level)
-    
+
     # Handle MP3 file upload to MinIO
     minio_info = None
     if audio_file and audio_file.filename:
@@ -225,25 +226,38 @@ async def create_song(
             song_id=song["id"],
             filename=audio_file.filename,
         )
-        
+
         if minio_info:
             # Update song with file reference
             update_song_file(song["id"], minio_info["object_name"], minio_info["size"])
-    
+
     # Add to semantic search index if lyrics provided
     if lyrics:
         add_song_vibe(song["id"], title, artist, lyrics)
-    
+
     # Invalidate songs list cache
     await invalidate_list_cache("all_songs")
-    
+
     # Log activity to MongoDB
     await track_analytics_event("song_created", {"song_id": song["id"], "title": title})
+
+    # Determine the public base URL for the hosted environment
+    # Use environment variable or derive from request headers
+    public_base = os.getenv("PUBLIC_BASE_URL")
+    if not public_base:
+        # Try to derive from request headers (works with proxies like HF Spaces)
+        forwarded_proto = request.headers.get("x-forwarded-proto")
+        forwarded_host = request.headers.get("x-forwarded-host")
+        if forwarded_proto and forwarded_host:
+            public_base = f"{forwarded_proto}://{forwarded_host}"
+        else:
+            # Fallback to request base URL
+            public_base = str(request.base_url).rstrip("/")
     
     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"
+        public_stream_url = f"{public_base}/songs/{song['id']}/stream"
+        public_download_url = f"{public_base}/songs/{song['id']}/download"
         response["audio"] = {
             "uploaded": True,
             "size": minio_info["size"],
@@ -319,39 +333,49 @@ async def upload_audio(request: Request, song_id: int, audio_file: UploadFile =
     song = get_song_by_id(song_id)
     if not song:
         raise HTTPException(status_code=404, detail="Song not found")
-    
+
     # Validate file type
     if not audio_file.filename.lower().endswith('.mp3'):
         raise HTTPException(status_code=400, detail="Only MP3 files are supported")
-    
+
     # Read and upload file
     contents = await audio_file.read()
     file_stream = io.BytesIO(contents)
-    
+
     minio_info = upload_mp3_file(
         file_data=file_stream,
         file_size=len(contents),
         song_id=song_id,
         filename=audio_file.filename,
     )
-    
+
     if not minio_info:
         raise HTTPException(status_code=500, detail="Failed to upload file to MinIO")
-    
+
     # Update song with file reference
     update_song_file(song_id, minio_info["object_name"], minio_info["size"])
-    
+
     # Invalidate cache
     await invalidate_song_cache(song_id)
     await invalidate_list_cache("all_songs")
-    
+
+    # Determine the public base URL for the hosted environment
+    public_base = os.getenv("PUBLIC_BASE_URL")
+    if not public_base:
+        forwarded_proto = request.headers.get("x-forwarded-proto")
+        forwarded_host = request.headers.get("x-forwarded-host")
+        if forwarded_proto and forwarded_host:
+            public_base = f"{forwarded_proto}://{forwarded_host}"
+        else:
+            public_base = str(request.base_url).rstrip("/")
+
     return {
         "status": "success",
         "audio": {
             "uploaded": True,
             "size": minio_info["size"],
-            "stream_url": f"{request.base_url}songs/{song_id}/stream",
-            "download_url": f"{request.base_url}songs/{song_id}/download",
+            "stream_url": f"{public_base}/songs/{song_id}/stream",
+            "download_url": f"{public_base}/songs/{song_id}/download",
             "minio_presigned_url": minio_info["presigned_url"],
         },
     }