Update app.py

app.py

@@ -201,8 +201,8 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
     # Header section with improved instructions
     gr.Markdown("""
     # 🔮 AI Story Studio
-**Collaborate with AI to craft your own adventure, one scene at a time.**  
-Pick a genre, start with a prompt or write your own, and guide the story with your choices.  
+    **Collaborate with AI to craft your own adventure, one scene at a time.**  
+    Pick a genre, start with a prompt or write your own, and guide the story with your choices.  
 
     > **Tip:** The more detail you provide, the deeper the story becomes.
     """)
@@ -213,7 +213,7 @@ Pick a genre, start with a prompt or write your own, and guide the story with yo
         with gr.Column(scale=3):
             # Chat window + user input - USING LIST FORMAT
             chatbot = gr.Chatbot(
-                height=500,
+                height=400,
                 bubble_full_width=True,
                 show_copy_button=True,
                 avatar_images=(None, wizard_avatar),
@@ -253,14 +253,13 @@ Pick a genre, start with a prompt or write your own, and guide the story with yo
             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]
 
             # Simplified update function
             def update_starter_buttons(selected_genre):
                 examples = get_examples_for_genre(selected_genre)
                 results = []
-                for i in range(4):
+                for i in range(3):
                     if i < len(examples):
                         results.append(examples[i])
                     else:
@@ -319,9 +318,11 @@ Pick a genre, start with a prompt or write your own, and guide the story with yo
     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",
@@ -329,13 +330,13 @@ Pick a genre, start with a prompt or write your own, and guide the story with yo
             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")