Fixes

app.py

@@ -73,7 +73,7 @@ def prepare_download(vtt_content: str, audio_filename: str) -> str | None:
     with open(download_path, 'w', encoding='utf-8') as f:
         f.write(vtt_content)
 
-    return str(download_path)
+    return download_path
 
 
 with gr.Blocks(title="Transcription & Diarization") as app:
@@ -132,17 +132,17 @@ with gr.Blocks(title="Transcription & Diarization") as app:
 
                 validation_status = gr.Markdown("⚪ No content", container=True)
 
-            with gr.Row():
-                clean_btn = gr.Button("Clean & improve VTT", variant="secondary", interactive=False)
-                download_file = gr.File(label="Download VTT", visible=False)
-                download_btn = gr.Button("Download VTT", variant="secondary", interactive=False)
-
-            with gr.Accordion("🎭 Rename speakers", open=False):
+            with gr.Accordion("🎭 Rename speakers", open=True):
                 with gr.Row():
                     old_speaker_name = gr.Textbox(label="Current speaker name (e.g., SPEAKER_00)", placeholder="SPEAKER_00", value="SPEAKER_00")
                     new_speaker_name = gr.Textbox(label="New speaker name", placeholder="Davide")
 
                 rename_btn = gr.Button("Rename")
+            
+            with gr.Row():
+                clean_btn = gr.Button("Fix", variant="secondary", interactive=False)
+                download_btn = gr.DownloadButton("Download", variant="primary", interactive=False)
+
 
     def check_inputs(openai_key: str, hf_key: str, audio) -> gr.Button:
         """
@@ -177,7 +177,7 @@ with gr.Blocks(title="Transcription & Diarization") as app:
         return (
             status,
             gr.Button(interactive=is_valid),  # clean_btn
-            gr.Button(interactive=is_valid)   # download_btn
+            gr.DownloadButton(interactive=is_valid)   # download_btn
         )
 
     # Enable/disable submit button based on API keys and audio input
@@ -230,7 +230,6 @@ with gr.Blocks(title="Transcription & Diarization") as app:
     download_btn.click(
         fn=prepare_download,
         inputs=[output_vtt, audio_filename_state],
-        outputs=[download_file]
     )
 
     # Speaker renaming

src/vtt_utils.py

@@ -43,7 +43,7 @@ def validate_vtt(vtt_content: str) -> Tuple[str, str]:
     try:
         # Check if starts with WEBVTT
         if not vtt_content.strip().startswith("WEBVTT"):
-            return "❌ Invalid: Missing WEBVTT header", "error"
+            return "🔴 Invalid: Missing WEBVTT header", "error"
 
         lines = vtt_content.split('\n')
         has_timestamps = False
@@ -58,7 +58,7 @@ def validate_vtt(vtt_content: str) -> Tuple[str, str]:
             # Validate timestamp format
             match = re.match(r'(\d{2}:\d{2}:\d{2}\.\d{3})\s*-->\s*(\d{2}:\d{2}:\d{2}\.\d{3})', line)
             if not match:
-                return "⚠️ Warning: Malformed timestamp found", "warning"
+                return "🟡 Warning: Malformed timestamp found", "warning"
 
             # Parse and validate timestamps
             start_str, end_str = match.groups()
@@ -66,26 +66,26 @@ def validate_vtt(vtt_content: str) -> Tuple[str, str]:
             end_ms = parse_timestamp(end_str)
 
             if start_ms is None or end_ms is None:
-                return "⚠️ Warning: Invalid timestamp values", "warning"
+                return "🟡 Warning: Invalid timestamp values", "warning"
 
             if start_ms >= end_ms:
-                return "⚠️ Warning: Start timestamp >= end timestamp", "warning"
+                return "🟡 Warning: Start timestamp >= end timestamp", "warning"
 
             timestamps.append((start_ms, end_ms))
 
         if not has_timestamps:
-            return "❌ Invalid: No timestamps found", "error"
+            return "🔴 Invalid: No timestamps found", "error"
 
         # Check for overlapping timestamps
         for i in range(len(timestamps) - 1):
             current_end = timestamps[i][1]
             next_start = timestamps[i + 1][0]
             if current_end > next_start:
-                return "⚠️ Warning: Overlapping timestamps detected", "warning"
+                return "🟡 Warning: Overlapping timestamps detected", "warning"
 
-        return "✅ Valid VTT format", "success"
+        return "🟢 Valid", "success"
     except Exception as e:
-        return f"❌ Validation error: {str(e)}", "error"
+        return f"🔴 Validation error: {str(e)}", "error"
 
 
 def clean_vtt(vtt_content: str) -> str: