Update app.py

app.py

@@ -65,18 +65,6 @@ css = """
   margin-bottom: 10px;
 }
 
-.repo-id {
-  margin-bottom: 10px;
-}
-
-.langchain-status {
-  margin-bottom: 10px;
-}
-
-.load-pdf {
-  margin-bottom: 10px;
-}
-
 .chatbot {
   height: 350px;
   border: 1px solid #ccc;
@@ -109,61 +97,29 @@ title = """
 </div>
 """
 
-
-def pdf_changes(inputs, outputs):
-  """Updates the text of the langchain status textbox."""
-  pdf_doc = inputs["pdf_doc"]
-  repo_id = inputs["repo_id"]
-
-  langchain_status.update_text(f"Loading PDF to langchain...")
-
-  # Load the PDF to langchain
-  langchain.load_pdf(pdf_doc, repo_id)
-
-  langchain_status.update_text(f"PDF loaded to langchain.")
-
-
-def add_text(inputs, outputs):
-  """Adds text to the chatbot."""
-  chatbot = inputs["chatbot"]
-  question = inputs["question"]
-
-  # Add the user question to the chatbot
-  chatbot.add_user_message(question)
-
-  # Get the response from the langchain chatbot
-  response = langchain.get_response(question)
-
-  # Add the bot response to the chatbot
-  chatbot.add_bot_message(response)
-
-
-# Create the chatbot
-chatbot = gr.Chatbot([], elem_id="chatbot").style(height=350)
-
-# Create the langchain object
-langchain = gr.Langchain()
-
-# Create the UI
 with gr.Blocks(css=css) as demo:
-  with gr.Column(elem_id="col-container"):
-    gr.HTML(title)
-
-    with gr.Column():
-      pdf_doc = gr.File(label="Load a pdf", file_types=['.pdf'], type="file")
-      repo_id = gr.Dropdown(label="LLM", choices=["google/flan-ul2", "OpenAssistant/oasst-sft-1-pythia-12b", "bigscience/bloomz"], value="google/flan-ul2")
-      with gr.Row():
-        langchain_status = gr.Textbox(label="Status", placeholder="", interactive=False)
-        load_pdf = gr.Button("Load pdf to langchain")
-        chatbot.style(height=350)
-
-    question = gr.Textbox(label="Question", placeholder="Type your Question and hit Enter ")
-    submit_btn = gr.Button("Send message")
-
-  # Respond to events
-  load_pdf.click(pdf_changes, inputs=[pdf_doc, repo_id], outputs=[langchain_status], queue=False)
-  question.submit(add_text, [chatbot, question], [chatbot, question]).then(bot, chatbot, chatbot)
-  submit_btn.click(add_text, [chatbot, question], [chatbot, question]).then(bot, chatbot, chatbot)
+    with gr.Column(elem_id="col-container"):
+        gr.HTML(title)
+        
+        with gr.Column():
+            pdf_doc = gr.File(label="Load a pdf", file_types=['.pdf'], type="file")
+            repo_id = gr.Dropdown(label="LLM", choices=["google/flan-ul2", "OpenAssistant/oasst-sft-1-pythia-12b", "bigscience/bloomz", "TheBloke/Llama-2-7b-Chat-GGUF"], value="google/flan-ul2")
+            with gr.Row():
+                langchain_status = gr.Textbox(label="Status", placeholder="", interactive=False)
+                load_pdf = gr.Button("Load pdf to langchain")
+        
+        chatbot = gr.Chatbot([], elem_id="chatbot").style(height=350)
+        question = gr.Textbox(label="Question", placeholder="Type your question and hit Enter ")
+        submit_btn = gr.Button("Send message")
+    #load_pdf.click(loading_pdf, None, langchain_status, queue=False)    
+    repo_id.change(pdf_changes, inputs=[pdf_doc, repo_id], outputs=[langchain_status], queue=False)
+    load_pdf.click(pdf_changes, inputs=[pdf_doc, repo_id], outputs=[langchain_status], queue=False)
+    question.submit(add_text, [chatbot, question], [chatbot, question]).then(
+        bot, chatbot, chatbot
+    )
+    submit_btn.click(add_text, [chatbot, question], [chatbot, question]).then(
+        bot, chatbot, chatbot
+    )
 
 # Start the demo
 demo.show(True)