Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -291,6 +291,7 @@ def extract_window_coords(residues, window_indices):
|
|
| 291 |
return np.array(coords_list)
|
| 292 |
|
| 293 |
|
|
|
|
| 294 |
def generate_windows_from_selection(selected_indices, window_size, window_type):
|
| 295 |
"""Generate windows from selected residue indices"""
|
| 296 |
if len(selected_indices) < window_size:
|
|
@@ -298,22 +299,12 @@ def generate_windows_from_selection(selected_indices, window_size, window_type):
|
|
| 298 |
|
| 299 |
windows = []
|
| 300 |
if window_type == "contiguous":
|
| 301 |
-
# Only sliding windows
|
| 302 |
for i in range(len(selected_indices) - window_size + 1):
|
| 303 |
windows.append(selected_indices[i:i+window_size])
|
| 304 |
else: # non-contiguous
|
| 305 |
from itertools import combinations
|
| 306 |
-
|
| 307 |
-
|
| 308 |
-
# Get the contiguous windows (to exclude them)
|
| 309 |
-
contiguous_windows = []
|
| 310 |
-
for i in range(len(selected_indices) - window_size + 1):
|
| 311 |
-
contiguous_windows.append(tuple(selected_indices[i:i+window_size]))
|
| 312 |
-
|
| 313 |
-
# Filter: keep only combinations that are NOT in contiguous_windows
|
| 314 |
-
for combo in all_combos:
|
| 315 |
-
if combo not in contiguous_windows:
|
| 316 |
-
windows.append(list(combo))
|
| 317 |
|
| 318 |
return windows
|
| 319 |
|
|
|
|
| 291 |
return np.array(coords_list)
|
| 292 |
|
| 293 |
|
| 294 |
+
|
| 295 |
def generate_windows_from_selection(selected_indices, window_size, window_type):
|
| 296 |
"""Generate windows from selected residue indices"""
|
| 297 |
if len(selected_indices) < window_size:
|
|
|
|
| 299 |
|
| 300 |
windows = []
|
| 301 |
if window_type == "contiguous":
|
|
|
|
| 302 |
for i in range(len(selected_indices) - window_size + 1):
|
| 303 |
windows.append(selected_indices[i:i+window_size])
|
| 304 |
else: # non-contiguous
|
| 305 |
from itertools import combinations
|
| 306 |
+
windows = list(combinations(selected_indices, window_size))
|
| 307 |
+
windows = [list(w) for w in windows]
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 308 |
|
| 309 |
return windows
|
| 310 |
|