Load game supports media now?

app.py

@@ -2,6 +2,7 @@ import gradio as gr
 import random
 import json
 import re
+import os
 
 # Default configuration template
 default_config = {
@@ -169,7 +170,7 @@ def show_elements(text):
 #         if location == 'end':
 #             continue
         
-#         with gr.Accordion(f"Location: {location}"):
+#         with gr.Accordion(f"Location: {location} - Previous desciption {details['description']}", open=False):
 #             description = gr.Textbox(label="Description", value=details['description'], interactive=True)
 #             outputs.append(description)
             
@@ -182,16 +183,15 @@ def show_elements(text):
 #             transitions = gr.Textbox(label="Transitions", value=json.dumps(details['transitions']), interactive=True)
 #             outputs.append(transitions)
     
-#     def update_json():
+#     def update_json(*current_values):
 #         updated_data = {"masterlocation1": {}}
-#         for i, location in enumerate(masterlocation1.keys()):
-#             if location == 'end':
-#                 continue
+#         locations = [loc for loc in masterlocation1.keys() if loc != 'end']
+#         for i, location in enumerate(locations):
 #             updated_data["masterlocation1"][location] = {
-#                 "description": outputs[i*4].value,
-#                 "events": json.loads(outputs[i*4 + 1].value),
-#                 "choices": json.loads(outputs[i*4 + 2].value),
-#                 "transitions": json.loads(outputs[i*4 + 3].value)
+#                 "description": current_values[i*4],
+#                 "events": json.loads(current_values[i*4 + 1]),
+#                 "choices": json.loads(current_values[i*4 + 2]),
+#                 "transitions": json.loads(current_values[i*4 + 3])
 #             }
 #         updated_data["masterlocation1"]["end"] = masterlocation1["end"]
 #         return json.dumps(updated_data, indent=2)
@@ -199,7 +199,7 @@ def show_elements(text):
 #     update_button = gr.Button("Update JSON")
 #     json_output = gr.Textbox(label="Updated JSON", lines=10)
     
-#     update_button.click(update_json, outputs=json_output)
+#     update_button.click(update_json, inputs=outputs, outputs=json_output)
     
 #     return outputs + [update_button, json_output]
 
@@ -213,7 +213,7 @@ def show_elements_json_input(json_input):
         if location == 'end':
             continue
         
-        with gr.Accordion(f"Location: {location}"):
+        with gr.Accordion(f"Location: {location} - Previous description {details['description']}", open=False):
             description = gr.Textbox(label="Description", value=details['description'], interactive=True)
             outputs.append(description)
             
@@ -225,16 +225,21 @@ def show_elements_json_input(json_input):
             
             transitions = gr.Textbox(label="Transitions", value=json.dumps(details['transitions']), interactive=True)
             outputs.append(transitions)
+            
+            # New media field
+            media = gr.Textbox(label="Media", value=json.dumps(details.get('media', [])), interactive=True)
+            outputs.append(media)
     
     def update_json(*current_values):
         updated_data = {"masterlocation1": {}}
         locations = [loc for loc in masterlocation1.keys() if loc != 'end']
         for i, location in enumerate(locations):
             updated_data["masterlocation1"][location] = {
-                "description": current_values[i*4],
-                "events": json.loads(current_values[i*4 + 1]),
-                "choices": json.loads(current_values[i*4 + 2]),
-                "transitions": json.loads(current_values[i*4 + 3])
+                "description": current_values[i*5],
+                "events": json.loads(current_values[i*5 + 1]),
+                "choices": json.loads(current_values[i*5 + 2]),
+                "transitions": json.loads(current_values[i*5 + 3]),
+                "media": json.loads(current_values[i*5 + 4])  # New media field
             }
         updated_data["masterlocation1"]["end"] = masterlocation1["end"]
         return json.dumps(updated_data, indent=2)
@@ -246,6 +251,19 @@ def show_elements_json_input(json_input):
     
     return outputs + [update_button, json_output]
 
+def create_media_component(file_path):
+    _, extension = os.path.splitext(file_path)
+    extension = extension.lower()[1:]  # Remove the dot and convert to lowercase
+
+    if extension in ['jpg', 'jpeg', 'png', 'gif']:
+        return gr.Image(value=file_path, label="Image Input")
+    elif extension in ['mp4', 'avi', 'mov']:
+        return gr.Video(value=file_path, label="Video Input")
+    elif extension in ['mp3', 'wav', 'ogg']:
+        return gr.Audio(value=file_path, label="Audio Input")
+    else:
+        return gr.Textbox(value=file_path, label=f"File: {os.path.basename(file_path)}")
+
 def convert_timeline_to_game_structure(timeline):
     lines = timeline.split('\n')
     game_structure = {}
@@ -490,7 +508,7 @@ def start_game(starting_location='village', starting_state='start'):
     description, choices, game_log = game_session.get_current_state_info()
     return description, choices, game_log, game_session
 
-def make_choice(choice, game_session):
+def make_choice(choice, game_session): #Calls the nested make choice function in the game session class
     if not choice:
         description, choices, game_log = game_session.get_current_state_info()
         return description, choices, "Please select a choice before proceeding.", game_session
@@ -500,7 +518,7 @@ def make_choice(choice, game_session):
 
     return result[0], gr.update(choices=result[1]), result[2], game_session
 
-def load_game(custom_config=None):
+def load_game(custom_config=None, with_media=False):
     global all_states
     if custom_config:
         try:
@@ -515,7 +533,17 @@ def load_game(custom_config=None):
             game_session = GameSession(starting_location, starting_state)
             description, choices, game_log = game_session.get_current_state_info()
             new_path_errors = validate_transitions(all_states)
-
+            media = None
+            output_media = None
+
+            if with_media:
+                if all_states[starting_location][starting_state][media]:
+                    media = all_states[starting_location][starting_state][media]
+                    for media_path in media:
+                        media_component = create_media_component(media_path)
+                        output_media.append(media_component)
+                    return gr.update(value=f"Custom configuration loaded successfully! \n{new_path_errors}"), game_log, description, gr.update(choices=choices), game_session, gr.update(value=custom_config), output_media
+                
             return gr.update(value=f"Custom configuration loaded successfully! \n{new_path_errors}"), game_log, description, gr.update(choices=choices), game_session, gr.update(value=custom_config)
         except json.JSONDecodeError as e:
             # Get the line number and column of the error
@@ -581,7 +609,6 @@ with gr.Blocks() as demo:
                     generate_no_media_timeline_points = gr.Slider(minimum=1, value=5, step=1, maximum=30, label="Choose the amount of media timeline points")
                     generate_with_media_check = gr.Checkbox(label="Generate with media", value=True)
                 generate_button = gr.Button("Generate Story and Timeline")
-                gr.HTML("Still under construction - The update json below doesnt work for so dont edit here yet")
 
                 @gr.render(inputs=game_structure_output_text_with_media)
                 def update(game_structure_output_text_with_media):
@@ -635,7 +662,10 @@ with gr.Blocks() as demo:
                     gr.Markdown("# Text-based Adventure Game")
 
                     wadescription = gr.Textbox(label="Current Situation", lines=4, value=initgameinfo[0])
-                    wamedia = gr.HTML("Placeholder to load all media tests")
+                    wamediabool = gr.State(value=True)
+                    with gr.Group() as wamedia: 
+                        gr.HTML("Placeholder to load all media tests")
+                    
                     wachoices = gr.Radio(label="Your Choices", choices=initgameinfo[1])
                     wasubmit_btn = gr.Button("Make Choice")
                     wagame_log = gr.Textbox(label="Game Log", lines=20, value=initgameinfo[2])
@@ -654,8 +684,8 @@ with gr.Blocks() as demo:
 
                         wacustom_configbtn.click(
                             load_game,
-                            inputs=[wacustom_config],
-                            outputs=[waerror_box, wagame_log, wadescription, wachoices, wagame_session, wacustom_config]
+                            inputs=[wacustom_config, wamediabool],
+                            outputs=[waerror_box, wagame_log, wadescription, wachoices, wagame_session, wacustom_config, wamedia]
                         )
     with gr.Tab("Custom JS Config Creator"):
         gr.HTML("Companion Space for zerogpu / client api workflow planning for a way to send a zip to the Basic Game Engine at the bottom of https://huggingface.co/spaces/KwabsHug/TestSvelteStatic (Also to test how much can be done majority on cpu)")