Update app.py

app.py

@@ -50,15 +50,16 @@ with gr.Blocks() as demo:
     """
     <h1 style="text-align: center;">Legal Document Drafting</h1>
     """)
-    with gr.Row():
-        with gr.Column():
+    with gr.Tab("Writing Assist"):
+#        with gr.Column():
             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.Column():
+#        with gr.Column():
+    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")