Fix loading json bugs

sessions/pokemon_types_flavoured_soft_drinks/data.json

@@ -4,21 +4,21 @@
             "start": "As the meteor dust settled, Pikachu and Charmander emerged, curious about their new surroundings. Nearby, a bustling city was abuzz with people trying to comprehend the arrival of these creatures. Suddenly, a mysterious figure appeared, offering a challenge: \"Can you help these Pokemon find their favorite soft drinks?\""
         },
         "2": {
-            "Pikachu's_Favorite_Drink_Quest": "Unexplored!",
+            "Pikachu's_Favorite_Drink_Quest": "As you approach Pikachu, the mysterious figure hands you a map with cryptic symbols. \"This will lead you to the Electric Berry Soda,\" they whisper. The path begins in the heart of the city, but strange shadows lurk in the alleyways. What could they be hiding?",
             "Charmander's_Favorite_Drink_Quest": "As you choose to help Charmander, the mysterious figure reveals a ticking clock: \"A rare solar eclipse will occur soon, and only during this time can Charmander identify its favorite drink. Hurry, the eclipse is approaching fast!\"",
             "Both_Pokemon's_Drink_Quest": "Unexplored!"
         },
         "3": {
-            "City_Search": "Unexplored!",
-            "Local_Expert": "A local expert, Professor Maple, quickly arrives with a cart full of uniquely flavored drinks. She explains that Charmander's favorite might be a fiery blend. The clock ticks loudly, signaling the eclipse's imminent arrival. Charmander's eyes widen as it sniffs the air.",
+            "City_Search": "The map leads you through the bustling streets, where a sudden flash of light reveals a group of shadowy figures blocking your path. They demand a challenge before letting you pass. Pikachu and Charmander stand ready, sparks flying and flames flickering. What will you do next?",
+            "Local_Expert": "You and your Pokemon cautiously step into the alley, where the shadows reveal a hidden group of Pokemon enthusiasts. They seem to be guarding a secret stash of rare Pokemon-themed soft drinks. One of them steps forward, intrigued by your presence, and asks, \"Are you here for the challenge or something more?\"",
             "Taste_Test": "Unexplored!"
         },
         "4": {
-            "Charmander_takes_a_sip": "Charmander takes a sip of the Cinnamon Flame Fizz, its eyes lighting up with delight. Just as the eclipse begins, a warm glow surrounds Charmander, signaling it has found its favorite drink. The crowd cheers, and Professor Maple smiles, knowing the challenge was a success."
+            "Charmander_takes_a_sip": "The shadowy figures pause, intrigued by your offer of negotiation. They whisper among themselves, then reveal their true form: a group of mischievous Ditto. They agree to a trade—solve their riddle, and they'll guide you to the hidden Electric Berry Soda fountain."
         },
         "5": {
-            "Celebrate_the_success": "Unexplored!",
-            "Learn_about_other_Pokemon": "Unexplored!",
+            "Celebrate_the_success": "As the eclipse reaches its peak, a shimmering portal opens, inviting the Pokemon back to their world. Charmander, grateful for the help, gifts you a fiery gem as a token of friendship. The crowd waves goodbye, feeling a sense of wonder and unity.",
+            "Learn_about_other_Pokemon": "With the Ditto's riddle solved, a path opens to the Electric Berry Soda fountain. Professor Maple smiles warmly, revealing a book filled with tales of other Pokemon. As you sip the soda, the Pokemon cheer, and a sense of peace settles over the city.",
             "Plan_future_adventures": "As the eclipse fades, the sky returns to its usual hue, and the crowd disperses with excitement. Professor Maple approaches you, suggesting a future journey to explore more Pokemon and their unique preferences. Pikachu and Charmander look eager, ready for more adventures by your side."
         },
         "6": {
@@ -47,6 +47,22 @@
         "7b000a21-8ee5-4f8f-9f80-3039f22c1340": {
             "question": "What_do_you_decide_to_do_next?",
             "options": "{'Embark_on_a_new_adventure_with_Pikachu_and_Charmander': 'New_Adventure_Begins', \"Reflect_on_the_impact_of_the_eclipse_and_the_Pokemon's_arrival\": 'Reflective_Moment', 'Bid_farewell_to_Professor_Maple_and_the_Pokemon,_returning_to_your_usual_life': 'Return_to_Normalcy'}"
+        },
+        "8681384d-fba7-45aa-8daa-6a4d951c552f": {
+            "question": "What_do_you_do_as_the_portal_closes?",
+            "options": "{'Wave_farewell_and_reflect_on_the_adventure': 'Reflective_Moment', 'Prepare_for_a_new_adventure_with_the_fiery_gem': 'New_Adventure_Begins', 'Return_to_your_daily_life_with_a_story_to_tell': 'Return_to_Normalcy'}"
+        },
+        "28399188-1fe7-4d89-ae01-51829063301f": {
+            "question": "Do_you_follow_the_map_or_investigate_the_shadows?",
+            "options": "{'Follow_the_map': 'City_Search', 'Investigate_the_shadows': 'Local_Expert', 'Seek_help_from_a_local': 'Taste_Test'}"
+        },
+        "d112564b-30cc-4d67-bb98-a2274e60a186": {
+            "question": "How_will_you_respond_to_the_shadowy_figures?",
+            "options": "{'Challenge_them_to_a_battle': 'Charmander_takes_a_sip', 'Try_to_negotiate_with_them': 'Charmander_takes_a_sip', 'Look_for_an_alternate_route': 'Charmander_takes_a_sip'}"
+        },
+        "089ed925-3c43-4f23-86d9-5c7c6718fdf3": {
+            "question": "How_would_you_like_to_end_this_adventure?",
+            "options": "{'Celebrate_with_the_Pokemon': 'Reflective_Moment', 'Plan_a_new_journey_with_Professor_Maple': 'New_Adventure_Begins', 'Return_to_the_city_and_share_your_story': 'Return_to_Normalcy'}"
         }
     },
     "state_question_map": {
@@ -54,16 +70,20 @@
             "start": "3a9f3748-d14b-43a7-beeb-9ca20b46427b"
         },
         "2": {
-            "Charmander's_Favorite_Drink_Quest": "154fe5ae-612c-4cc8-a46a-71021627aa28"
+            "Charmander's_Favorite_Drink_Quest": "154fe5ae-612c-4cc8-a46a-71021627aa28",
+            "Pikachu's_Favorite_Drink_Quest": "28399188-1fe7-4d89-ae01-51829063301f"
         },
         "3": {
-            "Local_Expert": "740b89c8-3b16-42ec-a819-b026163f3761"
+            "Local_Expert": "740b89c8-3b16-42ec-a819-b026163f3761",
+            "City_Search": "d112564b-30cc-4d67-bb98-a2274e60a186"
         },
         "4": {
             "Charmander_takes_a_sip": "a388a0ff-ee45-4d9e-a285-d9deeaff659e"
         },
         "5": {
-            "Plan_future_adventures": "7b000a21-8ee5-4f8f-9f80-3039f22c1340"
+            "Plan_future_adventures": "7b000a21-8ee5-4f8f-9f80-3039f22c1340",
+            "Celebrate_the_success": "8681384d-fba7-45aa-8daa-6a4d951c552f",
+            "Learn_about_other_Pokemon": "089ed925-3c43-4f23-86d9-5c7c6718fdf3"
         }
     },
     "categories": [
@@ -94,7 +114,7 @@
         }
     ],
     "categories_seen": {
-        "Electric Zing": false,
+        "Electric Zing": true,
         "Fire Blaze": true,
         "Water Splash": false,
         "Grass Burst": false,
@@ -104,6 +124,18 @@
         "3a9f3748-d14b-43a7-beeb-9ca20b46427b+Charmander-154fe5ae-612c-4cc8-a46a-71021627aa28+Consult_a_local_expert_on_Pokemon_tastes-740b89c8-3b16-42ec-a819-b026163f3761+Cinnamon_Flame_Fizz-a388a0ff-ee45-4d9e-a285-d9deeaff659e+Reflect_on_the_experience_and_plan_for_future_adventures-7b000a21-8ee5-4f8f-9f80-3039f22c1340+Embark_on_a_new_adventure_with_Pikachu_and_Charmander": [
             "Fire Blaze",
             "You chose to help Charmander, a Fire-type Pokémon, and selected the Cinnamon Flame Fizz, a fiery drink, as its favorite. This aligns with the traits of the Fire Blaze category, which is described as a fiery drink that ignites your taste buds with the heat of Fire-type Pokémon. Your choices reflect a preference for the spicy, warming, and invigorating traits associated with Fire Blaze."
+        ],
+        "3a9f3748-d14b-43a7-beeb-9ca20b46427b+Charmander-8681384d-fba7-45aa-8daa-6a4d951c552f+Consult_a_local_expert_on_Pokemon_tastes-8681384d-fba7-45aa-8daa-6a4d951c552f+Spicy_Chili_Soda": [
+            "Fire Blaze",
+            "You chose to help Charmander, a Fire-type Pokémon, and selected a Spicy Chili Soda, which aligns with the traits of a fiery drink that ignites your taste buds with the heat of Fire-type Pokémon."
+        ],
+        "3a9f3748-d14b-43a7-beeb-9ca20b46427b+Pikachu-28399188-1fe7-4d89-ae01-51829063301f+Investigate_the_shadows-740b89c8-3b16-42ec-a819-b026163f3761+Spicy_Chili_Soda-7b000a21-8ee5-4f8f-9f80-3039f22c1340+Reflect_on_the_experience_and_plan_for_future_adventures": [
+            "Fire Blaze",
+            "Your choice to help Pikachu first shows a willingness to engage with Electric-type Pokémon, but your decision to investigate the shadows and select the Spicy Chili Soda for Charmander aligns more closely with the traits of the Fire Blaze category. The spicy and invigorating nature of the drink you chose for Charmander, along with the warm glow and delight it brought, reflects the fiery and warming traits of Fire-type Pokémon. Additionally, your decision to reflect on the experience and plan for future adventures suggests an invigorating and forward-thinking mindset, which is characteristic of the Fire Blaze category."
+        ],
+        "3a9f3748-d14b-43a7-beeb-9ca20b46427b+Pikachu-d112564b-30cc-4d67-bb98-a2274e60a186+Follow_the_map-a388a0ff-ee45-4d9e-a285-d9deeaff659e+Try_to_negotiate_with_them-089ed925-3c43-4f23-86d9-5c7c6718fdf3+Thank_Professor_Maple_and_ask_about_other_Pokemon-089ed925-3c43-4f23-86d9-5c7c6718fdf3+Return_to_the_city_and_share_your_story": [
+            "Electric Zing",
+            "Your journey began with a choice to help Pikachu, an Electric-type Pokémon, and your adventure was centered around finding the Electric Berry Soda. Your decisions to follow the map and negotiate with the shadowy figures, who turned out to be Ditto, led you to the Electric Berry Soda fountain. This aligns with the traits of the Electric Zing category, which is associated with Electric-type Pokémon and their zesty, electrifying nature."
         ]
     }
 }

