Update app.py

app.py

@@ -102,17 +102,21 @@ For each response you MUST:
 2. Always use second-person perspective ("you", "your") to maintain reader immersion
 3. Include dialogue or your character's thoughts that reveal personality and motivations
 4. Create a strong sense of atmosphere appropriate for {selected_genre}
-5. End EVERY response with exactly three numbered choices like this:
-1. [Complete sentence in second-person starting with a verb (e.g., "You decide to..."/"You attempt to...")]
-2. [Complete sentence in second-person starting with a verb (e.g., "You sneak towards..."/"You call out to...")]
-3. [Complete sentence in second-person starting with a verb (e.g., "You examine..."/"You reach for...")]
-IMPORTANT:
-- Always maintain second-person perspective throughout the narrative
-- Always end with exactly three numbered choices
-- Never skip the choices or respond with just narrative
+5. End with EXACTLY THREE numbered choices and NOTHING ELSE AFTER THEM:
+
+1. [Complete sentence in second-person starting with a verb]
+2. [Complete sentence in second-person starting with a verb]
+3. [Complete sentence in second-person starting with a verb]
+
+CRITICAL RULES:
+- Provide only ONE set of three choices at the very end of your response
+- Never continue the story after giving choices
+- Never provide additional choices
+- Keep all narrative before the choices
+- End every response with exactly three numbered options
 - Each choice must start with "You" followed by a verb
-- Format choices exactly as shown above with numbers 1-3
-Keep the story cohesive by referencing previous events and choices."""
+
+Remember: The story continues ONLY when the player makes a choice."""
     return system_message
 
 def create_story_summary(chat_history):
@@ -178,15 +182,16 @@ def respond(message: str, chat_history: List[Tuple[str, str]], genre: Optional[s
         return "", chat_history + [(message, error_msg)]
 
 def save_story(chat_history):
-    """Convert chat history to markdown for download"""
+    """Convert chat history to markdown and return as downloadable file"""
     if not chat_history:
-        return "No story to save yet!"
+        return None
     
     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 as a file for download
     return story_text
 
 # Add this function to get a custom avatar image URL
@@ -198,7 +203,7 @@ def get_storyteller_avatar_url():
 with gr.Blocks(theme=gr.themes.Soft()) as demo:
     # Header section with improved instructions
     gr.Markdown("""
-    # 🔮 Interactive Story Time
+    # 🔮 AI Story Studio
 
     This AI-powered storytelling experience is a collaboration between you and AI—building a world, scene by scene. 
     You lead the way with your choices, and AI responds, expanding on your ideas.
@@ -332,15 +337,17 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
     # "Download My Story" row
     with gr.Row():
         save_btn = gr.Button("Download My Story", variant="secondary")
-        story_output = gr.Markdown(visible=False)
+        story_output = gr.File(
+            label="Download your story",
+            file_count="single",
+            type="string",
+            file_types=[".md"]
+        )
     
-    save_btn.click(save_story, inputs=[chatbot], outputs=[story_output])
     save_btn.click(
-        fn=lambda: True,
-        inputs=None,
-        outputs=story_output,
-        js="() => {document.getElementById('story_output').scrollIntoView();}",
-        queue=False
+        fn=lambda x: (save_story(x), "story.md") if x else None,
+        inputs=[chatbot],
+        outputs=[story_output]
     )
 
     # Initialize buttons with default fantasy genre examples