Update app.py

app.py

@@ -474,14 +474,14 @@ with gr.Blocks(css=STYLES) as demo:
           with gr.Column():
             gr.Markdown("For instant generation and concatenation, use `generate text` button. "
                         "Want to explore alternative choices? use `generate alternatives` button.")
-            
+
             with gr.Accordion("longer guideline", open=False):
-              gr.Markdown("`generate text` button generate continued text and attach it to the end. " 
+              gr.Markdown("`generate text` button generate continued text and attach it to the end. "
                         "on the other hand, `generate alternatives` button generate alternate texts "
                         "up to 3 and let you choose one of them. In both cases, **Write the next paragraph based on "
                         "the following stories so far.** is the default prompt. If you want to try your own designed "
                         "prompt, enter it in the textbox below.")
-              
+
             prompt = gr.Textbox(placeholder="design your own prompt", elem_classes=['no-label'])
 
           with gr.Row():
@@ -511,24 +511,26 @@ with gr.Blocks(css=STYLES) as demo:
               replace_type = gr.Dropdown(choices=['word', 'sentense', 'phrase', 'paragraph'], value='sentense', interactive=True, elem_classes=['no-label'])
 
       with gr.Tab("Chatting"):
-        chatbot = gr.Chatbot([], elem_classes=['no-label-chatbot'])
-        chat_txt = gr.Textbox(placeholder="enter question", elem_classes=['no-label'])
+        with gr.Column(elem_classes=['group-border']):
+          chatbot = gr.Chatbot([], elem_classes=['no-label-chatbot'])
+          chat_txt = gr.Textbox(placeholder="enter question", elem_classes=['no-label'])
 
-        with gr.Row():
-          clear_btn = gr.Button("clear", elem_classes=['control-label-font', 'control-button'])
-          regen_btn = gr.Button("regenerate", elem_classes=['control-label-font', 'control-button'])
+          with gr.Row():
+            clear_btn = gr.Button("clear", elem_classes=['control-label-font', 'control-button'])
+            regen_btn = gr.Button("regenerate", elem_classes=['control-label-font', 'control-button'])
 
       with gr.Tab("Exporting"):
-        gr.Markdown("generate cover art with [`nota-ai/bk-sdm-small`](https://huggingface.co/nota-ai/bk-sdm-small) model. "
-                    "design your own prompt in the textbox below, or just hit 'generate prompt for cover art` button. LLaMA2 "
-                    "model will suggest a prompt for you based on your story.")
-          
-        cover_art = gr.Image(interactive=False, elem_classes=['no-label-image'])
-        gen_cover_art_prompt = gr.Textbox(lines=5, max_lines=5, elem_classes=['no-label'])
-        # toggle between "generate prompt for cover art" and "generate cover art"
-        gen_cover_art_btn = gr.Button("generate prompt for cover art", elem_classes=['control-label-font', 'control-button'])
-        gen_pdf_btn = gr.Button("export as PDF", elem_classes=['control-label-font', 'control-button'])
-        pdf_file = gr.File(visible=False)
+        with gr.Column(elem_classes=['group-border']):
+          gr.Markdown("generate cover art with [`nota-ai/bk-sdm-base`](https://huggingface.co/nota-ai/bk-sdm-base) model. "
+                      "design your own prompt in the textbox below, or just hit 'generate prompt for cover art` button. LLaMA2 "
+                      "model will suggest a prompt for you based on your story.")
+
+          cover_art = gr.Image(interactive=False, elem_classes=['no-label-image'])
+          gen_cover_art_prompt = gr.Textbox(lines=5, max_lines=5, elem_classes=['no-label'])
+          # toggle between "generate prompt for cover art" and "generate cover art"
+          gen_cover_art_btn = gr.Button("generate prompt for cover art", elem_classes=['control-label-font', 'control-button'])
+          gen_pdf_btn = gr.Button("export as PDF", elem_classes=['control-label-font', 'control-button'])
+          pdf_file = gr.File(visible=False)
 
   progress_bar = gr.Textbox(elem_classes=['no-label'])