Trying to simplify

app.py

@@ -116,7 +116,12 @@ def generate_story_and_timeline(include_media=True):
 
     game_structure_with_media, game_structure_without_media = generate_game_structures(formatted_timeline, no_media_formatted_timeline)
 
-    return formatted_timeline, no_media_formatted_timeline, story, game_structure_with_media, game_structure_without_media, game_structure_with_media
+    # if include_media:
+    #     return json.dumps(game_structure_with_media, indent=2)
+    # else:
+    #     return json.dumps(game_structure_without_media, indent=2)
+
+    return formatted_timeline, no_media_formatted_timeline, story, json.dumps(game_structure_with_media, indent=2), json.dumps(game_structure_without_media, indent=2) #, game_structure_with_media
 
 media_file_types = ["image", "video", "audio"]
 
@@ -474,7 +479,7 @@ initgameinfo = start_game()
 
 with gr.Blocks() as demo:
     gr.HTML("""Main ideas for this space is (June 2024): 
-    <div style="display: flex; justify-content: space-between; margin-bottom: 20px; text-align: center;">
+    <div style="display: flex; justify-content: center; margin-bottom: 20px; align-items: center;">
         <div style="width: 20%;">A program exist around data </div>
         <div style="width: 20%;">We can generate almost any media data and more </div>
         <div style="width: 20%;">llms good at short questions </div>
@@ -496,16 +501,20 @@ with gr.Blocks() as demo:
             generate_button = gr.Button("Generate Story and Timeline")
             with gr.Accordion("Can copy in the Test Example State Machine tab - only linear path for now"):
                 media_checkbox = gr.Checkbox(label="Include Media", value=True)
-                game_structure_output_text = gr.Code(language="json")
-                screen_game_structure_with_media = gr.State()
-                screen_game_structure_without_media = gr.State()
-
-                media_checkbox.change(
-                    fn=show_game_structure,
-                    inputs=[media_checkbox, screen_game_structure_with_media, screen_game_structure_without_media],
-                    outputs=[game_structure_output_text]
-                )
-            generate_button.click(generate_story_and_timeline, inputs=[], outputs=[timeline_output_with_assets, timeline_output, story_output, screen_game_structure_with_media, screen_game_structure_without_media, game_structure_output_text])
+                with gr.Row():
+                    game_structure_output_text_with_media = gr.Code(language="json")
+                    game_structure_output_text = gr.Code(language="json")
+                # screen_game_structure_with_media = gr.State()
+                # screen_game_structure_without_media = gr.State()
+
+                # media_checkbox.change(
+                #     fn=show_game_structure,
+                #     inputs=[media_checkbox, screen_game_structure_with_media, screen_game_structure_without_media],
+                #     outputs=[game_structure_output_text]
+                # )
+
+
+            generate_button.click(generate_story_and_timeline, inputs=[], outputs=[timeline_output_with_assets, timeline_output, story_output, game_structure_output_text_with_media, game_structure_output_text]) #screen_game_structure_with_media, screen_game_structure_without_media, game_structure_output_text])
         # with gr.Row():
             # gr.Textbox(value=timeline_with_media, lines=4)
             # gr.Textbox(value=timeline_without_media, lines=4)
@@ -655,7 +664,7 @@ with gr.Blocks() as demo:
 
         with gr.Tab("Save files"):
             gr.HTML("For Dynamic events overnight or when player is not active what can LLMS edit? <br><br>eg. Waiting for a letter from a random npc can be decided by the llm <br>eg. Improved Stats on certain days (eg. bitrthday) <br>Privacy <br>User Directed DLC eg. Rockstar Editor with local llm guide")
-            gr.HTML("Placeholder for huggingface spaces that can assist  ")
+            gr.HTML("Placeholder for huggingface spaces that can assist - https://huggingface.co/nvidia/Nemotron-4-340B-Instruct (Purpose is supposed to be synthetic data generation) ")
             gr.HTML("Placeholder for models small enough to run on cpu here in this space that can assist (9b and under) <br>initial floor for testing can be https://huggingface.co/spaces/Qwen/Qwen2-0.5B-Instruct, https://huggingface.co/spaces/Qwen/Qwen2-1.5b-instruct-demo, https://huggingface.co/spaces/stabilityai/stablelm-2-1_6b-zephyr, https://huggingface.co/spaces/IndexTeam/Index-1.9B, https://huggingface.co/microsoft/Phi-3-mini-4k-instruct")
 
         with gr.Tab("Images"):