update stop api button for save cost from openai api

app.py

@@ -4,7 +4,7 @@ from queue import Empty, Queue
 from threading import Thread
 from bot.web_scrapping.crawler_and_indexer import content_crawler_and_index
 from bot.utils.callbacks import QueueCallback
-from bot.utils.constanst import set_api_key
+from bot.utils.constanst import set_api_key, stop_api_key
 from bot.utils.show_log import logger
 from bot.web_scrapping.default import *
 from langchain.chat_models import ChatOpenAI
@@ -13,8 +13,8 @@ from langchain.schema import AIMessage, BaseMessage, HumanMessage, SystemMessage
 
 
 def apply_api_key(api_key):
-    set_api_key(api_key=api_key)
-    return 'Successfully openai api key'
+    api_key = set_api_key(api_key=api_key)
+    return f'Successfully set {api_key}'
 
 
 human_message_prompt_template = HumanMessagePromptTemplate.from_template("{text}")
@@ -166,15 +166,23 @@ def main():
                         [system_prompt, model_name, temperature],
                         [chat, message, chatbot, messages],
                     )
-            with gr.Column():
-                status = gr.Textbox(label='API KEY STATUS')
-                api_key_set = gr.Textbox(label='Set your OPENAI API KEY')
-                api_key_set_button = gr.Button("Set API key")
-                api_key_set_button.click(
-                    apply_api_key,
-                    [api_key_set],
-                    [status]
-                )
+            with gr.Row():
+                with gr.Column():
+                    status = gr.Textbox(label='API KEY STATUS')
+                    api_key_set = gr.Textbox(label='Set your OPENAI API KEY')
+                    api_key_set_button = gr.Button("Set API key")
+                    api_key_set_button.click(
+                        apply_api_key,
+                        [api_key_set],
+                        [status]
+                    )
+                with gr.Column():
+                    status_2 = gr.Textbox(label='STOP API KEY STATUS')
+                    stop_api_button = gr.Button('Stop API key')
+                    stop_api_button.click(
+                        stop_api_key,
+                        [],
+                        [status_2])
             with gr.Column():
                 system_prompt_area = gr.TextArea(
                     default_system_prompt, lines=4, label="prompt", interactive=True

bot/utils/constanst.py

@@ -5,4 +5,9 @@ API_KEY = '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'
 
 def set_api_key(api_key=API_KEY):
     os.environ['OPENAI_API_KEY'] = api_key
-    return True
+    return api_key
+
+
+def stop_api_key(value):
+    os.environ.pop('OPENAI_API_KEY', None)
+    return 'Stopped API key'