Update app.py

app.py

@@ -141,7 +141,7 @@ def generate_answer(question, openAI_key,model):
     answer = generate_text(openAI_key, prompt, model)
     return answer
 
-def question_answer(url, file, question, openAI_key, model):
+def question_answer1(url, file, question, openAI_key, model):
     try:
         if openAI_key.strip()=='':
             return '[ERROR]: Please enter you Open AI Key. Get your key here : https://platform.openai.com/account/api-keys'
@@ -170,6 +170,38 @@ def question_answer(url, file, question, openAI_key, model):
     except openai.error.InvalidRequestError as e:
         return f'[ERROR]: Either you do not have access to GPT4 or you have exhausted your quota!'
 
+# Modifying your code to include chat history feature
+
+def question_answer(chat_history, url, file, question, openAI_key, model):
+    try:
+        if openAI_key.strip()=='':
+            return '[ERROR]: Please enter your OpenAI Key. Get your key here : https://platform.openai.com/account/api-keys'
+        if url.strip() == '' and file == None:
+            return '[ERROR]: Both URL and PDF is empty. Provide at least one.'
+        if url.strip() != '' and file != None:
+            return '[ERROR]: Both URL and PDF is provided. Please provide only one (either URL or PDF).'
+        if model is None or model =='':
+            return '[ERROR]: You have not selected any model. Please choose an LLM model.'
+        if url.strip() != '':
+            glob_url = url
+            download_pdf(glob_url, 'corpus.pdf')
+            load_recommender('corpus.pdf')
+        else:
+            old_file_name = file.name
+            file_name = file.name
+            file_name = file_name[:-12] + file_name[-4:]
+            os.rename(old_file_name, file_name)
+            load_recommender(file_name)
+        if question.strip() == '':
+            return '[ERROR]: Question field is empty'
+        if model == "text-davinci-003":
+            return generate_answer_text_davinci_003(question, openAI_key)
+        else:
+            answer = generate_answer(question, openAI_key, model)
+            chat_history.append([question, answer])
+            return chat_history
+    except openai.error.InvalidRequestError as e:
+        return f'[ERROR]: Either you do not have access to GPT4 or you have exhausted your quota!'
 
 
 def generate_text_text_davinci_003(openAI_key,prompt, engine="text-davinci-003"):
@@ -235,29 +267,29 @@ with gr.Blocks() as demo:
         
         with gr.Group():
             gr.Markdown(f'<p style="text-align:center">Get your Open AI API key <a href="https://platform.openai.com/account/api-keys">here</a></p>')
-            #openAI_key=gr.Textbox(label='Enter your OpenAI API key here')
             openAI_key = gr.Textbox(label='Enter your OpenAI API key here', password=True)
             url = gr.Textbox(label='Enter PDF URL here')
             gr.Markdown("<center><h4>OR<h4></center>")
             file = gr.File(label='Upload your PDF/ Research Paper / Book here', file_types=['.pdf'])
             question = gr.Textbox(label='Enter your question here')
-            #Predefined questions
             gr.Examples(
                 [[q] for q in questions],
                 inputs=[question],
                 label="PRE-DEFINED QUESTIONS: Click on a question to auto-fill the input box, then press Enter!",
             )
             model = gr.Radio(['gpt-3.5-turbo', 'gpt-3.5-turbo-16k', 'gpt-3.5-turbo-0613', 'gpt-3.5-turbo-16k-0613', 'text-davinci-003','gpt-4','gpt-4-32k'], label='Select Model', default='gpt-3.5-turbo')
-            #model = gr.Dropdown(choices=['gpt-3.5-turbo', 'gpt-3.5-turbo-16k', 'gpt-3.5-turbo-0613', 'gpt-3.5-turbo-16k-0613', 'text-davinci-003'], label='Select Large Language Model', default='gpt-3.5-turbo')
             btn = gr.Button(value='Submit')
             btn.style(full_width=True)
 
         with gr.Group():
-            #answer = gr.Textbox(label='The answer to your question is :')
-            answer = gr.Textbox(label='The answer to your question is :').style(show_copy_button=True)
+            chatbot = gr.Chatbot(placeholder="Chat History", label="Chat History", lines=20)
 
+    question.submit(
+        question_answer,
+        [chatbot, url, file, question, openAI_key, model],
+        outputs=[chatbot],
+        show_progress=False,
+    )
 
-        #btn.click(question_answer, inputs=[url, file, question,openAI_key], outputs=[answer])
-        btn.click(question_answer, inputs=[url, file, question, openAI_key, model], outputs=[answer])
 #openai.api_key = os.getenv('Your_Key_Here') 
 demo.launch()