Update app.py

app.py

@@ -12,15 +12,10 @@ I request that you provide only the improved version of the text without any fur
 
 PREFIX_PROMPT = "Please refine the following text in academic English: \n"
 
-print(f'[INFO] Loading API key from env...')
-
-openai.api_key = os.getenv("OPENAI_API_KEY")
-
-if openai.api_key is None:
-    print('[WARN] OPENAI_API_KEY key not found in env')
-
-def submit(x, simple=False):
-    # everytime we restart a new conversation.
+def submit(x, api, simple=False):
+    # reset api key everytime, so it won't save api unsafely...?
+    openai.api_key = api 
+    # restart a new conversation.
     if simple:
         results = openai.ChatCompletion.create(
             model="gpt-3.5-turbo",
@@ -49,8 +44,6 @@ with gr.Blocks() as app:
     # allow setting API key in gui
     with gr.Row():
         api_input = gr.Textbox(label="OPENAI_API_KEY", value=openai.api_key, lines=1)
-        def save_api(x): openai.api_key = x
-        api_input.change(save_api, api_input)
 
     simple_checkbox = gr.Checkbox(value=False, label='simple mode (only send the prefix prompt, 0.00142$ cheaper per query)')
 
@@ -63,6 +56,6 @@ with gr.Blocks() as app:
             cost = gr.Number(label='cost of this query ($)')
 
     text_button = gr.Button("Submit")
-    text_button.click(submit, inputs=[text_input, simple_checkbox], outputs=[text_output, cost])
+    text_button.click(submit, inputs=[text_input, api_input, simple_checkbox], outputs=[text_output, cost])
     
 app.launch()