adding button for no one can use api key and have to set default

README.md

@@ -32,18 +32,19 @@ This project use Gradio to build User Interface, please use:
  python app.py
 ```
 Enjoy project with step by step follow this: 
-
+First step, you need to set your openai api key right here:
+!<img src='image/set_api_key.png'>
 Bot Learning with URL 
-![Screenshot 2023-11-13 at 19.56.40.png](..%2F..%2F..%2F..%2Fvar%2Ffolders%2Fzc%2Fcsmhsgrd0bz3bbkycljwdk2c0000gn%2FT%2FTemporaryItems%2FNSIRD_screencaptureui_zJFx9p%2FScreenshot%202023-11-13%20at%2019.56.40.png)
+!<img src='image/bot_learning.png'>
 
 Please input URL and your file format and then press Training button, when the training process is finished, it's will notice you Training Completed 
 
 After that, you can starting chat with your custom bot about the topic in your URL
-![Screenshot 2023-11-13 at 20.00.10.png](..%2F..%2F..%2F..%2Fvar%2Ffolders%2Fzc%2Fcsmhsgrd0bz3bbkycljwdk2c0000gn%2FT%2FTemporaryItems%2FNSIRD_screencaptureui_xDIQGv%2FScreenshot%202023-11-13%20at%2020.00.10.png)
+!<img src='image/chatbot.png'>
 
 The vector index storage by this structure: 
 
-![Screenshot 2023-11-13 at 20.03.04.png](..%2F..%2F..%2F..%2Fvar%2Ffolders%2Fzc%2Fcsmhsgrd0bz3bbkycljwdk2c0000gn%2FT%2FTemporaryItems%2FNSIRD_screencaptureui_ZTP7r9%2FScreenshot%202023-11-13%20at%2020.03.04.png)
+!<img src='image/index_storage.png'>
 
 I've already adding the data from this page: https://www.presight.io/privacy-policy.html
 You can check and chat with bot for Information in that or You can training more knowledge for it

app.py

@@ -11,7 +11,12 @@ from langchain.chat_models import ChatOpenAI
 from langchain.prompts import HumanMessagePromptTemplate
 from langchain.schema import AIMessage, BaseMessage, HumanMessage, SystemMessage
 
-set_api_key(api_key='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')
+
+def apply_api_key(api_key):
+    set_api_key(api_key=api_key)
+    return 'Successfully openai api key'
+
+
 human_message_prompt_template = HumanMessagePromptTemplate.from_template("{text}")
 
 
@@ -161,7 +166,15 @@ 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.Column():
                 system_prompt_area = gr.TextArea(
                     default_system_prompt, lines=4, label="prompt", interactive=True
@@ -184,6 +197,4 @@ def main():
 if __name__ == '__main__':
     demo = main()
     demo.queue()
-    demo.launch()
-
-
+    demo.launch(share=True)