Reupload changed app.py

app.py

@@ -23,7 +23,6 @@ import time
 import shutil
 
 
-
 #env_path = Path('.') / 'environ_vars.env'
 #load_dotenv(dotenv_path=env_path)
 # Print variable FOO to test if this worked
@@ -32,8 +31,6 @@ import shutil
 openai.api_key  = os.environ['OPENAI_API_KEY']
 hf_token  = os.environ['HUGGINGFACEHUB_API_TOKEN']
 
-def loading_pdf_message():
-    return "Pdf(s) loading..."
 
 def augment_llm_with_docs(files):
     from langchain.document_loaders import PyPDFLoader
@@ -116,7 +113,6 @@ def augment_llm_with_docs(files):
         memory=memory
     )
 
-    return "Pdf(s) loaded"
 
 def respond(message, chat_history):
     result = qa({"question": message})
@@ -126,23 +122,15 @@ def respond(message, chat_history):
     return "", chat_history
 
 with gr.Blocks() as demo:
-    with gr.Column():
-        upload_pdfs = gr.File(file_types=[".pdf"], file_count="multiple")
-        with gr.Row():
-            status_vec_db = gr.Textbox(label="DB status", placeholder="This will show status of pdf database", interactive=False)
-            load_pdfs = gr.Button("Load pdf(s) to database")
-
-
+    upload_pdfs = gr.File(file_types=[".pdf"], file_count="multiple")
     chatbot = gr.Chatbot()
-    prompt_msg = gr.Textbox(label="Prompt")
-    submit_btn = gr.Button("Submit")
-    clear = gr.ClearButton([prompt_msg, chatbot])
-
-    #upload_pdfs.upload(augment_llm_with_docs, inputs=upload_pdfs)
-    load_pdfs.click(loading_pdf_message, None, status_vec_db)    
-    load_pdfs.click(augment_llm_with_docs, inputs=upload_pdfs, outputs=status_vec_db)
-    submit_btn.click(respond, inputs=[prompt_msg, chatbot], outputs=[prompt_msg, chatbot])
-    prompt_msg.submit(respond, inputs=[prompt_msg, chatbot], outputs=[prompt_msg, chatbot])
+    msg = gr.Textbox(label="Prompt")
+    btn = gr.Button("Submit")
+    clear = gr.ClearButton([msg, chatbot, upload_pdfs])
+
+    upload_pdfs.upload(augment_llm_with_docs, inputs=upload_pdfs)
+    btn.click(respond, inputs=[msg, chatbot], outputs=[msg, chatbot])
+    msg.submit(respond, inputs=[msg, chatbot], outputs=[msg, chatbot])
     
     gr.Markdown("## Input Pdf File Examples")
     gr.Examples(examples=["docs/finance/Week_1_Intro.pdf",
@@ -157,4 +145,4 @@ with gr.Blocks() as demo:
     first ask these questions, then dive deeper into the topics such as asking to know more about the time
     value of money.""")
 
-demo.launch()
+demo.launch()