Update app.py

app.py

@@ -9,10 +9,8 @@ from langchain.vectorstores import Chroma # for the vectorization part
 from langchain.chains import RetrievalQA # for conversing with chatGPT
 from langchain.chat_models import ChatOpenAI # the LLM model we'll use (ChatGPT)
 
-def loading_pdf():
-    return "Loading..."
 
-def pdf_changes(pdf_doc, open_ai_key):
+def load_pdf(pdf_doc, open_ai_key):
     if openai_key is not None:
         os.environ['OPENAI_API_KEY'] = open_ai_key
         #Load the pdf file
@@ -36,40 +34,33 @@ def pdf_changes(pdf_doc, open_ai_key):
 
 def answer_query(query):
     question = query
-    return pdf_qa.run(question)
+    return chain.run(question)
+
     
-css="""
-#col-container {max-width: 700px; margin-left: auto; margin-right: auto;}
-"""
+html = """
+<div style="text-align:center; max width: 700px;">
+    <h1>ChatPDF</h1>
+    <p> Upload a PDF File, then click on Load PDF File <br>
+    Once the document has been loaded you can begin chatting with the PDF =)
+</div>"""
+css = """container{max-width:700px; margin-left:auto; margin-right:auto,padding:20px}"""
+with gr.Blocks(css=css,theme=gr.themes.Monochrome()) as demo:
+    gr.HTML(html)
+    with gr.Column():
+        gr.Markdown('ChatPDF')
+        pdf_doc = gr.File(label="Load a pdf",file_types=['.pdf','.docx'],type='file')
+        with gr.Row():
+            load_pdf = gr.Button('Load pdf file')
+            status = gr.Textbox(label="Status",placeholder='',interactive=False)
 
-title = """
-<div style="text-align: center;max-width: 700px;">
-    <h1>Chatbot for PDFs - GPT-4</h1>
-    <p style="text-align: center;">Upload a .PDF, click the "Load PDF to LangChain" button, <br />
-    Wait for the Status to show Ready, start typing your questions. <br />
-    The app is built on GPT-4</p>
-</div>
-"""
 
+        with gr.Row():
+            input = gr.Textbox(label="type in your question")
+            output = gr.Textbox(label="output")
+        submit_query = gr.Button("submit")
 
-with gr.Blocks(css=css) as demo:
-    with gr.Column(elem_id="col-container"):
-        gr.HTML(title)
-        
-        with gr.Column():
-            openai_key = gr.Textbox(label="Your GPT-4 OpenAI API key", type="password")
-            pdf_doc = gr.File(label="Load a pdf", file_types=['.pdf'], type="file")
-            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)    
-    load_pdf.click(pdf_changes, inputs=[pdf_doc, openai_key], outputs=[langchain_status], queue=False)
-    submit_btn.click(answer_query,question,chatbot)
-    
+        load_pdf.click(load_pdf,inputs=pdf_doc,outputs=status)
+
+        submit_query.click(answer_query,input,output)
 
 demo.launch()