Update app.py

app.py

@@ -18,18 +18,23 @@ def vote_story(index):
 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)
-    if st.button(f"Vote for Story {index + 1}"):
+    
+    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)
-        
-    # 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:
@@ -38,9 +43,9 @@ if 'df' not in st.session_state:
     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. 🚗🏥",
+                "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]
         })
@@ -51,16 +56,19 @@ st.title('Medical Story Voting 🗳️')
 
 # Display stories and voting buttons
 for index, row in st.session_state.df.iterrows():
-    col1, col2 = st.columns([3,1])
-    with col1:
-        display_story(index, row)
-    with col2:
-        st.write(f"Votes: {row['votes']}")
+    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()