Update app.py

app.py

@@ -8,7 +8,7 @@ import paddlehub as hub
 import random
 from encoder import get_encoder
 
-openai.api_key = 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
+openai.api_key = os.getenv("OPENAI_API_KEY")
 
 from utils import get_tmt_client, getTextTrans_tmt
 tmt_client = get_tmt_client()
@@ -351,6 +351,10 @@ total_tokens = 4096
 max_output_tokens = 1024
 max_input_tokens = total_tokens - max_output_tokens
 
+def set_openai_api_key(api_key):
+    if api_key and api_key.startswith("sk-") and len(api_key) > 50:
+        openai.api_key = api_key
+
 def get_response_from_openai(input, chat_history, model_radio):
     error_1 = 'You exceeded your current quota, please check your plan and billing details.'
     def openai_create(input_list, model_radio):
@@ -418,6 +422,7 @@ def get_response_from_openai(input, chat_history, model_radio):
             return ret, response.usage                           
         except Exception as e:
             logger.info(f"openai_create_error__{e}")
+            ret = f"Openai said: {e} Perhaps enter your OpenAI API key."
             return ret, {"completion_tokens": -1, "prompt_tokens": -1, "total_tokens": -1}
     
     # logger.info(f'chat_history = {chat_history}')
@@ -475,6 +480,8 @@ with gr.Blocks(title='Talk to chatGPT') as demo:
             chat_radio = gr.Radio(["Talk to chatGPT", "Text to Image"], elem_id="chat_radio",value="Talk to chatGPT", show_label=False, visible=True)
             model_radio = gr.Radio(["GPT-3.0", "GPT-3.5"], elem_id="model_radio", value="GPT-3.5", 
                                 label='GPT model: ', show_label=True,interactive=True, visible=True) 
+            openai_api_key_textbox = gr.Textbox(placeholder="Paste your OpenAI API key (sk-...) and hit Enter",
+                                    show_label=False, lines=1, type='password')
         with gr.Row(elem_id="btns_row"):
             with gr.Column(id="submit_col"):
                 submit_btn = gr.Button(value = "submit",elem_id="submit-btn").style(
@@ -495,6 +502,15 @@ with gr.Blocks(title='Talk to chatGPT') as demo:
         with gr.Row(elem_id='tab_img', visible=False).style(height=5):
            tab_img = gr.TabbedInterface(tab_actions, tab_titles)  
 
-     
+        openai_api_key_textbox.change(set_openai_api_key,
+                                      inputs=[openai_api_key_textbox],
+                                      outputs=[])
+        openai_api_key_textbox.submit(set_openai_api_key,
+                                      inputs=[openai_api_key_textbox],
+                                      outputs=[])
+        chat_radio.change(fn=chat_radio_change, 
+                        inputs=[chat_radio], 
+                        outputs=[model_radio, openai_api_key_textbox],
+                        ) 
 
 demo.launch(debug = True)