Update src/pyscripts/run.py

src/pyscripts/run.py

@@ -136,6 +136,24 @@ def getConferencePoster(item: ConferencePosterInfo):
         file_url = f"https://pub-429c75a96a8f4597984dd7ebc525d652.r2.dev/{object_name}"
         return {"url": file_url}
 
+@app.post("/getGuestComposePoster")
+def getGuestComposePoster(item):
+    # item.conference_rundown = [list(i) for i in item.conference_rundown]
+    sorted_conference_guest = sorted(item["guest"], key=lambda x: x['rank'])
+    img_stream = img_process.getGuestComposePoster(
+        item["name"],
+        item["eng_name"],
+        sorted_conference_guest)
+    file_dir = "ConferencePoster"
+    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}"
+        return {"url": file_url}
+
 
 
 interface = gr.Interface(lambda x: "Hello, " + x + "!", "textbox", "textbox")
@@ -143,9 +161,10 @@ app = gr.mount_gradio_app(app, interface, path=CUSTOM_PATH)
 
 
 if __name__ == "__main__":
-    MOCK_URL = "https://mock.jsont.run/XFwcOq3q3wgN3gTglSwld"
+    MOCK_URL = "https://mock.jsont.run/iKAdks2DIVBtx-R7Fbw3y"
     res = requests.get(MOCK_URL)
     mock_data = res.json()
+    # mock_data = ConferencePosterInfo(**mock_data)
     print(mock_data)
-    mock_data = ConferencePosterInfo(**mock_data)
-    getConferencePoster(mock_data)
+    # getConferencePoster(mock_data)
+    getGuestComposePoster(mock_data["conference_info"])