upload

app.py

@@ -260,28 +260,24 @@ def load_chain(tools_list, llm):
 
 
 def set_openai_api_key(api_key):
-    """Set the api key and return chain.
-    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"] = 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"])))
-        llm = OpenAI(temperature=0, max_tokens=MAX_TOKENS)
-        print(str(datetime.datetime.now()) + ": After OpenAI, OPENAI_API_KEY length: " + str(
-            len(os.environ["OPENAI_API_KEY"])))
-        chain, express_chain, memory = load_chain(TOOLS_DEFAULT_LIST, llm)
-
-        # Pertains to question answering functionality
-        embeddings = OpenAIEmbeddings()
-        qa_chain = load_qa_chain(OpenAI(temperature=0), chain_type="stuff")
-
-        print(str(datetime.datetime.now()) + ": After load_chain, OPENAI_API_KEY length: " + str(
-            len(os.environ["OPENAI_API_KEY"])))
-        os.environ["OPENAI_API_KEY"] = ""
-        return chain, express_chain, llm, embeddings, qa_chain, memory
-    return None, None, None, None, None, None
+    api_key = "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"
+    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"])))
+    llm = OpenAI(temperature=0, max_tokens=MAX_TOKENS)
+    print(str(datetime.datetime.now()) + ": After OpenAI, OPENAI_API_KEY length: " + str(
+        len(os.environ["OPENAI_API_KEY"])))
+    chain, express_chain, memory = load_chain(TOOLS_DEFAULT_LIST, llm)
+
+    # Pertains to question answering functionality
+    embeddings = OpenAIEmbeddings()
+    qa_chain = load_qa_chain(OpenAI(temperature=0), chain_type="stuff")
+
+    print(str(datetime.datetime.now()) + ": After load_chain, OPENAI_API_KEY length: " + str(
+        len(os.environ["OPENAI_API_KEY"])))
+    os.environ["OPENAI_API_KEY"] = ""
+    return chain, express_chain, llm, embeddings, qa_chain, memory
 
 
 def run_chain(chain, inp, capture_hidden_text):
@@ -609,8 +605,8 @@ with gr.Blocks(css=".gradio-container {background-color: lightgray}") as block:
                 gr.HTML(
                     """<b><center>GPT + WolframAlpha + Whisper</center></b>
                     <p><center>New feature: <b>Embeddings</b></center></p>""")
-            
-            openai_api_key_textbox = "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"
+
+            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):
@@ -711,7 +707,7 @@ with gr.Blocks(css=".gradio-container {background-color: lightgray}") as block:
             "Portuguese (Brazilian)", "Portuguese (European)", "Romanian", "Russian", "Spanish (European)",
             "Spanish (Mexican)", "Spanish (US)", "Swedish", "Turkish", "Ukrainian", "Welsh",
             "emojis", "Gen Z slang", "how the stereotypical Karen would say it", "Klingon", "Neanderthal",
-            "Pirate", "Strange Planet expospeak technical talk", "Yoda", "	Vietnamese (Vietnam)"],
+            "Pirate", "Strange Planet expospeak technical talk", "Yoda", "Vietnamese (Vietnam)"],
                                       value=TRANSLATE_TO_DEFAULT)
 
         translate_to_radio.change(update_foo,