Update app.py

app.py

@@ -226,6 +226,16 @@ async def process_video(video_url, segments, combine, audio_only, progress=gr.Pr
         return
 
     output_dir = Path('output')
+    
+    # Clean up the output directory
+    if output_dir.exists():
+        try:
+            shutil.rmtree(output_dir)
+            yield 0, "Cleaned up existing output directory", None
+        except Exception as e:
+            yield 0, f"Error cleaning up output directory: {str(e)}", None
+            return
+    
     output_dir.mkdir(exist_ok=True)
 
     try:
@@ -247,16 +257,18 @@ async def process_video(video_url, segments, combine, audio_only, progress=gr.Pr
                 video_output = output_dir / f"{sanitize_title(video_title)}_video_segment_{i+1}.mp4"
                 async for output in download_segment_async(video_url, start_time, end_time, str(video_output)):
                     progress((i / total_segments) * 0.8 + (1 / total_segments) * 0.4, f"Downloading video segment {i+1}/{total_segments}: {output}")
-                if not video_output.exists():
+                if video_output.exists():
+                    video_segments.append(str(video_output))
+                else:
                     raise FileNotFoundError(f"Video segment file not found: {video_output}")
-                video_segments.append(str(video_output))
 
             audio_output = output_dir / f"{sanitize_title(video_title)}_audio_segment_{i+1}.m4a"
             async for output in download_segment_async(audio_url or video_url, start_time, end_time, str(audio_output)):
                 progress((i / total_segments) * 0.8 + (1 / total_segments) * 0.8, f"Downloading audio segment {i+1}/{total_segments}: {output}")
-            if not audio_output.exists():
+            if audio_output.exists():
+                audio_segments.append(str(audio_output))
+            else:
                 raise FileNotFoundError(f"Audio segment file not found: {audio_output}")
-            audio_segments.append(str(audio_output))
         
         except Exception as e:
             yield (i / total_segments) * 100, f"Error downloading segment {i+1}: {str(e)}", None
@@ -271,16 +283,10 @@ async def process_video(video_url, segments, combine, audio_only, progress=gr.Pr
                 output_path = output_dir / f"{sanitize_title(video_title)}_combined.mp4"
             
             output_path = await combine_segments_async(video_segments if not audio_only else [], audio_segments, str(output_path))
-            if Path(output_path).exists():
-                yield 100, f"Segments combined and saved as {output_path}", output_path
-            else:
-                raise FileNotFoundError(f"Combined output file not found: {output_path}")
+            yield 100, f"Segments combined and saved as {output_path}", output_path
         else:
             output_path = audio_segments[0] if audio_only else video_segments[0]
-            if Path(output_path).exists():
-                yield 100, "All segments downloaded successfully", output_path
-            else:
-                raise FileNotFoundError(f"Output file not found: {output_path}")
+            yield 100, "All segments downloaded successfully", output_path
 
     except Exception as e:
         yield 100, f"Error: {str(e)}", None
@@ -320,95 +326,80 @@ def get_video_qualities(video_url):
 utils.colab_check = lambda: True
 
 custom_css = """
-    :root {
-        --primary-hue: 210;
-        --secondary-hue: 160;
-        --neutral-hue: 0;
-        --primary-color: hsl(var(--primary-hue), 70%, 55%);
-        --secondary-color: hsl(var(--secondary-hue), 70%, 55%);
-        --neutral-color: hsl(var(--neutral-hue), 0%, 95%);
-        --text-color: hsl(var(--neutral-hue), 0%, 20%);
-        --border-radius: 8px;
+    #component-0 {
+        max-width: 100%;
+        margin: 0 auto;
+        padding: 10px;
+        box-sizing: border-box;
     }
-    body { font-family: 'Inter', sans-serif; margin: 0; padding: 0; height: 100vh; overflow: hidden; }
-    #component-0 { height: 100vh; overflow-y: auto; padding: 1rem; box-sizing: border-box; }
-    .container { max-width: 1200px; margin: 0 auto; display: flex; flex-direction: column; height: 100%; }
-    .header { text-align: center; margin-bottom: 1rem; }
-    .main-content { display: flex; flex-wrap: wrap; gap: 1rem; flex-grow: 1; overflow-y: auto; }
-    .input-section, .segment-controls, .options-section, .output-section {
-        background-color: var(--neutral-color);
-        border-radius: var(--border-radius);
-        padding: 1rem;
-        box-shadow: 0 2px 4px rgba(0,0,0,0.1);
+    .container {
+        display: flex;
+        flex-direction: column;
+        height: calc(100vh - 100px);
+        max-width: 1200px;
+        margin: 0 auto;
+    }
+    .header {
+        text-align: center;
+        margin-bottom: 1rem;
+    }
+    .main-content {
+        display: flex;
+        flex-wrap: wrap;
+        gap: 1rem;
+        overflow-y: auto;
+        flex: 1;
+    }
+    .input-section, .options-section {
         flex: 1 1 300px;
+        min-width: 300px;
     }
-    .segment-input, .time-input, .button-row {
+    .output-section {
+        flex-basis: 100%;
+    }
+    .segment-input {
         display: flex;
         flex-wrap: wrap;
         gap: 0.5rem;
         align-items: center;
     }
-    .time-input { width: 60px !important; text-align: center; }
-    .time-label, .separator { font-weight: bold; color: var(--primary-color); margin-bottom: 0.5rem; }
-    .button-row { justify-content: space-between; }
+    .time-input {
+        width: 60px !important;
+    }
     @media (max-width: 768px) {
-        .main-content { flex-direction: column; }
-        .input-section, .segment-controls, .options-section, .output-section { flex: 1 1 100%; }
+        .main-content {
+            flex-direction: column;
+        }
+        .input-section, .options-section {
+            flex-basis: 100%;
+        }
     }
-    .url-input { flex-grow: 1; }
-    .quality-dropdown { width: 100px !important; }
 """
 
-with gr.Blocks(
-    title="YouTube Segment Downloader",
-    theme=gr.themes.Soft(
-        primary_hue="blue",
-        secondary_hue="green",
-        neutral_hue="gray"
-    ).set(
-        body_text_color="*neutral_800",
-        background_fill_primary="*neutral_50",
-        button_primary_background_fill="*primary_500",
-        button_primary_background_fill_hover="*primary_600",
-        button_secondary_background_fill="*secondary_500",
-        button_secondary_background_fill_hover="*secondary_600",
-    ),
-    css=custom_css
-) as iface:
+with gr.Blocks(css=custom_css) as iface:
     with gr.Column(elem_classes="container"):
         gr.Markdown("# 🎬 YouTube Segment Downloader", elem_classes="header")
         
         with gr.Row(elem_classes="main-content"):
             with gr.Column(elem_classes="input-section"):
-                with gr.Row():
-                    video_url = gr.Textbox(
-                        label="YouTube URL",
-                        placeholder="Paste URL here",
-                        elem_classes="url-input"
-                    )
-                    quality = gr.Dropdown(
-                        label="Quality",
-                        choices=[],
-                        interactive=True,
-                        elem_classes="quality-dropdown",
-                        visible=False
-                    )
+                video_url = gr.Textbox(label="YouTube URL", placeholder="Paste URL here")
+                quality = gr.Dropdown(label="Quality", choices=[], interactive=True)
                 url_status = gr.Markdown(visible=False)
                 
-                with gr.Column(elem_classes="segment-controls"):
-                    gr.Markdown("### Add Segments", elem_classes="time-label")
+                with gr.Column():
+                    gr.Markdown("### Add Segments")
                     with gr.Row(elem_classes="segment-input"):
                         start_hours = gr.Number(label="Start HH", minimum=0, maximum=23, step=1, value=0, elem_classes="time-input")
                         start_minutes = gr.Number(label="MM", minimum=0, maximum=59, step=1, value=0, elem_classes="time-input")
                         start_seconds = gr.Number(label="SS", minimum=0, maximum=59, step=1, value=0, elem_classes="time-input")
                         
-                        gr.Markdown("to", elem_classes="separator")
+                        gr.Markdown("to")
                         
                         end_hours = gr.Number(label="End HH", minimum=0, maximum=23, step=1, value=0, elem_classes="time-input")
                         end_minutes = gr.Number(label="MM", minimum=0, maximum=59, step=1, value=0, elem_classes="time-input")
-                        end_seconds = gr.Number(label="SS", minimum=0, maximum=59, step=1, value=0, elem_classes="time_input")
+                        end_seconds = gr.Number(label="SS", minimum=0, maximum=59, step=1, value=0, elem_classes="time-input")
                     
-                    add_btn = gr.Button("Add Segment", variant="primary")
+                    add_btn = gr.Button("Add Segment")
                 
                 segments = gr.Dataframe(
                     headers=["Segment"],
@@ -423,10 +414,10 @@ with gr.Blocks(
                 combine = gr.Checkbox(label="Combine segments", value=True)
                 audio_only = gr.Checkbox(label="Audio only", value=False)
                 remove_index = gr.Number(label="Remove segment", minimum=0, step=1, value=0)
-                remove_btn = gr.Button("Remove", variant="secondary")
+                remove_btn = gr.Button("Remove")
                 old_index = gr.Number(label="Move from", minimum=0, step=1, value=0)
                 new_index = gr.Number(label="to", minimum=0, step=1, value=0)
-                move_btn = gr.Button("Move", variant="secondary")
+                move_btn = gr.Button("Move")
                 
                 submit_btn = gr.Button("🚀 Download", variant="primary")