Upload app.py

app.py

@@ -254,7 +254,7 @@ def set_openai_api_key(api_key):
     If no api_key, then None is returned.
     """
     if api_key and api_key.startswith("sk-") and len(api_key) > 50:
-        os.environ["OPENAI_API_KEY"] = "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" # api_key
+        os.environ["OPENAI_API_KEY"] = api_key
         print("\n\n ++++++++++++++ Setting OpenAI API key ++++++++++++++ \n\n")
         print(str(datetime.datetime.now()) + ": Before OpenAI, OPENAI_API_KEY length: " + str(
             len(os.environ["OPENAI_API_KEY"])))
@@ -594,15 +594,15 @@ with gr.Blocks(css=".gradio-container {background-color: lightgray}") as block:
     use_embeddings_state = gr.State(False)
 
     with gr.Tab("Chat"):
-      #  with gr.Row():
+        with gr.Row():
             
           #  with gr.Column():
              #   gr.HTML(
               #      """<b><center>GPT + WolframAlpha + Whisper</center></b>
                #     <p><center>New feature: <b>Embeddings</b></center></p>""")
 
-            #openai_api_key_textbox = gr.Textbox(placeholder="Paste your OpenAI API key (sk-...)",
-             #                                   show_label=False, lines=1, type='password')
+            openai_api_key_textbox = gr.Textbox(placeholder="Paste your OpenAI API key (sk-...)",
+                                                show_label=False, lines=1, type='password')
 
         with gr.Row():
             with gr.Column(scale=1, min_width=TALKING_HEAD_WIDTH, visible=True):
@@ -845,7 +845,7 @@ with gr.Blocks(css=".gradio-container {background-color: lightgray}") as block:
         Powered by <a href='https://github.com/hwchase17/langchain'>LangChain 🦜️🔗</a>
         </center>""")
 
-    message.submit(chat, inputs=[ message, history_state, chain_state, trace_chain_state,
+    message.submit(chat, inputs=[openai_api_key_textbox, message, history_state, chain_state, trace_chain_state,
                                  speak_text_state, talking_head_state, monologue_state,
                                  express_chain_state, num_words_state, formality_state,
                                  anticipation_level_state, joy_level_state, trust_level_state, fear_level_state,
@@ -855,7 +855,7 @@ with gr.Blocks(css=".gradio-container {background-color: lightgray}") as block:
                    outputs=[chatbot, history_state, video_html, my_file, audio_html, tmp_aud_file, message])
     # outputs=[chatbot, history_state, audio_html, tmp_aud_file, message])
 
-    submit.click(chat, inputs=[ message, history_state, chain_state, trace_chain_state,
+    submit.click(chat, inputs=[openai_api_key_textbox, message, history_state, chain_state, trace_chain_state,
                                speak_text_state, talking_head_state, monologue_state,
                                express_chain_state, num_words_state, formality_state,
                                anticipation_level_state, joy_level_state, trust_level_state, fear_level_state,
@@ -864,10 +864,10 @@ with gr.Blocks(css=".gradio-container {background-color: lightgray}") as block:
                                qa_chain_state, docsearch_state, use_embeddings_state],
                  outputs=[chatbot, history_state, video_html, my_file, audio_html, tmp_aud_file, message])
     # outputs=[chatbot, history_state, audio_html, tmp_aud_file, message])
-    """
+
     openai_api_key_textbox.change(set_openai_api_key,
                                   inputs=[openai_api_key_textbox],
                                   outputs=[chain_state, express_chain_state, llm_state, embeddings_state,
                                            qa_chain_state, memory_state])
-    """
+
 block.launch(debug=True)