from fastapi import FastAPI
from fastapi.responses import FileResponse
import os
import uvicorn

app = FastAPI()

image_directory='./images'
if not os.path.exists(image_directory):
    os.makedirs(image_directory)

@app.get("/images/{image_name}")
async def get_image(image_name: str):
    image_path = os.path.join(image_directory, image_name)
    
    if os.path.exists(image_path):
        return FileResponse(image_path)
    else:
        return {"error": "Image not found"}

if __name__ == "__main__":
    uvicorn.run(app, host="127.0.0.1", port=5000)