Update app.py

app.py

@@ -585,13 +585,20 @@ def main():
                     current_selection = st.session_state['ref_selections'].get(selected_ref['name'], [])
                     default_names = [structure_info[i]['full_name'] for i in current_selection] if current_selection else []
                     
+                    #selected_names = st.multiselect(
+                    #    "Select residues",
+                    #    options=[info['full_name'] for info in structure_info],
+                    #    default=[structure_info[i]['full_name'] for i in range(2, len(structure_info)-2)],
+                    #    key=f"specific_ref_{selected_ref['name']}"
+                    #)
                     selected_names = st.multiselect(
                         "Select residues",
                         options=[info['full_name'] for info in structure_info],
-                        default=[structure_info[i]['full_name'] for i in range(2, len(structure_info)-2)],
+                        default=default_names if default_names else [structure_info[i]['full_name'] for i in range(2, len(structure_info)-2)],  # ✅ Use saved selection
                         key=f"specific_ref_{selected_ref['name']}"
                     )
                     
+                    
                     name_to_idx = {info['full_name']: info['index'] for info in structure_info}
                     selected_indices = [name_to_idx[name] for name in selected_names]
                     selected_indices.sort()
@@ -684,10 +691,17 @@ def main():
                     current_selection = st.session_state['query_selections'].get(selected_query['name'], [])
                     default_names = [structure_info[i]['full_name'] for i in current_selection] if current_selection else []
                     
+                    #selected_names = st.multiselect(
+                    #    "Select residues",
+                    #    options=[info['full_name'] for info in structure_info],
+                    #    default=[structure_info[i]['full_name'] for i in range(2, len(structure_info)-2)],
+                    #    key=f"specific_query_{selected_query['name']}"
+                    #)
+
                     selected_names = st.multiselect(
                         "Select residues",
                         options=[info['full_name'] for info in structure_info],
-                        default=[structure_info[i]['full_name'] for i in range(2, len(structure_info)-2)],
+                        default=default_names if default_names else [structure_info[i]['full_name'] for i in range(2, len(structure_info)-2)],  # ✅ Use saved selection
                         key=f"specific_query_{selected_query['name']}"
                     )