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,41 @@ 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!'
 
+def question_answer(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']
+        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":
+            answer = generate_answer_text_davinci_003(question, openAI_key)
+        else:
+            answer = generate_answer(question, openAI_key, model)
+        return [{USER_NAME: question, BOT_NAME: answer}]
+    except openai.error.InvalidRequestError as e:
+        return [{USER_NAME: question, '[ERROR]': 'Either you do not have access to GPT4 or you have exhausted your quota!'}]
+
+# define gr.Chatbox for answer
+answer = gr.Chatbox(label='The answer to your question is :', placeholder='Type your question and press enter')
+
+# define gr.Chatbox for question
+question = gr.Chatbox(placeholder='Enter your question here', label='Enter your question here', message_pairs=[[USER_NAME, '']])
 
 
 def generate_text_text_davinci_003(openAI_key,prompt, engine="text-davinci-003"):
@@ -258,7 +293,9 @@ with gr.Blocks() as demo:
 
 
         #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])
         btn.click(question_answer, inputs=[url, file, question, openAI_key, model], outputs=[answer])
+      
 #openai.api_key = os.getenv('Your_Key_Here') 
 demo.launch()