fix: api key not clean after page refreshed

app.py

@@ -1,21 +1,11 @@
 from openai import OpenAI
 import gradio as gr
 
-api_key = ""
-client = None
 
-
-def initialize_openai(api_key):
-    print("Initializing OpenAI client for key: ", api_key[:10])
-    global client
-    client = OpenAI(api_key=api_key)
-    print("OpenAI API Initialized")
-
-
-def generate_response(message, history):
-    if not client:
+def generate_response(message, history, api_key_input):
+    if not api_key_input:
         return "Please enter an API key to initialize the OpenAI API."
-
+    client = OpenAI(api_key=api_key_input)
     formatted_history = []
     for msg in history:
         formatted_history.append({"role": msg["role"], "content": msg["content"]})
@@ -25,7 +15,7 @@ def generate_response(message, history):
     response = client.chat.completions.create(
         model="gpt-3.5-turbo", messages=formatted_history, temperature=1.0
     )
-
+    client.close()
     return response.choices[0].message.content
 
 
@@ -33,9 +23,13 @@ with gr.Blocks() as demo:
     api_key_input = gr.Textbox(
         label="Enter API Key", type="password", placeholder="Enter your OpenAI API key"
     )
-    api_key_input.change(initialize_openai, inputs=api_key_input, outputs=None)
     chatbot = gr.Chatbot(height=300, type="messages")
-    gr.ChatInterface(fn=generate_response, type="messages", chatbot=chatbot)
+    gr.ChatInterface(
+        fn=generate_response,
+        type="messages",
+        additional_inputs=api_key_input,
+        chatbot=chatbot,
+    )
 
 
 demo.launch()