Update app.py

app.py

@@ -1,5 +1,5 @@
 import os
-from random import random
+from random import randint
 import streamlit as st
 from types import SimpleNamespace
 
@@ -165,17 +165,22 @@ ask = col2.button("Ask", type="primary")
 
 if ask:
     if not api_key:
-        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.")
+        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.
+                """
+           )
         # stop with a prob = 0.9
-        
-        if random() > 1/6:
+
+        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, 
             api_key_ = os.environ["OPENAI_API_KEY"]
     
     if not api_key_:
-        st.write(f"**{BOOK_NAME}:** no luck, try again?")
+        st.write(f"**{BOOK_NAME}:** got {dice}, no luck, try again?")
         st.stop()
     else:
         os.environ["OPENAI_API_KEY"] = api_key