Update app.py

app.py

@@ -168,31 +168,31 @@ def generate_answer(question,openAI_key):
     return answer
 
 
-def question_answer(url, file, question,openAI_key):
-    if openAI_key.strip()=='':
-        return '[ERROR]: Please enter you Open AI Key. Get your key here : https://platform.openai.com/account/api-keys'
+def question_answer(url, file, question, openAI_key, start_page, end_page):
+    if openAI_key.strip() == '':
+        return '[ERROR]: Please enter your 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 atleast one.'
+        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 (eiter URL or PDF).'
+        return '[ERROR]: Both URL and PDF is provided. Please provide only one (either URL or PDF).'
 
     if url.strip() != '':
         glob_url = url
         download_pdf(glob_url, 'corpus.pdf')
-        load_recommender('corpus.pdf')
+        load_recommender('corpus.pdf', start_page=start_page, end_page=end_page)
 
     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)
+        load_recommender(file_name, start_page=start_page, end_page=end_page)
 
     if question.strip() == '':
         return '[ERROR]: Question field is empty'
 
-    return generate_answer(question,openAI_key)
+    return generate_answer(question, openAI_key)
 
 
 recommender = SemanticSearch()
@@ -223,5 +223,25 @@ with gr.Blocks() as demo:
             answer = gr.Textbox(label='The answer to your question is :')
 
         btn.click(question_answer, inputs=[url, file, question,openAI_key], outputs=[answer])
+        with gr.Row():
+    
+        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')
+            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')
+            start_page = gr.Textbox(label='Start on page (optional):', numeric=True)
+            end_page = gr.Textbox(label='End on page (optional):', numeric=True)
+            btn = gr.Button(value='Submit')
+            btn.style(full_width=True)
+    
+        with gr.Group():
+            answer = gr.Textbox(label='The answer to your question is :')
+    
+        btn.click(question_answer, inputs=[url, file, question, openAI_key, start_page, end_page], outputs=[answer])
+    
+    
 #openai.api_key = os.getenv('Your_Key_Here') 
 demo.launch()