Update src/pyscripts/run.py

src/pyscripts/run.py

@@ -2,6 +2,7 @@ from fastapi import FastAPI, Path, Query, Response, BackgroundTasks
 from fastapi.responses import FileResponse, HTMLResponse
 import gradio as gr
 import sys
+import re
 import os
 project_root = os.path.abspath(os.path.join(os.path.dirname(__file__), '..', '..'))
 sys.path.append(project_root)
@@ -164,9 +165,38 @@ def getGuestComposePoster(item:ConferenceInfo):
         return {"url": file_url}
 
 
+class GuestEScreenInfo(BaseModel):
+    room_types: list
+    person: object
+
+
+
+@app.post("/getGuestEScreen")
+def getGuestEScreen(item: GuestEScreenInfo):
+    file_dir = "GuestEScreen"
+    file_urls = []
+    for room_type in item.room_types:
+        rt = ""
+        if room_type.startswith("M"):
+            rt = "M"
+        elif re.match(r"^\d+", room_type):
+            rt = "half"
+        else:
+            rt = "full"
+        img_stream = img_process.getGuestEScreen(rt, item.person)
+        object_name = file_dir + "/" + item.name + ".jpg"
+        if os.path.exists("src/assets/output"):
+            with open(f"src/assets/output/{item.name + '.jpg'}", 'wb') as f:
+                f.write(img_stream.getvalue())
+        else:
+            oss.upload_file_stream_to_s3(img_stream, "host", object_name)
+            file_url = f"https://pub-429c75a96a8f4597984dd7ebc525d652.r2.dev/{object_name}"
+            file_urls.append(file_url)
+    return {"url": "\n".join(file_urls)}
+
 
 interface = gr.Interface(lambda x: "Hello, " + x + "!", "textbox", "textbox")
-app = gr.mount_gradio_app(app, interface, path=CUSTOM_PATH)
+app = gr.getGuestEScreen(app, interface, path=CUSTOM_PATH)
 
 
 if __name__ == "__main__":