fastapi

app.py

@@ -1,21 +1,34 @@
-import gradio as gr
-import os
+# main.py
+
+from fastapi import FastAPI
 from PIL import Image
+import base64
+
+app = FastAPI()
+
 
+@app.get("/")
+async def root():
+    images = []
 
-def image_mod(image_path):
-    image = Image.open(image_path)
-    return image.rotate(45)
+    with open('image1.jpg', 'rb') as open_file:
+        byte_content = open_file.read()
+    base64_bytes = base64.b64encode(byte_content)
+    base64_string = base64_bytes.decode('utf-8')
+    images.append(base64_string)
 
+    with open('image2.jpg', 'rb') as open_file:
+        byte_content = open_file.read()
+    base64_bytes = base64.b64encode(byte_content)
+    base64_string = base64_bytes.decode('utf-8')
+    images.append(base64_string)
 
-#demo = gr.Interface(image_mod, gr.Image(type="pil"), "image",
-demo = gr.Interface(image_mod, "text", "image",
-    flagging_options=["blurry", "incorrect", "other"], examples=[
-        os.path.join(os.path.dirname(__file__), "workdir/cheetah.jpg"),
-        os.path.join(os.path.dirname(__file__), "workdir/lion.jpg"),
-        os.path.join(os.path.dirname(__file__), "workdir/logo.png")
-        ])
+    #image_path='lion.jpg'
+    #pilim = Image.open(image_path)
+    #pilimrot = pilim.rotate(45)
+    return {"data": images}
 
-if __name__ == "__main__":
-    demo.launch()
+@app.get("/items/{item_id}")
+async def read_item(item_id):
+    return {"item_id": item_id}