Update app.py

app.py

@@ -50,19 +50,21 @@ with gr.Blocks() as demo:
     """
     <h1 style="text-align: center;">Legal Document Drafting</h1>
     """)
-    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])
-    gr.Markdown("Legal Document drafting demo")
+    with gr.Column():
+        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():
+        gr.Markdown("Legal Document drafting demo")
     
 if __name__ == "__main__":
     demo.launch()