Update app.py

app.py

@@ -161,6 +161,7 @@ col1.write(f"**You:** {user_input}")
 ask = col2.button("Ask", type="primary")
 
 if ask:
+    api_key_ = api_key
     if not api_key:
         st.markdown(f"""**{BOOK_NAME}:** Whoops looks like you forgot your API key buddy. 
                 We throw a dice. If it's 6, you can ask one question for free.
@@ -170,36 +171,31 @@ if ask:
 
         dice = randint(1, 6)
         logger.info(f" dice: {dice}")
-        if dice != 6:
-            api_key_ = api_key
-            # st.stop()
-        else:  # use space secret[OPENAI_API_KEY/envion, 
+        if dice == 6:
             api_key_ = os.environ["OPENAI_API_KEY"]
             st.write(f"**{BOOK_NAME}:** got {dice}, lucky you!")
-    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()
+
+    # 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:
-        # 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, or two, for your first question because some stuff needs to be downloaded 🥺👉🏻👈🏻"
-            st.session_state.key = 'value'
-            # ns.counter = 1            
-        with st.spinner(msg):
-            try:
-                answer, pages, extract = get_answer(question=user_input)
-                logger.info(f"answer: {answer}")
-            except Exception as exc:
-                st.write(f"**{BOOK_NAME}:**: {exc}")
-                st.stop()
+        msg = "Um... excuse me but... this can take about a minute, or two, for your first question because some stuff needs to be downloaded 🥺👉🏻👈🏻"
+        st.session_state.key = 'value'
+        # ns.counter = 1            
+    with st.spinner(msg):
+        try:
+            answer, pages, extract = get_answer(question=user_input)
+            logger.info(f"answer: {answer}")
+        except Exception as exc:
+            logger.error(exc)
+            st.write(f"**{BOOK_NAME}:**: {exc}")
+            st.stop()
 
     st.write(f"**{BOOK_NAME}:** {answer}")