Update app.py

app.py

@@ -1,20 +1,18 @@
 import streamlit as st
 import pandas as pd
+import time
+import matplotlib.pyplot as plt
 
-# Initialize session state for the vote counts
-if 'df' not in st.session_state:
-    st.session_state.df = pd.DataFrame({
-        'story': [
-            "A 45-year-old man presents with a long history of ulcers on the bottom of his feet. 👣🚑",
-            "A 24-year-old man, an information technology professional, gets himself tested for serum immunoglobulin M (IgM) levels. 💻🧪",
-            "A 33-year-old woman who was recently involved in a motor vehicle accident presents to a medical clinic for a follow-up visit. 🚗🏥",
-        ],
-        'votes': [0, 0, 0]
-    })
+# Function to persist and reload the dataframe
+def persist_and_reload(df):
+    df.to_csv('votes.csv', index=False)
+    new_df = pd.read_csv('votes.csv')
+    return new_df
 
 # Function to vote on a story
 def vote_story(index):
     st.session_state.df.loc[index, 'votes'] += 1
+    st.session_state.df = persist_and_reload(st.session_state.df)
 
 # Function to break story into sentences
 def break_into_sentences(story):
@@ -27,6 +25,26 @@ def display_story(index, row):
         st.text(sentence)
     if st.button(f"Vote for Story {index + 1}"):
         vote_story(index)
+        
+    # Display a small bar chart for vote count
+    fig, ax = plt.subplots()
+    ax.barh(['Votes'], [row['votes']])
+    st.pyplot(fig)
+
+# Initialize session state for the vote counts
+if 'df' not in st.session_state:
+    try:
+        st.session_state.df = pd.read_csv('votes.csv')
+    except FileNotFoundError:
+        st.session_state.df = pd.DataFrame({
+            'story': [
+                "A 45-year-old man presents with a long history of ulcers on the bottom of his feet. 👣🚑",
+                "A 24-year-old man, an information technology professional, gets himself tested for serum immunoglobulin M (IgM) levels. 💻🧪",
+                "A 33-year-old woman who was recently involved in a motor vehicle accident presents to a medical clinic for a follow-up visit. 🚗🏥",
+            ],
+            'votes': [0, 0, 0]
+        })
+        st.session_state.df = persist_and_reload(st.session_state.df)
 
 # Main app
 st.title('Medical Story Voting 🗳️')
@@ -42,3 +60,11 @@ for index, row in st.session_state.df.iterrows():
 # Display total votes
 st.markdown("### 📊 Vote Summary")
 st.table(st.session_state.df)
+
+# Reload votes from file every 30 seconds
+if 'last_updated' not in st.session_state:
+    st.session_state.last_updated = time.time()
+
+if time.time() - st.session_state.last_updated > 30:
+    st.session_state.df = pd.read_csv('votes.csv')
+    st.session_state.last_updated = time.time()