Update app.py

app.py

@@ -1,4 +1,6 @@
 import os
+
+import openai
 from random import randint
 import streamlit as st
 from types import SimpleNamespace
@@ -177,12 +179,13 @@ if ask:
     else:
         # api_key_ = os.environ["OPENAI_API_KEY"]
         api_key_ = api_key
-        
+
+    openai.api_key = api_key_
     if not api_key_:
         st.write(f"**{BOOK_NAME}:** got {dice}, no luck, try again?")
         st.stop()
     else:
-        os.environ["OPENAI_API_KEY"] = api_key_
+        # os.environ["OPENAI_API_KEY"] = api_key_
         # if ns.counter:   # this does not work
         if 'key' in st.session_state:
             msg = "Just one sec"