refactor: change the render order to one response a time

app.py

@@ -21,6 +21,7 @@ from utils.completion_reward_utils import (
     create_certificate,
     complete_reward,
     check_is_in_completion_reward,
+    get_llm_response_once,
 )
 
 seafoam = Seafoam()
@@ -49,8 +50,12 @@ def get_player_info(player_backend_user_id):
 
         temp_log = """你開始了星空探險隊的冒險！ 你在離家 2 天後，找到了迷失已久的星際繪圖師，並在他的引導下進入了下一個未知的領域。 在完成了 41 關學習內容後，你獲得了遙遠星球上的神秘遺產。 在某天，你在冰晶極地找到了一顆具有神秘力量的寒冰寶石。 你在奇特的花園中找到了一種新的能源。 在雲端星球中，你遇見了一位雲之精靈，並從她那裡獲得了一個神祕的能量水球。 132 天之後，你們獲得了一個神秘的地圖，這將引導你們找到隱藏的神秘能量源。\n你們在一個古老的地下洞穴中找到了一個巨大的藍色能量晶體，這將極大地援助你們的旅程。 你在混亂的黑洞中找到了一片完好無損的石碑，上面記載著光束寶石的由來和使用方式。 你用寶石的力量召喚傳說中的先知，獲得了狐狸貓族的力量, 向黑洞發動了終極一擊。 你在繼續活動了 7 天後，你的行為感動了星際，你獲得了一個可以轉換成任何形象的能量護盾。 你在一場巨大的黑洞爆炸後，成功找回了去往家鄉的方向。 連日交戰後達成了學習 209 個內容的紀錄，獲得了星系長老的讚賞，並贈與你們神秘的光護符。 你成功與神秘的月亮寶石建立連結，從而獲得月亮之力。 你在遙遠的星海之上發現了一顆寂靜的星體，星體的中心藏有一個帶著神秘能量的寶藏。 你在神祕的狐狸貓族祖先墳場找到了一塊古老的碑文，裡面記載了如何使用和提煉能量的方法。 在經歷了若干日月後，你終於在 229 號洞穴找到了第一塊結晶。 你們在探索神秘洞穴的過程中，發現了一個神秘寶箱，裡面有 229 個神秘道具。 你們在冬季島上找到了冰晶葉，並獲得了 678 點的能量。 你剛剛結束了 14 天的劍橋學堂，你們獲得了 726 點的能量供給。 你剛剛結束了 14 天的劍橋學堂，你們獲得了 726 點的能量供給。 你在星空探險隊的冒險期間和狐貍貓一起度過了聖誕佳節 🎄 你從超空間黑洞脫離時發現了一串神秘的密碼，這個密碼包含著黑洞能量的謎題。 你們在一個神秘的星球上探險，在那裡你們發現了新的能量源。 你們在一片未知的領域中尋找到了狐狸貓族的古老遺蹟，並獲得了 780 的能量點數。 你收集了 836 個痕跡之星，並用風暴龍的力量打開了神秘之門。 你收集了 941 個痕跡之星，並用風暴龍的力量打開了神秘之門。 你結束了 2023 與狐貍貓的星空探險！"""
         processed_datas["[email protected]"] = temp_log
-        processed_datas["http://googleid.junyiacademy.org/115084976189396533674"] = temp_log
-        processed_datas["http://googleid.junyiacademy.org/106428943548495187296"] = temp_log
+        processed_datas[
+            "http://googleid.junyiacademy.org/115084976189396533674"
+        ] = temp_log
+        processed_datas[
+            "http://googleid.junyiacademy.org/106428943548495187296"
+        ] = temp_log
 
         with open("processed_adventure_logs.json", "w") as f:
             json.dump(processed_datas, f)
@@ -340,9 +345,11 @@ with gr.Blocks(theme=seafoam, css=get_content("css/style.css")) as demo:
         queue=False,
     )
 
