Update app.py

app.py

@@ -10,7 +10,7 @@ from langchain import OpenAI
 from langchain import PromptTemplate
 from loguru import logger
 
-ns = SimpleNamespace(counter=0)
+# ns = SimpleNamespace(counter=0)
 
 st.set_page_config(page_title="Talk2Book", page_icon="📖")
 
@@ -138,7 +138,7 @@ with st.sidebar:
                             type = "password",
                             help = "This isn't saved 🙈"
                            )
-    os.environ["OPENAI_API_KEY"] = api_key
+    # os.environ["OPENAI_API_KEY"] = api_key
 
     st.markdown("---")
 
@@ -162,18 +162,27 @@ col1.write(f"**You:** {user_input}")
 # ask button to the right of the displayed question
 ask = col2.button("Ask", type="primary")
 
-
 if ask:
-
     if api_key is "":
-        st.write(f"**{BOOK_NAME}:** Whoops looks like you forgot your API key buddy")
+        st.write(f"**{BOOK_NAME}:** Whoops looks like you forgot your API key buddy. We throw a dice to decide wether you can ask one question for free.")
+        # stop with a prob = 0.9
+        if random() > 1/6:
+            st.stop()
+        else:  # use space secret[OPENAI_API_KEY/envion, 
+            api_key = os.environ["OPENAI_API_KEY"]
+    
+    if not api_key:
+        st.write(f"**{BOOK_NAME}:** no luck, try again?")
         st.stop()
     else:
-        if ns.counter: 
+        os.environ["OPENAI_API_KEY"] = api_key
+        # if ns.counter:   # this does not work
+        if 'key' in st.session_state:
             msg = "Just one sec"
         else:
             msg = "Um... excuse me but... this can take about a minute for your first question because some stuff have to be downloaded 🥺👉🏻👈🏻"
-            ns.counter = 1            
+            st.session_state.key = 'value'
+            # ns.counter = 1            
         with st.spinner(msg):
             try:
                 answer, pages, extract = get_answer(question=user_input)