Update app.py

app.py

@@ -27,11 +27,9 @@ def pdf_changes(pdf_doc):
 
 def add_text(history, text):
     history = history + [(text, None)]
-    print(history)
     return history, ""
 
 def bot(history):
-    print(history[-1][0])
     response = infer(history[-1][0])
     history[-1][1] = response['result']
     return history
@@ -43,21 +41,37 @@ def infer(question):
 
     return result
 
+css="""
+#col-container {max-width: 700px; margin-left: auto; margin-right: auto;}
+"""
+
+title = """
+<div style="text-align: center;max-width: 700px;>
+    <h1>Chat with PDF</h1>
+    <p style="text-align: center;">Upload a .PDF from your computer, click the "Load PDF to LangChain" button, <br />
+    when everything is ready, you can start asking questions about the pdf ;)</p>
+</div>
+"""
+
+
 with gr.Blocks() as demo:
-    with gr.Column():
-        pdf_doc = gr.File(label="Load a pdf", file_types=['.pdf'], type="file")
-        langchain_status = gr.Textbox()
-        load_pdf = gr.Button("Load pdf to langchain")
-        
-        chatbot = gr.Chatbot([], elem_id="chatbot").style(height=350)
-        question = gr.Textbox(label="Question")
+    with gr.Column(elem_id="col-container"):
+        gr.HTML(title)
+        with gr.Row():
+            with gr.Column():
+                pdf_doc = gr.File(label="Load a pdf", file_types=['.pdf'], type="file")
+                langchain_status = gr.Textbox()
+                load_pdf = gr.Button("Load pdf to langchain")
+            with gr.Column():
+                chatbot = gr.Chatbot([], elem_id="chatbot").style(height=350)
+                with gr.Row():
+                    question = gr.Textbox(label="Question")
+                    clear = gr.Button("Clear")
         
-    
     load_pdf.click(pdf_changes, pdf_doc, langchain_status, queue=False)
-    
-
     question.submit(add_text, [chatbot, question], [chatbot, question]).then(
         bot, chatbot, chatbot
     )
+    clear.click(lambda: None, None, chatbot, queue=False)
 
 demo.launch()