Update app.py

app.py

@@ -22,7 +22,7 @@ import accelerate
 
 def load_doc(list_file_path, chunk_size, chunk_overlap):
     # Processing for one document only
-    # loader = PyPDFLoader(file_path)
+    # loader = Py PDFLoader(file_path)
     # pages = loader.load()
     loaders = [PyPDFLoader(x) for x in list_file_path]
     pages = []
@@ -167,21 +167,41 @@ def demo():
         qa_chain = gr.State()
         collection_name = gr.State()
 
-    with gr.Tab("Chat"):
-        chatbot = gr.Chatbot(height=300)
-        with gr.Accordion("Advanced - Document references", open=False):
-            with gr.Row():
-                doc_source1 = gr.Textbox(label="Reference 1", lines=2, container=True, scale=20)
-                source1_page = gr.Number(label="Page", scale=1)
-            with gr.Row():
-                doc_source2 = gr.Textbox(label="Reference 2", lines=2, container=True, scale=20)
-                source2_page = gr.Number(label="Page", scale=1)
-            with gr.Row():
-                doc_source3 = gr.Textbox(label="Reference 3", lines=2, container=True, scale=20)
-                source3_page = gr.Number(label="Page", scale=1)
+        gr.Markdown("""RAG USING MIXTRAL""")
+
+        with gr.Row():
+            chatbot = gr.Chatbot(height=300)
+            with gr.Accordion("Advanced - Document references", open=False):
+                with gr.Row():
+                    doc_source1 = gr.Textbox(label="Reference 1", lines=2, container=True, scale=20)
+                    source1_page = gr.Number(label="Page", scale=1)
+                with gr.Row():
+                    doc_source2 = gr.Textbox(label="Reference 2", lines=2, container=True, scale=20)
+                    source2_page = gr.Number(label="Page", scale=1)
+                with gr.Row():
+                    doc_source3 = gr.Textbox(label="Reference 3", lines=2, container=True, scale=20)
+                    source3_page = gr.Number(label="Page", scale=1)
             with gr.Row():
                 msg = gr.Textbox(placeholder="Type message", container=True)
             with gr.Row():
                 submit_btn = gr.Button("Submit")
                 clear_btn = gr.ClearButton([msg, chatbot])
+
+        # Chatbot events
+        submit_btn.click(conversation, \
+            inputs=[qa_chain, msg, chatbot], \
+            outputs=[qa_chain, msg, chatbot, doc_source1, source1_page, doc_source2, source2_page, doc_source3, source3_page], \
+            queue=False)
+        clear_btn.click(lambda:[None,"",0,"",0,"",0], \
+            inputs=None, \
+            outputs=[chatbot, doc_source1, source1_page, doc_source2, source2_page, doc_source3, source3_page], \
+            queue=False)
+    demo.queue().launch(debug=True)
+
+    # Replace the placeholders with your actual functions
+    def conversation(*args, **kwargs):
+        pass
+
+
+