Update app.py

app.py

@@ -196,8 +196,8 @@ def save_story(chat_history):
             story_text += f"**Player:** {chat_history[i]}\n\n"
             story_text += f"**Story:** {chat_history[i+1]}\n\n---\n\n"
     
-    # Return as a file for download
-    return story_text
+    # Return as bytes for download
+    return story_text.encode('utf-8')
 
 # Add this function to get a custom avatar image URL
 def get_storyteller_avatar_url():
@@ -339,12 +339,11 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
         story_output = gr.File(
             label="Download your story",
             file_count="single",
-            type="string",
             file_types=[".md"]
         )
     
     save_btn.click(
-        fn=lambda x: (save_story(x), "story.md") if x else None,
+        fn=lambda x: {"name": "story.md", "data": save_story(x)} if x else None,
         inputs=[chatbot],
         outputs=[story_output]
     )