|
|
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)
|
|
|
|
|
|
|
