Update app.py

app.py

@@ -142,35 +142,34 @@ def generate_answer(question, openAI_key,model):
     return answer
 
 def question_answer(url, file, question, openAI_key, model):
-    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'
+    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":
+            return generate_answer_text_davinci_003(question, openAI_key)
+        else:
+            return generate_answer(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!'
 
-    if model == "text-davinci-003":
-        return generate_answer_text_davinci_003(question, openAI_key)
-    else:
-        return generate_answer(question, openAI_key, model)
 
 
 def generate_text_text_davinci_003(openAI_key,prompt, engine="text-davinci-003"):