add section titles and rescale buttons

app.py

@@ -208,17 +208,18 @@ def demo():
 
         gr.Markdown(
             """
-            <center><h2>PDF-based chatbot</center></h2>
-            <h3>Ask any questions about your PDF documents</h3>
-            """
-        )
-        gr.Markdown(
-            """<b>Note:</b> This AI assistant, using Langchain and open-source LLMs, performs retrieval-augmented generation (RAG) from your PDF documents. \
-            This chatbot takes past questions into account when generating answers (via conversational memory), and includes document references for clarity purposes.<br>
+            <center><h1>Chat with your PDF!</center></h1>
+            <center><h3>Ask any questions about your PDF documents</h3><center>
             """
         )
+        # gr.Markdown(
+        #    """<b>Note:</b> This AI assistant, using Langchain and open-source LLMs, performs retrieval-augmented generation (RAG) from your PDF documents. \
+        #    This chatbot takes past questions into account when generating answers (via conversational memory), and includes document references for clarity purposes.<br>
+        #    """
+        # )
 
         with gr.Tab("Chatbot configuration"):
+            gr.Markdown("1. Upload the PDF(s)")
             with gr.Row():
                 document = gr.Files(
                     height=100,
@@ -229,6 +230,7 @@ def demo():
                 )
                 # upload_btn = gr.UploadButton("Loading document...", height=100, file_count="multiple", file_types=["pdf"], scale=1)
 
+            gr.Markdown("2. Configure the vector database")
             with gr.Row():
                 with gr.Row():
                     db_btn = gr.Radio(
@@ -261,13 +263,14 @@ def demo():
                             info="Chunk overlap",
                             interactive=True,
                         )
-                with gr.Row():
-                    db_btn = gr.Button("Generate vector database")
+            with gr.Row():
+                db_btn = gr.Button("Generate vector database", size="sm")
             with gr.Row():
                 db_progress = gr.Textbox(
                     label="Vector database initialization", value="0% Configure the DB"
                 )
 
+            gr.Markdown("3. Configure the LLM model")
             with gr.Row():
                 with gr.Row():
                     llm_btn = gr.Radio(
@@ -308,11 +311,13 @@ def demo():
                             info="Model top-k samples",
                             interactive=True,
                         )
-                with gr.Row():
-                    qachain_btn = gr.Button("Initialize Question Answering chain")
+            with gr.Row():
+                qachain_btn = gr.Button(
+                    "Initialize Question Answering chain", size="sm"
+                )
             with gr.Row():
                 llm_progress = gr.Textbox(
-                    value="None", label="0% Configure the QA chain"
+                    value="QA chain initialization", label="0% Configure the QA chain"
                 )
 
         with gr.Tab("Chatbot"):