diff --git a/main.py b/main.py index 36eee87..cd4b3ac 100644 --- a/main.py +++ b/main.py @@ -1,5 +1,5 @@ import aiohttp -from fastapi import FastAPI, HTTPException +from fastapi import FastAPI, HTTPException, Request from fastapi.responses import FileResponse import asyncmy import asyncio @@ -72,7 +72,7 @@ async def root(): return {"message": "yes the api works, maybe i will create a small landing page later here"} @app.get("/random") -async def get_random_bnuy(): +async def get_random_bnuy(request: Request): async with app.state.pool.acquire() as conn: async with conn.cursor() as cursor: await cursor.execute("SELECT filename, subreddit, url FROM images ORDER BY RAND() LIMIT 1;") @@ -80,7 +80,7 @@ async def get_random_bnuy(): if result: filepath = os.path.join("data/images", result[0]) if os.path.exists(filepath): - return {"file": f"/images/{result[0]}", "source": f"https://www.reddit.com/r/{result[1]}/", "original_url": result[2]} + return {"url": f"{request.base_url}/images/{result[0]}", "source": f"https://www.reddit.com/r/{result[1]}/", "original_url": result[2]} else: raise HTTPException(status_code=404, detail="Image file not found") else: