Update app.py

app.py

@@ -29,17 +29,24 @@ from langchain.indexes import VectorstoreIndexCreator
 import tempfile
 from langchain.document_loaders import DirectoryLoader
 
-os.environ["OPENAI_API_KEY"] = os.environ['OpenApi_Key']
+os.environ["OPENAI_API_KEY"] = "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"
 query1=" "
 limit = 0
 def loading_pdf():
     return "Loading..."
 
-    
+def remove_last_slash_and_characters(input_string):
+    last_slash_index = input_string.rfind('/')
+    if last_slash_index != -1:
+        result_string = input_string[:last_slash_index]
+    else:
+        result_string = input_string
+
+    return result_string
 
 def pdf_changes(pdf_doc, prompt):
     
-    loader = DirectoryLoader("/tmp/gradio")
+    loader = DirectoryLoader(remove_last_slash_and_characters(pdf_doc.name))
     #loader = OnlinePDFLoader(pdf_doc.name)
     #loader = PyPDFLoader(pdf_doc.name)
     documents = loader.load()
@@ -63,7 +70,7 @@ def pdf_changes(pdf_doc, prompt):
       )
 
 
-    system_template=prompt+"""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}"""
 
@@ -77,7 +84,7 @@ def pdf_changes(pdf_doc, prompt):
     from langchain.chains import RetrievalQAWithSourcesChain
     global query1
     chain_type_kwargs = {"prompt": prompt2}
-    llm = ChatOpenAI(model_name="gpt-3.5-turbo", temperature=0, max_tokens=512)  # Modify model_name if you have access to GPT-4
+    llm = ChatOpenAI(model_name="gpt-3.5-turbo", temperature=0, max_tokens=512)
     global chain
     chain = RetrievalQAWithSourcesChain.from_chain_type(
         llm=llm,
@@ -100,7 +107,7 @@ def bot(history):
 def infer(question):
     global query1
     global limit
-    openai.api_key = os.environ['OpenApi_Key']
+    openai.api_key = "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"
     prompt_text = question
     if prompt_text:
         query1 = query1 + "\nUser: " + prompt_text + "\nBot: "
@@ -134,7 +141,7 @@ with gr.Blocks(css=css,theme = gr.themes.Soft()) as demo:
         with gr.Row():
           with gr.Column(scale=1):
             #gr.File(file_count="multiple")
-              pdf_doc = gr.File(label="Load a pdf", file_count="multiple")
+              pdf_doc = gr.File(label="Load a pdf", file_count="single")
               prompt = gr.Textbox(label="Behaviour Prompt (optional)", 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():
@@ -145,8 +152,6 @@ with gr.Blocks(css=css,theme = gr.themes.Soft()) as demo:
             with gr.Row():
               question = gr.Textbox(label="Question", placeholder="Type your question and hit Enter ",scale=6,show_label=False)
               submit_btn = gr.Button("Send",scale=1)
-    #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,prompt], outputs=[langchain_status], queue=False)
     question.submit(add_text, [chatbot, question], [chatbot, question]).then(
         bot, chatbot, chatbot