Update app.py

app.py

@@ -1,19 +1,20 @@
 import streamlit as st
 import pandas as pd
 
-# Create a dataset for stories and votes
-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]
-})
+# 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 vote on a story
 def vote_story(index):
-    df.loc[index, 'votes'] += 1
+    st.session_state.df.loc[index, 'votes'] += 1
 
 # Function to break story into sentences
 def break_into_sentences(story):
@@ -24,18 +25,20 @@ def display_story(index, row):
     sentences = break_into_sentences(row['story'])
     for sentence in sentences:
         st.text(sentence)
-    st.button(f"Vote for Story {index + 1}", on_click=vote_story, args=(index,))
+    if st.button(f"Vote for Story {index + 1}"):
+        vote_story(index)
 
 # Main app
 st.title('Medical Story Voting 🗳️')
 
 # Display stories and voting buttons
-for index, row in df.iterrows():
+for index, row in st.session_state.df.iterrows():
     col1, col2 = st.columns([3,1])
     with col1:
         display_story(index, row)
-    col2.write(f"Votes: {row['votes']}")
+    with col2:
+        st.write(f"Votes: {row['votes']}")
 
 # Display total votes
 st.markdown("### 📊 Vote Summary")
-st.table(df)
+st.table(st.session_state.df)