Update app.py

app.py

@@ -11,17 +11,34 @@ teams = [
     ('Team 5', '🐉', 'Dragons', 'Houston')
 ]
 
+def create_sankey(votes, turn):
+    labels = [f"{team[1]} {team[2]}" for team in teams] + ['Upvotes', 'Downvotes']
+    source = []
+    target = []
+    value = []
+
+    for i, team in enumerate(teams):
+        source += [i, i]
+        target += [len(teams), len(teams) + 1]
+        value += [votes[team[0]][0], votes[team[0]][1]]
+
+    fig = go.Figure(data=[go.Sankey(
+        node=dict(pad=15, thickness=20, line=dict(color='black', width=0.5), label=labels),
+        link=dict(source=source, target=target, value=value))])
+
+    fig.update_layout(title_text=f'Location Simulator by Nickname (Turn {turn})', title_font=dict(size=24, color='blue'))
+
+    return fig
+
 def run_scenario(duration=100, click_card_limit=None):
     start_time = time.time()
     votes = {team[0]: [0, 0] for team in teams}  # Initialize upvotes and downvotes
     click_cards = 0
     chat = []
+    turn = 0
 
-    fig = go.Figure(data=[go.Sankey(
-        node=dict(pad=15, thickness=20, line=dict(color='black', width=0.5),
-                  label=[f"{team[1]} {team[2]}" for team in teams] + ['Upvotes', 'Downvotes']),
-        link=dict(source=[], target=[], value=[]))])
-
+    st.header("Sankey Graph")
+    fig = create_sankey(votes, turn)
     st.plotly_chart(fig, use_container_width=True)
 
     while time.time() - start_time < duration:
@@ -39,19 +56,8 @@ def run_scenario(duration=100, click_card_limit=None):
             votes[team[0]][1] += clicks
         chat.append((team, vote_type, clicks))
 
-        source, target, value = [], [], []
-        for i, team in enumerate(teams):
-            source += [i, i]
-            target += [len(teams), len(teams) + 1]
-            value += [votes[team[0]][0], votes[team[0]][1]]
-
-        fig = go.Figure(data=[go.Sankey(
-            node=dict(pad=15, thickness=20, line=dict(color='black', width=0.5),
-                      label=[f"{team[1]} {team[2]}" for team in teams] + ['Upvotes', 'Downvotes']),
-            link=dict(source=source, target=target, value=value))])
-
-        fig.update_layout(title_text='Location Simulator by Nickname', title_font=dict(size=24, color='blue'))
-
+        turn += 1
+        fig = create_sankey(votes, turn)
         st.plotly_chart(fig, use_container_width=True)
 
         time.sleep(random.uniform(0, 1))  # Random sleep between 0 and 1 seconds
@@ -79,6 +85,3 @@ st.header("Final Results")
 for team, vote_counts in votes.items():
     st.write(f"{team}: {vote_counts[0]} upvotes, {vote_counts[1]} downvotes")
 
-st.header("Sankey Graph")
-fig = create_sankey(votes)
-st.plotly_chart(fig, use_container_width=True)