Update src/pyscripts/run.py

src/pyscripts/run.py

@@ -149,7 +149,7 @@ def getConferencePoster(item: ConferencePosterInfo):
 @app.post("/getGuestComposePoster")
 def getGuestComposePoster(item:ConferenceInfo):
     # item.conference_rundown = [list(i) for i in item.conference_rundown]
-    sorted_conference_guest = sorted(item.guest, key=lambda x: x['rank'])
+    sorted_conference_guest = sorted(item.guest, key=lambda x: x['rank'] if 'rank' in x else 0)
     img_stream = img_process.getGuestComposePoster(
         item.name,
         item.eng_name,
@@ -183,7 +183,6 @@ def getGuestEScreen(item: GuestEScreenInfo):
             rt = "half"
         else:
             rt = "full"
-        print(item.person)
         img_stream = img_process.getGuestEScreen(rt, item.person)
         object_name = file_dir + "/" + rt + "_" + item.person["name"] + ".jpg"
         if os.path.exists("src/assets/output"):
@@ -201,13 +200,16 @@ interface = gr.Interface(lambda x: "Hello, " + x + "!", "textbox", "textbox")
 
 
 if __name__ == "__main__":
-    MOCK_URL = "https://mock.jsont.run/tjTCU1IWOGoIunWRovNJt"
+    # MOCK_URL = "https://mock.jsont.run/1H9o2vvkIBMB83oeVQd1q"
+    MOCK_URL = "https://mock.jsont.run/c3oQ2XLm1ZIHoVWZa18Ui"
+    
     res = requests.get(MOCK_URL)
     mock_data = res.json()
-    # mock_data = ConferencePosterInfo(**mock_data)
-    # getConferencePoster(mock_data)
     print(mock_data)
+
+    mock_data = ConferencePosterInfo(**mock_data)
+    getConferencePoster(mock_data)
     # mock_data = ConferenceInfo(**mock_data["conference_info"])
     # getGuestComposePoster(mock_data)
-    mock_data = GuestEScreenInfo(**mock_data)
-    getGuestEScreen(mock_data)
+    # mock_data = GuestEScreenInfo(**mock_data)
+    # getGuestEScreen(mock_data)