Update app.py

app.py

@@ -146,8 +146,6 @@ with st.sidebar:
     st.info("Based on [Talk2Book](https://github.com/batmanscode/Talk2Book)")
 
 
-    
-
 ##### main ####
 # user_input = st.text_input("Your question", "Who are you?", key="input")
 
@@ -178,12 +176,14 @@ if ask:
             # st.stop()
         else:  # use space secret[OPENAI_API_KEY/envion, 
             api_key_ = os.environ["OPENAI_API_KEY"]
-    
+            st.write(f"**{BOOK_NAME}:** got {dice}, lucky you!")
+    else:
+        api_key_ = os.environ["OPENAI_API_KEY"]
+        
     if not api_key_:
         st.write(f"**{BOOK_NAME}:** got {dice}, no luck, try again?")
         st.stop()
     else:
-        st.write(f"**{BOOK_NAME}:** got {dice}, lucky you!")
         os.environ["OPENAI_API_KEY"] = api_key_
         # if ns.counter:   # this does not work
         if 'key' in st.session_state: