Update app.py

app.py

@@ -3,6 +3,7 @@ from collections import defaultdict
 
 import streamlit as st
 import streamlit.components.v1 as components
+import plotly.graph_objects as go
 import plotly.express as px
 import numpy as np
 
@@ -22,14 +23,10 @@ st.markdown("""
 """, unsafe_allow_html=True)
 
 
-max_history = 4
-
+max_history = 10
 
 def set_state(x):
     if x == 1: st.toast('The journey begins!', icon='😍')
-    if x == 2: st.toast('Welcome!', icon='🙃')
-    if x == 3: st.toast("Let's play!", icon='🥰')
-    if x == 4: st.toast("Meet your friends!", icon='🤪')
     st.session_state.stage = x
 
         
@@ -40,11 +37,13 @@ def reset_game():
     st.session_state.history = []
     st.session_state.preds = defaultdict(lambda: defaultdict(int))
     st.session_state.pnl = [0]
+    st.session_state.min = [0]
+    st.session_state.max = [0]
 
 if 'stage' not in st.session_state:
     reset_game()
 
-st.title('How random can you truly be?')
+st.title('Can your brain be random?')
 
 if st.session_state.stage == 0:
     st.button('Begin', on_click=set_state, args=[1],use_container_width=True)
@@ -60,49 +59,75 @@ if st.session_state.stage == 0:
         
     
 
-def get_prev_seqs(history):
-    # st.write(history)
+# def get_prev_seqs(history):
+#     # st.write(history)
+    
+#     seqs = []
+#     for h in range(max_history+1):
+#         if len(history) >= h:
+#             previous_seq = ''.join(history[-h:]) if h > 0 else ""
+#             # st.write(previous_seq)
+#             seqs.append(previous_seq) # from small to largest
+#     return seqs
     
-    seqs = []
-    for h in range(max_history+1):
-        if len(history) >= h:
-            previous_seq = ''.join(history[-h:]) if h > 0 else ""
-            # st.write(previous_seq)
-            seqs.append(previous_seq) # from small to largest
-    return seqs
+# def refresh_preds():
+#     played = st.session_state.history[-1]
+#     seqs = get_prev_seqs(st.session_state.history[:-1])
+#     for seq in seqs:
+#         # st.write(f'"{seq}"',played)
+#         st.session_state.preds[seq][played] += 1
+
+def make_pred(max_history=max_history,alpha=0.5):
+    history = st.session_state.history
     
-def refresh_preds():
-    played = st.session_state.history[-1]
-    seqs = get_prev_seqs(st.session_state.history[:-1])
-    for seq in seqs:
-        # st.write(f'"{seq}"',played)
-        st.session_state.preds[seq][played] += 1
-
-def make_pred():
-    seqs = get_prev_seqs(st.session_state.history)
-    # st.write('seqs',seqs)
-    scores = defaultdict(float)
     denominator = 0
-    for i,seq in enumerate(seqs):
-        weight = (i+1)**2
-        preds = st.session_state.preds[seq]
-        total = sum(preds.values())
+    scores = {str(i):0 for i in range(st.session_state.n_buttons)}
+    recent = np.array(history[-max_history:])
+    for i in range(1,len(history)):
+        past = np.array(history[-i-max_history:-i])
         
-        if total:
-            for played,value in preds.items():
-                scores[played] += weight*value/total
-            denominator += weight
-    if denominator:
-        scores = {played:value/denominator for played,value in scores.items()}
-        return scores
-    else:
-        return {str(i):1/st.session_state.n_buttons for i in range(st.session_state.n_buttons)}
+        played = history[-i]
+        decay = np.exp(-alpha*np.linspace(1,0,len(past)))
+        similarity = ((past == recent[-len(past):])*decay).sum() / decay.sum()
+        weight = len(history)/(len(history)+i) * similarity
+        scores[played] += weight
+        denominator += weight
+    if not denominator:
+        return get_random_play()
+    result = {str(i):scores[str(i)]/denominator for i in range(st.session_state.n_buttons)}
+    return result
+    
+def get_random_play():
+    return {str(i):1/st.session_state.n_buttons for i in range(st.session_state.n_buttons)}
+
+# def make_pred():
+#     seqs = get_prev_seqs(st.session_state.history)
+#     # st.write('seqs',seqs)
+#     scores = defaultdict(float)
+#     denominator = 0
+#     for i,seq in enumerate(seqs):
+#         weight = (i+1)**2
+#         preds = st.session_state.preds[seq]
+#         total = sum(preds.values())
+        
+#         if total:
+#             for played,value in preds.items():
+#                 scores[played] += weight*value/total
+#             denominator += weight
+#     if denominator:
+#         scores = {played:value/denominator for played,value in scores.items()}
+#         return scores
+#     else:
+#         return get_random_play()
         
 
 def update_pnl(user_win):
     current_score = st.session_state.pnl[-1]
     current_score += -1 if user_win else (st.session_state.n_buttons-1)
     st.session_state.pnl.append(current_score)
+    expected_change = (2**0.5) * ((st.session_state.n_buttons-1)/st.session_state.n_buttons)
+    st.session_state.min.append(st.session_state.min[-1]-expected_change)
+    st.session_state.max.append(st.session_state.max[-1]+expected_change)
     
 def user_select(i,choice):
     st.session_state.history.append(str(i))
@@ -113,7 +138,7 @@ def user_select(i,choice):
         st.toast('Well done!', icon='😍')
         update_pnl(user_win=True)
     
-    refresh_preds()
+    # refresh_preds()
 
 def compute_perf():
     data = np.array(st.session_state.pnl)
@@ -135,8 +160,10 @@ if st.session_state.stage == 1:
         col.button(str(i),on_click=user_select, args=[str(i),choice],
         use_container_width=True)
 
-    st.title(f'My earnings so far... {compute_perf()} :-)')
-    st.plotly_chart(px.line(st.session_state.pnl), use_container_width=True)
+    st.subheader(f'My earnings so far... {compute_perf()} :-)')
+    fig = px.line(st.session_state.pnl)
+    fig.update_layout(showlegend=False)
+    st.plotly_chart(fig, use_container_width=True)
     
     st.button('Start over', on_click=reset_game, args=[],
         use_container_width=True)