Update app.py

app.py

@@ -178,14 +178,17 @@ def respond(message, chat_history, genre=None, use_full_memory=True):
         error_message = f"Story magic temporarily interrupted. Please try again. (Error: {str(e)})"
         yield chat_history + [(message, error_message)]
 
-def update_examples(genre):
-    """Update the example prompts based on the selected genre"""
-    examples = get_examples_for_genre(genre)
-    examples.insert(0, "Begin my adventure")  # Always include this option
-    return gr.Examples(
-        examples=[[ex] for ex in examples],
-        inputs=msg
-    )
+def save_story(chat_history):
+    """Convert chat history to markdown for download"""
+    if not chat_history:
+        return "No story to save yet!"
+    
+    story_text = "# My Interactive Adventure\n\n"
+    for user_msg, bot_msg in chat_history:
+        story_text += f"**Player:** {user_msg}\n\n"
+        story_text += f"**Story:** {bot_msg}\n\n---\n\n"
+    
+    return story_text
 
 # Create interface with additional customization options
 with gr.Blocks(theme=gr.themes.Soft()) as demo:
@@ -226,27 +229,52 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
                 info="When enabled, the AI will try to remember the entire story"
             )
             
-            # Initialize with fantasy examples
-            examples_box = gr.Examples(
-                examples=[[ex] for ex in get_examples_for_genre("fantasy")],
-                inputs=msg,
-                label="Examples"
+            # Create story starter buttons that automatically submit
+            gr.Markdown("## Story Starters")
+            story_starters = []
+            
+            def create_example_click_handler(example_text):
+                def example_click():
+                    return example_text
+                return example_click
+            
+            # Create a start button
+            start_btn = gr.Button("Begin my adventure", variant="secondary")
+            start_btn.click(
+                fn=create_example_click_handler("Begin my adventure"),
+                outputs=[msg],
+                queue=False
+            ).then(
+                fn=respond,
+                inputs=[msg, chatbot, genre, full_memory],
+                outputs=[chatbot]
             )
+            
+            # Create example buttons for the current genre
+            example_buttons = []
+            for i, example in enumerate(get_examples_for_genre("fantasy")):
+                btn = gr.Button(example, elem_id=f"example_{i}")
+                btn.click(
+                    fn=create_example_click_handler(example),
+                    outputs=[msg],
+                    queue=False
+                ).then(
+                    fn=respond,
+                    inputs=[msg, chatbot, genre, full_memory],
+                    outputs=[chatbot]
+                )
+                example_buttons.append(btn)
     
-    # Update examples when genre changes
-    def update_example_text(genre):
-        return "\n".join(get_examples_for_genre(genre))
-    
-    examples_textbox = gr.Textbox(
-        value=update_example_text("fantasy"),
-        label="Example Story Starters",
-        interactive=False
-    )
+    # Function to update example buttons when genre changes
+    def update_example_buttons(genre):
+        examples = get_examples_for_genre(genre)
+        return [gr.Button.update(value=example) for example in examples[:4]]  # Limit to 4 examples
     
+    # Connect genre dropdown to update example buttons
     genre.change(
-        fn=update_example_text,
+        fn=update_example_buttons,
         inputs=[genre],
-        outputs=[examples_textbox]
+        outputs=example_buttons
     )
     
     # Set up event handlers for the chatbot
@@ -258,17 +286,6 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
     clear.click(lambda: "", None, msg, queue=False)
     
     # Add a download story button
-    def save_story(chat_history):
-        if not chat_history:
-            return "No story to save yet!"
-        
-        story_text = "# My Interactive Adventure\n\n"
-        for user_msg, bot_msg in chat_history:
-            story_text += f"**Player:** {user_msg}\n\n"
-            story_text += f"**Story:** {bot_msg}\n\n---\n\n"
-        
-        return story_text
-    
     with gr.Row():
         save_btn = gr.Button("Save Story as Markdown", variant="secondary")
         story_output = gr.Markdown(visible=False)