Update app.py

app.py

@@ -318,25 +318,25 @@ 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
-with gr.Row():
-    save_btn = gr.Button("Download My Story", variant="secondary")
+    # "Download My Story" row
+    with gr.Row():
+        save_btn = gr.Button("Download My Story", variant="secondary")
+        
+        with gr.Column(scale=1, min_width=150):  # Adjust width to make it smaller
+            story_output = gr.File(
+                label="Download your story",
+                file_count="single",
+                file_types=[".md"],
+                interactive=False,
+                visible=True
+            )
     
-    with gr.Column(scale=1, min_width=150):  # Adjust width to make it smaller
-        story_output = gr.File(
-            label="Download your story",
-            file_count="single",
-            file_types=[".md"],
-            interactive=False,
-            visible=True
-        )
-
-save_btn.click(
-    fn=save_story,
-    inputs=[chatbot],
-    outputs=story_output,
-    queue=False  # Process immediately
-)
+    save_btn.click(
+        fn=save_story,
+        inputs=[chatbot],
+        outputs=story_output,
+        queue=False  # Process immediately
+    )
 
     # Initialize buttons with default fantasy genre examples
     initial_examples = get_examples_for_genre("fantasy")