update

chat_dov.py

@@ -86,16 +86,26 @@ def create_chatbot(corpus, character_name, chatbot_type, retrieval_docs, summary
 
 def main():
     # parametros fixos para Dov Tzamir, arquivos ja processados , exceto indice que são em memoria
-    chatbot = st.cache_resource(create_chatbot)(
-            "data/tzamir.txt",    #args.corpus,  
-            "Dov",               #args.character_name,
-            "retrieval",         #args.chatbot_type,
-            "raw",               #args.retrieval_docs,
-            "map_reduce",        #args.summary_type,
-        )
+
     st.title("Data Driven Characters")
     st.write("Create your own character chatbots, grounded in existing corpora.")
-    st.divider()
+
+    openai_api_key = st.text_input(
+        label="Your OpenAI API KEY",
+        placeholder="Your OpenAI API KEY",
+        type="password",
+    )
+    os.environ["OPENAI_API_KEY"] = openai_api_key
+
+    chatbot = st.cache_resource(create_chatbot)(
+        "data/tzamir.txt",    #args.corpus,  
+        "Dov",               #args.character_name,
+        "retrieval",         #args.chatbot_type,
+        "raw",               #args.retrieval_docs,
+        "map_reduce",        #args.summary_type,
+    )
+
+    #st.divider()
     """
     st.markdown(f"**chatbot type**: *{args.chatbot_type}*")
     if "retrieval" in args.chatbot_type: