Update app.py

app.py

@@ -2,77 +2,40 @@ import streamlit as st
 import pandas as pd
 import time
 import matplotlib.pyplot as plt
+import json
 
-# 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
+# Load USMLE questions from a JSON file
+def load_questions():
+    return pd.DataFrame([{"question": "What is the serum level of X?", "answer": "High"},
+                         {"question": "What causes Y?", "answer": "Z"}])
 
-# Function to vote on a story
+# Vote for 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)
+    st.session_state.df.to_csv('votes.csv', index=False)
 
-# Function to break story into sentences
-def break_into_sentences(story):
-    return story.split(". ")
+# Streamlit App
+st.title("Medical App 🎙🗳️")
 
-# 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
+# Initialize session state
 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()
+        st.session_state.df = pd.DataFrame({'story': ['Story 1', 'Story 2'], 'votes': [0, 0]})
+        st.session_state.df.to_csv('votes.csv', index=False)
+
+# Tab layout
+tab = st.selectbox("Choose a Tab", ["USMLE Questions", "Story Voting"])
+
+if tab == "USMLE Questions":
+    questions_df = load_questions()
+    selected_q = st.selectbox('Select a question:', questions_df['question'])
+    st.write(f"Answer: {questions_df.loc[questions_df['question'] == selected_q, 'answer'].values[0]}")
+
+elif tab == "Story Voting":
+    for index, row in st.session_state.df.iterrows():
+        if st.button(f"Vote for: {row['story']}"):
+            vote_story(index)
+    st.table(st.session_state.df)
+    st.bar_chart(st.session_state.df.set_index('story')['votes'])