Update app.py

app.py

@@ -1035,7 +1035,7 @@ def main():
                 
                 with col1:
                     # Reference structure
-                    ref_filename = f"ref_{selected_row['Reference'].replace('.pdb', '')}_{'_'.join(map(str, [i+1 for i in selected_row['Ref_Indices']]))}.pdb"
+                    ref_filename = f"ref_{selected_row['Reference'].replace('.pdb', '')}_{'_'.join(map(str, [i+1 for i in selected_row['Ref_Window']]))}.pdb"
                     st.download_button(
                         label="📥 Reference PDB",
                         data=ref_pdb,
@@ -1046,7 +1046,7 @@ def main():
                 
                 with col2:
                     # Query structure (original position)
-                    query_filename = f"query_{selected_row['Query'].replace('.pdb', '')}_{'_'.join(map(str, [i+1 for i in selected_row['Query_Indices']]))}.pdb"
+                    query_filename = f"query_{selected_row['Query'].replace('.pdb', '')}_{'_'.join(map(str, [i+1 for i in selected_row['Query_Window']]))}.pdb"
                     st.download_button(
                         label="📥 Query PDB (Original)",
                         data=query_pdb,
@@ -1057,7 +1057,7 @@ def main():
                 
                 with col3:
                     # Query structure (aligned)
-                    query_aligned_filename = f"query_aligned_{selected_row['Query'].replace('.pdb', '')}_{'_'.join(map(str, [i+1 for i in selected_row['Query_Indices']]))}.pdb"
+                    query_aligned_filename = f"query_aligned_{selected_row['Query'].replace('.pdb', '')}_{'_'.join(map(str, [i+1 for i in selected_row['Query_Window']]))}.pdb"
                     st.download_button(
                         label="📥 Query PDB (Aligned)",
                         data=query_aligned_pdb,