Update app.py

app.py

@@ -57,7 +57,7 @@ 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']
 
-# Create Expander and Columns UI for terms
+# Your existing expander and columns UI
 with st.expander("Search by Common Terms 📚"):
     cols = st.columns(4)
     for term in top_20_terms:
@@ -66,15 +66,17 @@ with st.expander("Search by Common Terms 📚"):
                 filtered_data = filter_by_keyword(data, term)
                 st.write(f"Filter on '{term}' 📊")
                 with st.sidebar:
-                    st.dataframe(filtered_data)
-                if not filtered_data.empty:
-                    html_blocks = []
+                    # Modified part: Display each row with a button
                     for idx, row in filtered_data.iterrows():
-                        question_text = row.get("question", "No question field")
-                        documentHTML5 = generate_html_with_textarea(question_text)
-                        html_blocks.append(documentHTML5)
-                    all_html = ''.join(html_blocks)
-                    components.html(all_html, width=1280, height=1024)
+                        if st.button(f"Row {idx}", key=f"row_{idx}"):
+                            st.session_state['last_clicked_row'] = idx
+
+                # Check if there's a selected row and it's in the filtered data
+                if st.session_state['last_clicked_row'] in filtered_data.index:
+                    row = filtered_data.loc[st.session_state['last_clicked_row']]
+                    question_text = row.get("question", "No question field")
+                    documentHTML5 = generate_html_with_textarea(question_text)
+                    components.html(documentHTML5, width=1280, height=1024)
 
 # Text input for search keyword
 search_keyword = st.text_input("Or, enter a keyword to filter data:")