Update app.py

app.py

@@ -28,12 +28,12 @@ from langchain.document_loaders import PyPDFLoader
 from langchain.indexes import VectorstoreIndexCreator
 import tempfile
 
-os.environ["OPENAI_API_KEY"] = "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"
+os.environ["OPENAI_API_KEY"] = "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"
 query1=" "
 def loading_pdf():
     return "Loading..."
 
-def pdf_changes(pdf_doc, repo_id):
+def pdf_changes(pdf_doc, prompt):
     
     loader = OnlinePDFLoader(pdf_doc.name)
     documents = loader.load()
@@ -56,7 +56,7 @@ def pdf_changes(pdf_doc, repo_id):
       )
 
 
-    system_template="""You are a helpful chatbot used by the user to chat with pdf documents. Only answer the questions by using information provided in the context provided to you. If there is no relavant context, tell 'Hmm, I'm not sure'.
+    system_template="""You are a helpful chatbot used by the user to chat with pdf documents. Only answer the questions by using information provided in the context provided to you. If there is no relavant context, tell 'Hmm, I'm not sure'."""+prompt+"""
     ----------------
     {summaries}"""
 
@@ -92,7 +92,7 @@ def bot(history):
 
 def infer(question):
     global query1
-    openai.api_key = "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"
+    openai.api_key = "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"
     prompt_text = question
     if prompt_text:
         query1 = query1 + "\nUser: " + prompt_text + "\nBot: "
@@ -112,9 +112,9 @@ css="""
 """
 
 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" button, <br />
+<div style="text-align: center; max-width: 700px;">
+    <h1 style="color: #4545FF;">Chat with PDF</h1>
+    <p style="text-align: center; color: #4545FF;">Upload a .PDF from your computer, click the "Load PDF" button, <br />
     when everything is ready, you can start asking questions about the pdf ;)</p>
 </div>
 """
@@ -122,10 +122,11 @@ title = """
 
 with gr.Blocks(css=css,theme = gr.themes.Soft()) as demo:
     with gr.Column(elem_id="col-container"):
-       # gr.HTML(title)
+        #gr.HTML(title)
         
         with gr.Column():
             pdf_doc = gr.File(label="Load a pdf", file_types=['.pdf'], type="file")
+            prompt = gr.Textbox(label="Behaviour Prompt", placeholder="Reply to all questions as a rap / Reply to all questions in Hindi etc. ")
             #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)
@@ -136,7 +137,7 @@ with gr.Blocks(css=css,theme = gr.themes.Soft()) as demo:
         submit_btn = gr.Button("Send message")
     #load_pdf.click(loading_pdf, None, langchain_status, queue=False)    
     #repo_id.change(pdf_changes, inputs=[pdf_doc], outputs=[langchain_status], queue=False)
-    load_pdf.click(pdf_changes, inputs=[pdf_doc], outputs=[langchain_status], queue=False)
+    load_pdf.click(pdf_changes, inputs=[pdf_doc,prompt], outputs=[langchain_status], queue=False)
     question.submit(add_text, [chatbot, question], [chatbot, question]).then(
         bot, chatbot, chatbot
     )
@@ -144,4 +145,4 @@ with gr.Blocks(css=css,theme = gr.themes.Soft()) as demo:
         bot, chatbot, chatbot
     )
 
-demo.launch()
+demo.launch(debug=True)