Update app.py

app.py

@@ -6,6 +6,7 @@ import tensorflow_hub as hub
 import openai
 import gradio as gr
 import os
+import dotenv
 from sklearn.neighbors import NearestNeighbors
 
 
@@ -147,8 +148,7 @@ def generate_answer(question, openAI_key):
 
 
 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'
+    openAI_key = os.environ.get('OPENAI_KEY')
     if url.strip() == '' and file == None:
         return '[ERROR]: Both URL and PDF is empty. Provide atleast one.'
 
@@ -192,7 +192,6 @@ with gr.Blocks() as iface:
     with gr.Row():
 
         with gr.Group():
-            openAI_key = gr.Textbox(label='Enter your OpenAI API key here', placeholder='sk-')
             url = gr.Textbox(label='Enter PDF URL here', placeholder='https://docs.pdf')
             gr.Markdown(
                 "<center><h4>----------------------------------------------------------------------------------------------------------------------------------------------------<h4></center>")
@@ -207,7 +206,8 @@ with gr.Blocks() as iface:
                                     lines=5, placeholder='Your answer here...')
 
         btn.click(question_answer, inputs=[
-                  url, file, question, openAI_key], outputs=[answer])
+                  url, file, question], outputs=[answer])
 # openai.api_key = os.getenv('Your_Key_Here')
+dotenv.load_dotenv()
 iface.launch()
-# iface.launch(share=True)
+# iface.launch(share=True)