import streamlit as st
import json
import pandas as pd
import streamlit.components.v1 as components
# Function to load JSONL file into a DataFrame
def load_jsonl(file_path):
data = []
with open(file_path, 'r') as f:
for line in f:
data.append(json.loads(line))
return pd.DataFrame(data)
# Your filter_by_keyword function
def filter_by_keyword(data, term):
# Your filtering logic here
return data[data['column_name'].str.contains(term)]
# Function to generate HTML with textarea
def generate_html_with_textarea(row):
first_three_columns_text = ' '.join([f"{col}: {row[col]}" for col in row.index[:3]])
return f'''
Read It Aloud
🔊 Read It Aloud
'''
# Initialize session state for tracking the last clicked row
if 'last_clicked_row' not in st.session_state:
st.session_state['last_clicked_row'] = None
# Streamlit App 🚀
st.title("AI Medical Explorer with Speech Synthesis 🎙")
# Dropdown for file selection
file_option = st.selectbox("Select file:", ["usmle_16.2MB.jsonl", "usmle_2.08MB.jsonl"])
st.write(f"You selected: {file_option}")
# Load data
large_data = load_jsonl("usmle_16.2MB.jsonl")
small_data = load_jsonl("usmle_2.08MB.jsonl")
data = large_data if file_option == "usmle_16.2MB.jsonl" else small_data
# Top 20 healthcare terms for USMLE
top_20_terms = ['Heart', 'Lung', 'Pain', 'Memory', 'Kidney', 'Diabetes', 'Cancer', 'Infection', 'Virus', 'Bacteria', 'Gastrointestinal', 'Skin', 'Blood', 'Surgery']
# Initialize session state for tracking the last clicked row
if 'last_clicked_row' not in st.session_state:
st.session_state['last_clicked_row'] = None
# Streamlit app
with st.expander("Search by Common Terms 📚"):
cols = st.columns(4)
for term in top_20_terms:
with cols[top_20_terms.index(term) % 4]:
if st.button(f"{term}"):
filtered_data = filter_by_keyword(data, term)
st.write(f"Filter on '{term}' 📊")
# Display clickable rows in the dataframe
for idx, row in filtered_data.iterrows():
link = f"[Row {idx}]('javascript:void(0);')"
st.markdown(link, unsafe_allow_html=True)
# Capture the clicked row index
row_index = st.experimental_get_query_params().get("row_index")
if row_index:
selected_row = filtered_data.loc[int(row_index[0])]
html_content = generate_html_with_textarea(selected_row)
components.html(html_content, width=1280, height=1024)
# Always show the first row
if not filtered_data.empty:
first_row_html = generate_html_with_textarea(filtered_data.iloc[0])
components.html(first_row_html, width=1280, height=1024)
# Inject HTML5 and JavaScript for styling
st.markdown("""
""", unsafe_allow_html=True)
# Markdown and emojis for the case presentation
st.markdown("# 🏥 Case Study: 32-year-old Woman's Wellness Check")
st.markdown("## 📋 Patient Information")
st.markdown("""
- **Age**: 32
- **Gender**: Female
- **Past Medical History**: Asthma, Hypertension, Anxiety
- **Current Medications**: Albuterol, Fluticasone, Hydrochlorothiazide, Lisinopril, Fexofenadine
- **Vitals**
- **Temperature**: 99.5°F (37.5°C)
- **Blood Pressure**: 165/95 mmHg
- **Pulse**: 70/min
- **Respirations**: 15/min
- **Oxygen Saturation**: 98% on room air
""")
# Clinical Findings
st.markdown("## 📋 Clinical Findings")
st.markdown("""
- Cardiac exam reveals a S1 and S2 heart sound with a normal rate.
- Pulmonary exam is clear to auscultation bilaterally with good air movement.
- Abdominal exam reveals a bruit, normoactive bowel sounds, and an audible borborygmus.
- Neurological exam reveals cranial nerves II-XII as grossly intact with normal strength and reflexes in the upper and lower extremities.
""")
# Next Step Options
st.markdown("## 🤔 What is the best next step in management?")
# Multiple Choice
options = ["Blood Test", "MRI Scan", "Ultrasound with Doppler", "Immediate Surgery"]
choice = st.selectbox("", options)
# Explanation
if st.button("Submit"):
if choice == "Ultrasound with Doppler":
st.success("Correct! 🎉")
st.markdown("""
### Explanation
The patient's high blood pressure coupled with an abdominal bruit suggests the possibility of renal artery stenosis.
An **Ultrasound with Doppler** is the best next step for assessing blood flow and evaluating for renal artery stenosis.
""")
else:
st.error("Incorrect. 😞")
st.markdown("""
The best next step is **Ultrasound with Doppler**.
""")