Update app.py

app.py

@@ -7,6 +7,13 @@ client = InferenceClient("HuggingFaceH4/zephyr-7b-beta")
 
 # Story genres with genre-specific example prompts
 GENRE_EXAMPLES = {
+    "fairy_tale": [
+        "I follow the shimmer of fairy dust into a hidden forest"
+        "I meet a talking rabbit who claims to know a secret about the king’s lost crown"
+        "A tiny dragon appears at my window, asking for help to find its mother"
+        "I step into a clearing where the trees whisper ancient riddles"
+        "A friendly witch invites me into her cozy cottage, offering a warm cup of tea"
+    ]
     "fantasy": [
         "I enter the ancient forest seeking the wizard's tower",
         "I approach the dragon cautiously with my shield raised",
@@ -65,10 +72,7 @@ GENRE_EXAMPLES = {
 
 def get_examples_for_genre(genre):
     """Get example prompts specific to the selected genre"""
-    if genre in GENRE_EXAMPLES:
-        return GENRE_EXAMPLES[genre]
-    else:
-        return GENRE_EXAMPLES["fantasy"]  # Default to fantasy
+    return GENRE_EXAMPLES.get(genre, GENRE_EXAMPLES["fantasy"])
 
 def get_enhanced_system_prompt(genre=None):
     """Generate a detailed system prompt with optional genre specification"""
@@ -103,17 +107,8 @@ def create_story_summary(chat_history):
     }
     return summary_instruction
 
-# NEW HELPER: Convert (user, bot) tuples into the "role/content" dictionaries Gradio needs
 def format_history_for_gradio(history_tuples):
-    """
-    Convert a list of (user_msg, bot_msg) tuples
-    into a list of dicts for 'type="messages"':
-    [
-      {"role": "user", "content": ...},
-      {"role": "assistant", "content": ...},
-      ...
-    ]
-    """
+    """Convert (user, bot) tuples into Gradio 'messages' format (role/content dicts)."""
     messages = []
     for user_msg, bot_msg in history_tuples:
         messages.append({"role": "user", "content": user_msg})
@@ -121,7 +116,7 @@ def format_history_for_gradio(history_tuples):
     return messages
 
 def respond(message, chat_history, genre=None, use_full_memory=True):
-    """Generate a response based on the current message and conversation history"""
+    """Generate a response based on the current message and conversation history."""
     system_message = get_enhanced_system_prompt(genre)
 
     # Convert your existing (user, bot) history into a format for the API request
@@ -155,7 +150,7 @@ def respond(message, chat_history, genre=None, use_full_memory=True):
     # Special handling for story initialization
     if not chat_history or message.lower() in ["start", "begin", "begin my adventure"]:
         api_messages.append({
-            "role": "system", 
+            "role": "system",
             "content": f"Begin a new {genre or 'fantasy'} adventure with an intriguing opening scene. Introduce the protagonist without assuming too much about them."
         })
     
@@ -193,10 +188,9 @@ def save_story(chat_history):
     
     return story_text
 
-# Create interface with additional customization options
 with gr.Blocks(theme=gr.themes.Soft()) as demo:
-    gr.Markdown("# 🔮 Interactive Story Adventure")
-    gr.Markdown("Immerse yourself in an interactive story where your choices shape the narrative.")
+    gr.Markdown("# 🔮 Interactive Story Time")
+    gr.Markdown("Create a completely unique literary world, one choice at a time. Dare to explore the unknown.")
     
     with gr.Row():
         with gr.Column(scale=3):
@@ -206,7 +200,7 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
                 bubble_full_width=False,
                 show_copy_button=True,
                 avatar_images=(None, "🧙"),
-                type="messages"  # Use OpenAI-style messages
+                type="messages"
             )
             msg = gr.Textbox(
                 placeholder="Describe what you want to do next in the story...",
@@ -223,10 +217,9 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
             genre = gr.Dropdown(
                 choices=list(GENRE_EXAMPLES.keys()),
                 label="Story Genre",
-                info="Select a genre for your next adventure",
+                info="Choose the theme of your next adventure",
                 value="fantasy"
             )
-            
             full_memory = gr.Checkbox(
                 label="Full Story Memory",
                 value=True,
@@ -235,38 +228,34 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
             
             gr.Markdown("## Story Starters")
             
-            # -- Create four placeholder buttons for story starters --
+            # Create four placeholder buttons for story starters
             starter_btn1 = gr.Button("Starter 1")
             starter_btn2 = gr.Button("Starter 2")
             starter_btn3 = gr.Button("Starter 3")
             starter_btn4 = gr.Button("Starter 4")
             starter_buttons = [starter_btn1, starter_btn2, starter_btn3, starter_btn4]
-            
-            # Function to update the labels of the 4 starter buttons
+
+            # 1) We'll return a list of 4 dicts, each dict updating 'value' & 'visible'
             def update_starter_buttons(selected_genre):
-                # Grab up to 4 examples from the chosen genre
                 examples = get_examples_for_genre(selected_genre)
-                button_updates = []
+                results = []
                 for i in range(4):
                     if i < len(examples):
-                        button_updates.append(gr.Button.update(value=examples[i], visible=True))
+                        results.append({"value": examples[i], "visible": True})
                     else:
-                        button_updates.append(gr.Button.update(value="(no starter)", visible=False))
-                return button_updates
-            
-            # Initialize the starter buttons with the default 'fantasy' prompts
-            # so they don't stay "Starter 1/2/3/4" on page load.
-            starter_init = update_starter_buttons("fantasy")
-            for btn, init_data in zip(starter_buttons, starter_init):
-                btn.update(**init_data)
-            
-            # Function that populates the msg with the chosen starter text, then calls respond
+                        results.append({"value": "(no starter)", "visible": False})
+                return tuple(results)  # we must return a tuple/list with 4 items for 4 outputs
+
+            # 2) Initialize them with "fantasy" so they don't stay "Starter X" on page load
+            # We'll just call the function and store the results in a variable, then apply them in a .load() event
+            initial_button_data = update_starter_buttons("fantasy")  # returns 4 dicts
+
+            # 3) We'll define a "pick_starter" function that sets msg to the chosen text
             def pick_starter(starter_text, chat_history, selected_genre, memory_flag):
-                return starter_text  # This goes into 'msg'
-            
-            # Hook each starter button:
-            #  1) Put the chosen text into 'msg'
-            #  2) Then call 'respond' to update the chatbot
+                # Putting 'starter_text' into the msg
+                return starter_text
+
+            # 4) Connect each starter button:
             for starter_button in starter_buttons:
                 starter_button.click(
                     fn=pick_starter,
@@ -279,15 +268,15 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
                     outputs=[chatbot],
                     queue=False
                 )
-            
-            # Connect the genre dropdown to dynamically update starter buttons
+
+            # 5) Dynamically update the 4 buttons if the user changes the genre
             genre.change(
                 fn=update_starter_buttons,
                 inputs=[genre],
                 outputs=starter_buttons
             )
-    
-    # -- Chat submission + button events --
+
+    # Handler for user input
     msg.submit(respond, [msg, chatbot, genre, full_memory], [chatbot])
     submit.click(respond, [msg, chatbot, genre, full_memory], [chatbot])
     
@@ -295,7 +284,7 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
     clear.click(lambda: [], None, chatbot, queue=False)
     clear.click(lambda: "", None, msg, queue=False)
     
-    # -- "Download My Story" row --
+    # "Download My Story" row
     with gr.Row():
         save_btn = gr.Button("Download My Story", variant="secondary")
         story_output = gr.Markdown(visible=False)
@@ -309,5 +298,13 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
         queue=False
     )
 
+    # 6) Finally, run a "load" event to apply initial_button_data to the 4 button outputs on page load
+    def load_initial_buttons():
+        # Just return our precomputed tuple of 4 dicts
+        return initial_button_data
+
+    demo.load(fn=load_initial_buttons, outputs=starter_buttons, queue=False)
+
+# Run the app
 if __name__ == "__main__":
     demo.launch(server_name="0.0.0.0", server_port=7860)