From 396c2420de2303c2ae661c66a7ab39ca4d201d4e Mon Sep 17 00:00:00 2001
From: Jigar vyas <80967728+jigarvyasidea@users.noreply.github.com>
Date: Mon, 10 Oct 2022 07:08:20 -0700
Subject: [PATCH] Add files via upload

---
 main.py | 37 +++++++++++++++++++++++++++++++++++++
 1 file changed, 37 insertions(+)
 create mode 100644 main.py

diff --git a/main.py b/main.py
new file mode 100644
index 00000000..5c31172f
--- /dev/null
+++ b/main.py
@@ -0,0 +1,37 @@
+from fastapi import FastAPI, Request
+from fastapi.responses import FileResponse, HTMLResponse
+from fastapi.staticfiles import StaticFiles
+from fastapi.templating import Jinja2Templates
+from imly_ai.db import get_img_url
+from imly_ai.imagine import get_images
+
+
+app = FastAPI()
+
+app.mount("/static", StaticFiles(directory="static"), name="static")
+
+
+templates = Jinja2Templates(directory="templates")
+
+
+
+@app.get("/", response_class=HTMLResponse)
+async def root(request: Request):
+    return templates.TemplateResponse(
+        "index.html", {"request": request}
+    )
+
+
+@app.post("/imagine")
+async def generate_images(query: str):
+    return {
+        "images": get_images(query)
+    }
+    
+    
+@app.get("/images/{img_id}")
+async def get_images(img_id: str):
+    if img:=get_img_url(img_id):
+        return {"url": FileResponse(img)}
+    return {"url": None}
+