Update app.py

app.py

@@ -1,5 +1,3 @@
-# Team.Click.Battle.Multiplayer
-
 import streamlit as st
 import random
 import time
@@ -19,6 +17,13 @@ def run_scenario(duration=100, click_card_limit=None):
     click_cards = 0
     chat = []
 
+    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.plotly_chart(fig, use_container_width=True)
+
     while time.time() - start_time < duration:
         if click_card_limit is None or click_cards < click_card_limit:
             click_cards += 1
@@ -33,36 +38,25 @@ def run_scenario(duration=100, click_card_limit=None):
         else:
             votes[team[0]][1] += clicks
         chat.append((team, vote_type, clicks))
-        time.sleep(random.uniform(0, 1))  # Random sleep between 0 and 1 seconds
 
-    return votes, chat
-
-def save_votes_to_file(votes, filename='upvotes.txt'):
-    with open(filename, 'w') as f:
-        for team, vote_counts in votes.items():
-            f.write(f"{team}: {vote_counts[0]} upvotes, {vote_counts[1]} downvotes\n")
+        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]]
 
-def create_sankey(votes):
-    labels = []
-    source = []
-    target = []
-    value = []
+        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))])
 
-    for i, team in enumerate(teams):
-        labels.append(f"{team[1]} {team[2]}")
-        source += [i, i]
-        target += [len(teams), len(teams) + 1]
-        value += [votes[team[0]][0], votes[team[0]][1]]
+        fig.update_layout(title_text='Location Simulator by Nickname', title_font=dict(size=24, color='blue'))
 
-    labels += ['Upvotes', 'Downvotes']
-
-    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))])
+        st.plotly_chart(fig, use_container_width=True)
 
-    fig.update_layout(title_text='Location Simulator by Nickname', title_font=dict(size=24, color='blue'))
+        time.sleep(random.uniform(0, 1))  # Random sleep between 0 and 1 seconds
 
-    return fig
+    return votes, chat
 
 st.title("Team Upvotes and Downvotes Emoji Game")
 
@@ -72,20 +66,19 @@ click_card_limit = st.slider("Click Card Limit", min_value=0, max_value=100, val
 st.write(f"Running scenario for {duration} seconds with {click_card_limit} click cards...")
 votes, chat = run_scenario(duration, click_card_limit)
 
-save_votes_to_file(votes)
-
 st.header("Results")
 for team, vote_counts in votes.items():
     st.write(f"{team}: {vote_counts[0]} upvotes, {vote_counts[1]} downvotes")
 
 st.header("Chat")
 for message in chat:
-    team, vote_type, clicks = message
-    st.write(f"{team[1]} {team[2]}: {clicks} {vote_type}s")
+team, vote_type, clicks = message
+st.write(f"{team[1]} {team[2]}: {clicks} {vote_type}s")
+
+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)
-
-
-
+st.plotly_chart(fig, use_container_width=True)