jiehou commited on
Commit
86c3d9a
·
verified ·
1 Parent(s): cfabc1b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -12
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
- all_combos = list(combinations(selected_indices, window_size))
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