diff --git a/main.py b/main.py index 8d79945..1c6b69c 100644 --- a/main.py +++ b/main.py @@ -84,15 +84,17 @@ def main(): console.print("[red]Could not find an active live chat for the given video ID.[/red]") return - console.print(f"[green]Found live chat ID: {live_chat_id}[/green]") - # Setup chat logging log_dir = "chat_logs" os.makedirs(log_dir, exist_ok=True) timestamp = datetime.now().strftime("%Y%m%d_%H%M%S") log_filename = os.path.join(log_dir, f"chat_{video_id}_{timestamp}.log") log_file = open(log_filename, "w", encoding="utf-8") - console.print(f"[green]Chat will be logged to {log_filename}[/green]") + + # Display initial messages using rich.Live for a few seconds + with Live(console=console, screen=True, refresh_per_second=4) as live: + live.update(Text(f"[green]Found live chat ID: {live_chat_id}[/green]\n[green]Chat will be logged to {log_filename}[/green]", justify="center")) + time.sleep(5) next_page_token = None message_count = 0