Update app.py

app.py

@@ -50,21 +50,22 @@ with gr.Blocks() as demo:
     """
     <h1 style="text-align: center;">Legal Document Drafting</h1>
     """)
-    with gr.Column(scale=1, min_width=600):
-        with gr.Tab("Writing Assist"):
-            result = gr.components.Textbox(lines=7, label="Writing Assist", placeholder=prompts[0])
-            prompt = gr.components.Textbox(lines=2, label="Prompt", placeholder=examples[0], visible=False)
-            gr.Examples(examples, label='Examples', inputs=[prompt])
-            prompt.change(generate, inputs=[prompt], outputs=[result])
-            btn = gr.Button("Next sentence")
-            btn.click(generate, inputs=[result], outputs=[result])
-        with gr.Tab("Random Generative"):
-            result2 = gr.components.Textbox(lines=7, label="Random Generative", show_label=True, placeholder=prompts[1])
-            gr.Examples(examples, label='Examples', inputs=[result2])
-            rnd_btn = gr.Button("Random Drafting")
-            rnd_btn.click(rnd_generate, inputs=[result2], outputs=[result2])
-    with gr.Column(scale=2, min_width=600):
-        gr.Markdown("Legal Document drafting demo")
+    with gr.Row() as row:
+        with gr.Column(scale=1, min_width=600):
+            with gr.Tab("Writing Assist"):
+                result = gr.components.Textbox(lines=7, label="Writing Assist", placeholder=prompts[0])
+                prompt = gr.components.Textbox(lines=2, label="Prompt", placeholder=examples[0], visible=False)
+                gr.Examples(examples, label='Examples', inputs=[prompt])
+                prompt.change(generate, inputs=[prompt], outputs=[result])
+                btn = gr.Button("Next sentence")
+                btn.click(generate, inputs=[result], outputs=[result])
+            with gr.Tab("Random Generative"):
+                result2 = gr.components.Textbox(lines=7, label="Random Generative", show_label=True, placeholder=prompts[1])
+                gr.Examples(examples, label='Examples', inputs=[result2])
+                rnd_btn = gr.Button("Random Drafting")
+                rnd_btn.click(rnd_generate, inputs=[result2], outputs=[result2])
+        with gr.Column(scale=2, min_width=600):
+            gr.Markdown("Legal Document drafting demo")
     
 if __name__ == "__main__":
     demo.launch()