Update app.py

app.py

@@ -7,14 +7,18 @@ import gradio as gr
 # CONFIG
 # -------------------------
 
-SPACE_NAME = os.getenv("SPACE_ID", "your-username/your-space")
-BASE_URL = f"https://{SPACE_NAME.split('/')[-1]}.hf.space"
-
 FILE_DIR = "files"
 os.makedirs(FILE_DIR, exist_ok=True)
 
+# Your Space URL (auto-detect friendly)
+SPACE_NAME = os.getenv("SPACE_ID")
+if SPACE_NAME:
+    BASE_URL = f"https://{SPACE_NAME.split('/')[-1]}.hf.space"
+else:
+    BASE_URL = "http://127.0.0.1:7860"
+
 # -------------------------
-# FASTAPI SERVER
+# FASTAPI APP
 # -------------------------
 
 server = FastAPI()
@@ -36,25 +40,24 @@ def download_file(filename: str):
 # -------------------------
 
 def generate_link(filename):
-    file_path = os.path.join(FILE_DIR, filename)
-
     if not filename:
         return "Enter filename"
 
+    file_path = os.path.join(FILE_DIR, filename)
+
     if not os.path.exists(file_path):
-        return "File does not exist"
+        return "File does not exist in /files"
 
     return f"{BASE_URL}/download/{filename}"
 
 # -------------------------
-# SIMPLE UI
+# GRADIO UI
 # -------------------------
 
 with gr.Blocks() as app:
+    gr.Markdown("# 🔗 Direct Link Generator (HF Spaces)")
 
-    gr.Markdown("# 🔗 HF Spaces Link Generator")
-
-    filename_input = gr.Textbox(label="Filename (must exist in /files)")
+    filename_input = gr.Textbox(label="Filename")
     link_output = gr.Textbox(label="Direct Link")
 
     btn = gr.Button("Generate Link")
@@ -65,8 +68,9 @@ with gr.Blocks() as app:
         outputs=link_output
     )
 
-# IMPORTANT: mount FastAPI
-app = gr.mount_gradio_app(server, app)
+# 🔥 Correct mounting (IMPORTANT FIX)
+app = gr.mount_gradio_app(server, app, path="/")
 
 # Launch for Spaces
+app.queue()
 app.launch()