-    player_info_query_btn.click(get_player_info, player_backend_id, player_info, queue=False).then(
-        get_player_logs, player_backend_id, player_logs, queue=False
-    ).then(lambda: gr.update(visible=True), None, start_make_reward, queue=False).then(
+    player_info_query_btn.click(
+        get_player_info, player_backend_id, player_info, queue=False
+    ).then(get_player_logs, player_backend_id, player_logs, queue=False).then(
+        lambda: gr.update(visible=True), None, start_make_reward, queue=False
+    ).then(
         render_player_data,
         player_info,
         [avatar, pet_gallery, badge_gallery, adventure_log, achievements, adventure],
@@ -391,7 +398,11 @@ with gr.Blocks(theme=seafoam, css=get_content("css/style.css")) as demo:
                 gr.update(visible=True),
             )
         else:
-            return gr.update(visible=True), gr.update(visible=True), gr.update(visible=False)
+            return (
+                gr.update(visible=True),
+                gr.update(visible=True),
+                gr.update(visible=False),
+            )
 
     confirm_player_name.click(
         lambda: (gr.update(interactive=False), gr.update(visible=False)),
@@ -408,7 +419,11 @@ with gr.Blocks(theme=seafoam, css=get_content("css/style.css")) as demo:
     )
 
     cancel_player_name.click(
-        lambda: (gr.update(interactive=True), gr.update(visible=True), gr.update(visible=False)),
+        lambda: (
+            gr.update(interactive=True),
+            gr.update(visible=True),
+            gr.update(visible=False),
+        ),
         None,
         [player_name, confirm_player_name, player_name_too_long],
         queue=False,
@@ -456,6 +471,34 @@ with gr.Blocks(theme=seafoam, css=get_content("css/style.css")) as demo:
         queue=False,
     )
 
+    get_first_llm_response_args = dict(
+        fn=get_llm_response_once,
+        inputs=[completion_reward, player_logs],
+        output=bot1,
+        queue=False,
+    )
+
+    get_second_llm_response_args = dict(
+        fn=get_llm_response_once,
+        inputs=[completion_reward, player_logs],
+        output=bot2,
+        queue=False,
+    )
+
+    get_third_llm_response_args = dict(
+        fn=get_llm_response_once,
+        inputs=[completion_reward, player_logs],
+        output=bot3,
+        queue=False,
+    )
+
+    get_fourth_llm_response_args = dict(
+        fn=get_llm_response_once,
+        inputs=[completion_reward, player_logs],
+        output=bot4,
+        queue=False,
+    )
+
     start_generate_story.click(
         lambda: gr.update(visible=False), None, start_generate_story, queue=False
     ).then(
@@ -464,7 +507,13 @@ with gr.Blocks(theme=seafoam, css=get_content("css/style.css")) as demo:
         [bot1, bot2, bot3, bot4, weaving],
         queue=False,
     ).then(
-        **get_llm_response_args
+        **get_first_llm_response_args
+    ).then(
+        **get_second_llm_response_args
+    ).then(
+        **get_third_llm_response_args
+    ).then(
+        **get_fourth_llm_response_args
     ).then(
         lambda: gr.update(visible=True), None, [select_story], queue=False
     ).then(

utils/completion_reward.py

@@ -53,7 +53,14 @@ class CompletionReward:
         self.aws_agent = AWSAgent()
         self.google_agent = GoogleAgent()
         self.mtk_agent = MTKAgent()
+        self.agent_list = [
+            self.openai_agent,
+            self.aws_agent,
+            self.google_agent,
+            self.mtk_agent,
+        ]
         self.shuffled_response_order = {}
+        self.pop_response_order = []
         self.paragraph_map = {
             "openai": self.paragraph_openai,
             "aws": self.paragraph_aws,
@@ -67,10 +74,10 @@ class CompletionReward:
         google_story = self.google_agent.get_story(player_logs)
         mtk_story = self.mtk_agent.get_story(player_logs)
         agents_responses = {
-            'openai': openai_story,
-            'aws': aws_story,
-            'google': google_story,
-            'mtk': mtk_story,
+            "openai": openai_story,
+            "aws": aws_story,
+            "google": google_story,
+            "mtk": mtk_story,
         }
         self.paragraph_openai = agents_responses["openai"]
         self.paragraph_aws = agents_responses["aws"]
@@ -91,6 +98,24 @@ class CompletionReward:
             [(None, shuffled_responses[3])],
         )
 
+    def get_llm_response_once(self, player_logs):
+        if self.agent_list:
+            # Randomly select and remove an agent from the list
+            agent = self.agent_list.pop(random.randint(0, len(self.agent_list) - 1))
+        else:
+            return "No agents left", None
+
+        story = agent.get_story(player_logs)
+        self.agents_responses[agent.name] = story
+        self.pop_response_order.append(agent.name)
+
+        if len(self.pop_response_order) == 4:
+            self.shuffled_response_order = {
+                str(index): agent for index, agent in enumerate(self.pop_response_order)
+            }
+
+        return [(None, story)]
+
     def set_player_name(self, player_name, player_backend_user_id):
         self.player_backend_user_id = player_backend_user_id
         self.player_name = player_name
@@ -206,6 +231,7 @@ class CompletionReward:
 
 class OpenAIAgent:
     def __init__(self):
+        self.name = "openai"
         self.temperature = 0.8
         self.frequency_penalty = 0
         self.presence_penalty = 0
@@ -285,6 +311,9 @@ class OpenAIAgent:
 
 
 class AWSAgent:
+    def __init__(self):
+        self.name = "aws"
+
     def get_story(self, user_log):
         system_prompt = """
             我正在舉辦一個學習型的活動，我為學生設計了一個獨特的故事機制，每天每個學生都會收到屬於自己獨特的冒險紀錄，現在我需要你協助我將這些冒險紀錄，製作成一段冒險故事，請
@@ -341,6 +370,9 @@ class GoogleAgent:
 
     gemini_pro_model = GenerativeModel("gemini-pro")
 
+    def __init__(self):
+        self.name = "google"
+
     def get_story(self, user_log):
         system_prompt = """
                     我正在舉辦一個學習型的活動，我為學生設計了一個獨特的故事機制，每天每個學生都會收到屬於自己獨特的冒險紀錄，現在我需要你協助我將這些冒險紀錄，製作成一段冒險故事，請
@@ -377,6 +409,9 @@ class GoogleAgent:
 
 
 class MTKAgent:
+    def __init__(self):
+        self.name = "mtk"
+
     def get_story(self, user_log):
         system_prompt = """
             我正在舉辦一個學習型的活動，我為學生設計了一個獨特的故事機制，每天每個學生都會收到屬於自己獨特的冒險紀錄，現在我需要你協助我將這些冒險紀錄，製作成一段冒險故事，請
@@ -491,10 +526,11 @@ class ImageProcessor:
         for line in paragraph.split("\n"):
             wrapped_lines = textwrap.wrap(line, width=63)
             for wrapped_line in wrapped_lines:
-                _, _, _, line_height = draw.textbbox((0, 0), wrapped_line, font=body_font)
+                _, _, _, line_height = draw.textbbox(
+                    (0, 0), wrapped_line, font=body_font
+                )
                 paragraph_height += line_height + 10
 
-
         # Draw the box
         padding = 40
         left, right = 50, img.width - 50

utils/completion_reward_utils.py

@@ -7,6 +7,10 @@ def get_llm_response(completion_reward, *args):
     return completion_reward.get_llm_response(*args)
 
 
+def get_llm_response_once(completion_reward, *args):
+    return completion_reward.get_llm_response_once(*args)
+
+
 def set_player_name(completion_reward, *args):
     return completion_reward.set_player_name(*args)
 
@@ -31,9 +35,13 @@ def check_is_in_completion_reward(player_backend_user_id):
     with open("./data/completion_reward_issue_status.json") as f:
         completion_reward_issue_status_dict = json.load(f)
         completion_reward_issue_status_dict["[email protected]"] = "not_issued"
-        completion_reward_issue_status_dict["http://googleid.junyiacademy.org/115084976189396533674"] = "not_issued"
-        completion_reward_issue_status_dict["http://googleid.junyiacademy.org/106428943548495187296"] = "not_issued"
-        
+        completion_reward_issue_status_dict[
+            "http://googleid.junyiacademy.org/115084976189396533674"
+        ] = "not_issued"
+        completion_reward_issue_status_dict[
+            "http://googleid.junyiacademy.org/106428943548495187296"
+        ] = "not_issued"
+
     preview_list = [
         "[email protected]",
         "[email protected]",
@@ -44,9 +52,10 @@ def check_is_in_completion_reward(player_backend_user_id):
         "http://id.junyiacademy.org/28f0f5bb784441c58ad4bafad00a2c3d",
     ]
 
-
-    if player_backend_user_id in completion_reward_issue_status_dict or player_backend_user_id in preview_list:
-        
+    if (
+        player_backend_user_id in completion_reward_issue_status_dict
+        or player_backend_user_id in preview_list
+    ):
         if player_backend_user_id in preview_list:
             value = completion_reward_issue_status_dict[player_backend_user_id]
             if value == "not_issued":