RO-Rtechs commited on
Commit
534e3d5
·
verified ·
1 Parent(s): f8e63fc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -3
app.py CHANGED
@@ -120,9 +120,16 @@ def move_segment(segments, old_index, new_index):
120
  def parse_segments(segments: pd.DataFrame) -> List[Tuple[int, int]]:
121
  parsed_segments = []
122
  for segment in segments['Segment']:
123
- start, end = map(lambda x: sum(int(i) * 60 ** j for j, i in enumerate(reversed(x.split(':')))), segment.split('-'))
124
- if start < end:
125
- parsed_segments.append((start, end))
 
 
 
 
 
 
 
126
  return parsed_segments
127
 
128
  def process_video(video_url, segments, combine, progress=gr.Progress()):
 
120
  def parse_segments(segments: pd.DataFrame) -> List[Tuple[int, int]]:
121
  parsed_segments = []
122
  for segment in segments['Segment']:
123
+ if not isinstance(segment, str) or '-' not in segment:
124
+ continue
125
+ try:
126
+ start, end = segment.split('-')
127
+ start_seconds = sum(int(i) * 60 ** j for j, i in enumerate(reversed(start.split(':'))) if i)
128
+ end_seconds = sum(int(i) * 60 ** j for j, i in enumerate(reversed(end.split(':'))) if i)
129
+ if start_seconds < end_seconds:
130
+ parsed_segments.append((start_seconds, end_seconds))
131
+ except ValueError:
132
+ continue # Skip invalid segments
133
  return parsed_segments
134
 
135
  def process_video(video_url, segments, combine, progress=gr.Progress()):