Update app.py

app.py

@@ -103,11 +103,9 @@ For each response you MUST:
 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 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
@@ -115,7 +113,6 @@ CRITICAL RULES:
 - Keep all narrative before the choices
 - End every response with exactly three numbered options
 - Each choice must start with "You" followed by a verb
-
 Remember: The story continues ONLY when the player makes a choice."""
     return system_message
 
@@ -186,19 +183,24 @@ def respond(message: str, chat_history: List[Tuple[str, str]], genre: Optional[s
 def save_story(chat_history):
     """Convert chat history to markdown and return as downloadable file"""
     if not chat_history:
-        return None
+        return gr.File.update(value=None)
     
     try:
         story_text = "# My Interactive Adventure\n\n"
-        # Process chat history (already in tuples of user, bot messages)
         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 {"name": "my_adventure.md", "data": story_text.encode('utf-8')}
+        # Create temporary file
+        temp_file = "temp_story.md"
+        with open(temp_file, "w", encoding="utf-8") as f:
+            f.write(story_text)
+        
+        return temp_file
+        
     except Exception as e:
         logging.error(f"Error saving story: {e}")
-        return None
+        return gr.File.update(value=None)
 
 # Add this function to get a custom avatar image URL
 def get_storyteller_avatar_url():
@@ -210,7 +212,6 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
     # Header section with improved instructions
     gr.Markdown("""
     # 🔮 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.
     
@@ -218,7 +219,6 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
     * Pick a genre that sparks your imagination
     * Choose a Story Starter or start from scratch
     * Work with AI—describe your actions, make decisions, and see how the story unfolds together
-
     
     > **Tip:** The more detail you provide, the deeper the story becomes.
     """)
@@ -342,13 +342,15 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
             label="Download your story",
             file_count="single",
             file_types=[".md"],
-            interactive=False
+            interactive=False,
+            visible=True
         )
     
     save_btn.click(
         fn=save_story,
         inputs=[chatbot],
-        outputs=[story_output]
+        outputs=story_output,
+        queue=False  # Process immediately
     )
 
     # Initialize buttons with default fantasy genre examples