Update app.py

app.py

@@ -99,7 +99,8 @@ def load_recommender(path, start_page=1):
     return 'Corpus Loaded.'
 
 
-def generate_text(prompt, engine="text-davinci-003"):
+def generate_text(prompt, engine="text-davinci-003",openAI_key):
+    openai.api_key = openAI_key
     completions = openai.Completion.create(
         engine=engine,
         prompt=prompt,
@@ -112,7 +113,7 @@ def generate_text(prompt, engine="text-davinci-003"):
     return message
 
 
-def generate_answer(question):
+def generate_answer(question,openAI_key):
     topn_chunks = recommender(question)
     prompt = ""
     prompt += 'search results:\n\n'
@@ -129,11 +130,11 @@ def generate_answer(question):
               "answer should be short and concise.\n\nQuery: {question}\nAnswer: "
     
     prompt += f"Query: {question}\nAnswer:"
-    answer = generate_text(prompt)
+    answer = generate_text(prompt,openAI_key)
     return answer
 
 
-def question_answer(url, file, question):
+def question_answer(url, file, question,openAI_key):
     if url.strip() == '' and file == None:
         return '[ERROR]: Both URL and PDF is empty. Provide atleast one.'
     
@@ -155,7 +156,7 @@ def question_answer(url, file, question):
     if question.strip() == '':
         return '[ERROR]: Question field is empty'
 
-    return generate_answer(question)
+    return generate_answer(question,openAI_key)
 
 
 recommender = SemanticSearch()
@@ -175,6 +176,7 @@ with gr.Blocks() as demo:
     with gr.Row():
         
         with gr.Group():
+            api_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'])
@@ -185,7 +187,7 @@ with gr.Blocks() as demo:
         with gr.Group():
             answer = gr.Textbox(label='The answer to your question is :')
 
-        btn.click(question_answer, inputs=[url, file, question], outputs=[answer])
+        btn.click(question_answer, inputs=[url, file, question,openAI_key], outputs=[answer])
 openai.api_key = os.getenv('Your_Key_Here') 
 demo.launch()