app.py

import streamlit as st
import pandas as pd
import time
import matplotlib.pyplot as plt

# 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):
    return story.split(". ")

# Function to extract high info words
def extract_high_info_words(sentence):
    words = sentence.split(" ")
    return [word for word in words if len(word) > 6]

# Function to display story and voting
def display_story(index, row):
    sentences = break_into_sentences(row['story'])
    for sentence in sentences:
        st.text(sentence)
    
    high_info_words = []
    for sentence in sentences:
        high_info_words.extend(extract_high_info_words(sentence))
    
    if st.button(f"Vote for: {' '.join(high_info_words)}"):
        vote_story(index)

# 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 🗳️')

# Display stories and voting buttons
for index, row in st.session_state.df.iterrows():
    display_story(index, row)

# Display total votes
st.markdown("### 📊 Vote Summary")
st.table(st.session_state.df)

# Show all votes in one graph
fig, ax = plt.subplots()
ax.barh(st.session_state.df.index, st.session_state.df['votes'])
ax.set_yticks(st.session_state.df.index)
ax.set_yticklabels(st.session_state.df['story'])
st.pyplot(fig)

# 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()