Update app.py

app.py

@@ -9,6 +9,7 @@ from yt_dlp import YoutubeDL
 import subprocess
 import shutil
 from typing import List, Tuple
+import pandas as pd
 
 def sanitize_title(title):
     return re.sub(r'[\\/*?:"<>|]', "", title)
@@ -103,22 +104,22 @@ def add_segment(start_hours, start_minutes, start_seconds, end_hours, end_minute
     start_time = f"{start_hours:02d}:{start_minutes:02d}:{start_seconds:02d}"
     end_time = f"{end_hours:02d}:{end_minutes:02d}:{end_seconds:02d}"
     new_segment = f"{start_time}-{end_time}"
-    new_row = [new_segment]
-    return segments + [new_row]
+    new_row = pd.DataFrame([new_segment], columns=["Segment"])
+    return pd.concat([segments, new_row], ignore_index=True)
 
 def remove_segment(segments, index):
-    return segments[:index] + segments[index+1:]
+    return segments.drop(index).reset_index(drop=True)
 
 def move_segment(segments, old_index, new_index):
-    segments_list = segments.values.tolist()  # Convert Dataframe to list
-    if 0 <= old_index < len(segments_list) and 0 <= new_index < len(segments_list):
-        segment = segments_list.pop(old_index)
-        segments_list.insert(new_index, segment)
-    return segments_list
+    if 0 <= old_index < len(segments) and 0 <= new_index < len(segments):
+        segment = segments.iloc[old_index]
+        segments = segments.drop(old_index).reset_index(drop=True)
+        segments = pd.concat([segments.iloc[:new_index], pd.DataFrame([segment]), segments.iloc[new_index:]]).reset_index(drop=True)
+    return segments
 
-def parse_segments(segments: List[str]) -> List[Tuple[int, int]]:
+def parse_segments(segments: pd.DataFrame) -> List[Tuple[int, int]]:
     parsed_segments = []
-    for segment in segments:
+    for segment in segments['Segment']:
         start, end = map(lambda x: sum(int(i) * 60 ** j for j, i in enumerate(reversed(x.split(':')))), segment.split('-'))
         if start < end:
             parsed_segments.append((start, end))
@@ -129,8 +130,7 @@ def process_video(video_url, segments, combine, progress=gr.Progress()):
         return 0, "Error: Please provide a valid YouTube URL", None
 
     # Extract segments from the Dataframe
-    segment_list = [segment[0] for segment in segments if segment[0].strip()]
-    parsed_segments = parse_segments(segment_list)
+    parsed_segments = parse_segments(segments)
     if not parsed_segments:
         return 0, "Error: No valid segments provided", None
 
@@ -201,6 +201,7 @@ with gr.Blocks(title="Advanced YouTube Segment Downloader", theme=gr.themes.Soft
                 headers=["Segment"],
                 row_count=5,
                 col_count=1,
+                datatype=["str"],
                 interactive=True,
                 label="Segments"
             )