src/agents/coordinator.py

@@ -180,9 +180,16 @@ def load_coordinator() -> None:
     combined_data = load_information(filepath)
 
     states = {int(k): v for k, v in combined_data['states'].items()}
+    for k, v in combined_data['questions'].items():
+        w = [c for c in v['options'].replace('\'', '\"')]
+        for i in range(1, len(w) - 1):
+            if w[i] == '\"' and w[i - 1].isalpha() and w[i + 1].isalpha():
+                w[i] = '\''
+        combined_data['questions'][k]['options'] = ''.join(w)
+
     questions = {k: Question(**{
             'question': v['question'],
-            'options': json.loads(v['options'].replace('\'', '\"'))
+            'options': json.loads(v['options'])
         }) for k, v in combined_data['questions'].items()}
     state_question_map = {int(k): v for k, v in combined_data['state_question_map'].items()}
     categories = [Category(**c) for c in combined_data['categories']]

src/frontend/chatbot.py

@@ -100,7 +100,7 @@ def control_screen(
     return {
         chatbot: gr.update(visible=is_visible),
         button_row: gr.Row(visible=is_visible),
-        restart_button: gr.update(visible=is_visible)
+        restart_button: gr.update(visible=False)
     }
 
 

src/frontend/continue_story_screen.py

@@ -22,7 +22,7 @@ information_text = gr.Text(
     key='continue_story_user_information',
     preserved_by_key='key',
     visible=False,
-    label=None
+    label=''
 )
 session_dir = os.path.join(
     os.path.dirname(__file__),
@@ -54,6 +54,7 @@ for i, dirname in enumerate(sorted(os.listdir(session_dir))):
     )
     text = gr.Text(
         value=story_context,
+        label='',
         interactive=False,
         visible=False,
         key=f'continue_story_{i}',

src/frontend/progress_screen.py

@@ -30,6 +30,7 @@ category_buttons = [
 ]
 category_text = gr.Text(
     value='Select a category!',
+    label='',
     visible=False,
     interactive=False,
     key='category_text',
@@ -45,7 +46,8 @@ graph_plot = gr.Plot(
 def control_screen(is_visible: bool) -> Dict[gr.Row, gr.Row]:
     return {
         categories_row: gr.Row(visible=is_visible),
-        graph_row: gr.Row(visible=is_visible)
+        graph_row: gr.Row(visible=is_visible),
+        category_text: gr.update(visible=is_visible)
     }
 
 
@@ -102,14 +104,19 @@ def update_graph():
             question = C.questions[question_uuid]
             question_str = question.question
             options = question.options
-            for option, foll_state in options.items():
-                edge = (
-                    f'{state} (Stage {level})',
-                    f'{foll_state} (Stage {level + 1})'
-                )
-                edges.append(edge)
-                edge_attributes['question'].append(question_str)
-                edge_attributes['option'].append(option)
+            prev_foll_state = None
+            for option, foll_state in sorted(options.items(), key=lambda x: x[1]):
+                if prev_foll_state is not None and prev_foll_state == foll_state:
+                    edge_attributes['option'][-1] += f', {option}'
+                else:
+                    edge = (
+                        f'{state} (Stage {level})',
+                        f'{foll_state} (Stage {level + 1})'
+                    )
+                    edges.append(edge)
+                    edge_attributes['question'].append(question_str)
+                    edge_attributes['option'].append(option)
+                prev_foll_state = foll_state
 
     graph.add_vertices(nodes, attributes=node_attributes)
     graph.add_edges(edges, attributes=edge_attributes)
@@ -148,7 +155,7 @@ def update_graph():
         .apply(lambda x: x.replace('\n', '<br>'))
     node_labels = node_labels['label'].to_list()
     edge_labels = [
-        question.replace('_', ' ') + '\n\n' + option.replace('_', ' ')
+        question.replace('_', ' ') + '\n\n[ ' + option.replace('_', ' ') + ' ]'
         for question, option in zip(
             edge_attributes['question'], edge_attributes['option']
         )
@@ -216,12 +223,13 @@ def control_screen(
         categories_row: gr.Row(visible=is_visible),
         graph_row: gr.Row(visible=is_visible)
     }
-    graph_update = {
-        graph_plot: update_graph() if is_visible else gr.update(visible=False)
+    other_update = {
+        graph_plot: update_graph() if is_visible else gr.update(visible=False),
+        category_text: gr.update(visible=is_visible)
     }
     category_button_updates = update_categories()
 
-    return row_updates | graph_update | category_button_updates
+    return row_updates | other_update | category_button_updates
 
 
 def render():