Update app.py

app.py

@@ -126,10 +126,11 @@ with gr.Blocks() as demo:
         with gr.Column():
             textbox_input = gr.Textbox(label="Prompt", lines=1, placeholder="Enter prompt here...")
             button = gr.Button("Submit")
+            textbox_output = gr.Markdown(label="robot answer:", elem_id="scroll_output")
         with gr.Column():
             StoryBox = gr.Textbox(label="Story", lines=50, placeholder="Story here...", value=story_default_text)
             
-    textbox_output = gr.Markdown(label="robot answer:", elem_id="scroll_output")
+    
 
     start_btn.click(fn=start_chat, outputs=status_box)
     end_btn.click(fn=finish_chat, outputs=status_box)