Update main.py

main.py

@@ -18,7 +18,7 @@ from logging.handlers import RotatingFileHandler
 
 from dotenv import load_dotenv
 from fastapi import FastAPI, Request
-from fastapi.responses import JSONResponse
+from fastapi.responses import FileResponse, JSONResponse
 from pyrogram import Client
 from pyrogram.enums import ParseMode
 from pyrogram.errors import FloodWait
@@ -237,7 +237,8 @@ async def handle_media(chat_id: int, reply_to_id: int, file_id: str, filename: s
         os.remove(downloaded)
 
         entry     = fm.save_file(file_data, filename)
-        link      = fm.get_url(entry)
+        # Build direct download link served by our own FastAPI endpoint
+        link = f"{SPACE_URL}/download/{entry['id']}"
         file_size = len(file_data)
         size_str  = (f"{file_size/1024/1024:.2f} MB"
                      if file_size > 1024 * 1024
@@ -273,6 +274,29 @@ async def lifespan(app: FastAPI):
 
 app = FastAPI(lifespan=lifespan)
 
+# ============================================================
+# File serving endpoints
+# ============================================================
+@app.get("/")
+async def root():
+    return {"status": "ok", "service": "Link Generator Bot"}
+
+
+@app.get("/download/{file_id}")
+async def download_file(file_id: str):
+    """Serve a stored file by its UUID."""
+    entry = fm.get_entry_by_id(file_id)
+    if not entry:
+        return JSONResponse({"error": "File not found or expired"}, status_code=404)
+    path = entry["path"]
+    if not os.path.exists(path):
+        return JSONResponse({"error": "File not found on disk"}, status_code=404)
+    return FileResponse(
+        path=path,
+        filename=entry["filename"],
+        media_type="application/octet-stream",
+    )
+
 # ============================================================
 # Webhook endpoint
 # ============================================================