main.py

import sys
import os
from PyQt6.QtWidgets import (QApplication, QMainWindow, QWidget, QVBoxLayout, 
                             QHBoxLayout, QLineEdit, QPushButton, QProgressBar, 
                             QLabel, QFileDialog, QFrame, QStackedWidget, QTextEdit)
from PyQt6.QtCore import Qt, QSize
from yt_downloader import DownloadWorker
from PyQt6.QtGui import QIcon


# Professional Zinc/Blue Theme
MODERN_STYLE = """
QMainWindow { background-color: #09090b; }
QFrame#Sidebar { 
    background-color: #121214; 
    border-right: 1px solid #27272a; 
    min-width: 220px;
}
QLabel#Logo { 
    color: #ffffff; font-size: 16px; font-weight: 800; 
    padding: 25px; letter-spacing: 1px;
}
QPushButton#NavBtn { 
    background-color: transparent; color: #a1a1aa; 
    border: none; border-radius: 8px; text-align: left; 
    padding: 12px 20px; font-weight: 500; margin: 2px 10px;
}
QPushButton#NavBtn:hover { background-color: #1a1a1c; color: #ffffff; }
QPushButton#NavBtn[active="true"] { background-color: #3d5afe; color: #ffffff; }

QLineEdit { 
    background-color: #18181b; color: #fafafa; 
    border: 1px solid #27272a; border-radius: 8px; 
    padding: 12px; font-size: 13px;
}
QLineEdit:focus { border: 1px solid #3d5afe; }

QFrame#Card { 
    background-color: #121214; border: 1px solid #27272a; border-radius: 12px; 
}
QPushButton#PrimaryBtn { 
    background-color: #3d5afe; color: white; border-radius: 8px; 
    font-weight: 600; padding: 12px;
}
QPushButton#PrimaryBtn:hover { background-color: #536dfe; }

QProgressBar { 
    border: none; border-radius: 3px; background-color: #27272a; 
    text-align: center; color: transparent; height: 6px;
}
QProgressBar::chunk { background-color: #3d5afe; border-radius: 3px; }
"""

class TubeStreamPro(QMainWindow):
    def __init__(self):
        super().__init__()
        self.setWindowTitle("TubeStream Pro")
        self.resize(1100, 750)
        self.setStyleSheet(MODERN_STYLE)
        self.download_mode = "playlist" # Default

        # Main Layout
        central_widget = QWidget()
        self.setCentralWidget(central_widget)
        self.main_layout = QHBoxLayout(central_widget)
        self.main_layout.setContentsMargins(0, 0, 0, 0)
        self.main_layout.setSpacing(0)

        self.init_sidebar()
        self.init_content()

    def init_sidebar(self):

        # Standard icon size for the sidebar
        icon_size = QSize(20, 20)

        self.sidebar = QFrame()
        self.sidebar.setObjectName("Sidebar")
        layout = QVBoxLayout(self.sidebar)
        
        logo = QLabel("⚡ TUBESTREAM PRO")
        logo.setObjectName("Logo")
        layout.addWidget(logo)

        # Nav Buttons
        self.btn_nav_dl = QPushButton("  Downloader")
        self.btn_nav_dl.setObjectName("NavBtn")
        self.btn_nav_dl.setIcon(QIcon("assets/download.svg"))
        self.btn_nav_dl.setProperty("active", "true")
        
        self.btn_nav_log = QPushButton("  System Logs")
        self.btn_nav_log.setObjectName("NavBtn")
        self.btn_nav_dl.setIcon(QIcon("assets/logs.svg"))
        self.btn_nav_log.setIconSize(icon_size)

        layout.addWidget(self.btn_nav_dl)
        layout.addWidget(self.btn_nav_log)
        layout.addStretch()
        
        # Mode Switcher in Sidebar
        mode_label = QLabel("DOWNLOAD MODE")
        mode_label.setStyleSheet("color: #52525b; font-size: 10px; font-weight: bold; margin-left: 20px;")
        layout.addWidget(mode_label)
        
        self.mode_btn = QPushButton("Current: Playlist")
        self.mode_btn.setObjectName("NavBtn")
        self.mode_btn.clicked.connect(self.toggle_mode)
        layout.addWidget(self.mode_btn)
        layout.addSpacing(20)

        self.main_layout.addWidget(self.sidebar)

    def init_content(self):
        self.stack = QStackedWidget()
        
        # --- PAGE 1: DOWNLOADER ---
        dl_page = QWidget()
        dl_layout = QVBoxLayout(dl_page)
        dl_layout.setContentsMargins(40, 40, 40, 40)
        dl_layout.setSpacing(25)

        welcome = QLabel("Start a New Download")
        welcome.setStyleSheet("color: white; font-size: 24px; font-weight: bold;")
        dl_layout.addWidget(welcome)

        # Input Section
        self.url_input = QLineEdit()
        self.url_input.setPlaceholderText("Paste YouTube Video or Playlist URL here...")
        dl_layout.addWidget(self.url_input)

        path_h = QHBoxLayout()
        self.path_input = QLineEdit()
        self.path_input.setPlaceholderText("Destination Folder...")
        btn_browse = QPushButton("📁 Browse")
        btn_browse.setStyleSheet("color: #3d5afe; background: transparent; font-weight: bold; border: none;")
        btn_browse.clicked.connect(self.select_folder)
        path_h.addWidget(self.path_input)
        path_h.addWidget(btn_browse)
        dl_layout.addLayout(path_h)

        # Task Card
        self.card = QFrame()
        self.card.setObjectName("Card")
        card_layout = QVBoxLayout(self.card)
        card_layout.setContentsMargins(20, 20, 20, 20)

        self.title_label = QLabel("No active task")
        self.title_label.setStyleSheet("color: #e4e4e7; font-size: 14px; font-weight: 500;")
        card_layout.addWidget(self.title_label)

        self.progress_bar = QProgressBar()
        card_layout.addWidget(self.progress_bar)

        self.status_label = QLabel("Status: Idle")
        self.status_label.setStyleSheet("color: #71717a; font-size: 12px;")
        card_layout.addWidget(self.status_label)
        
        dl_layout.addWidget(self.card)

        # Controls
        ctrl_h = QHBoxLayout()
        self.start_btn = QPushButton("START DOWNLOAD")
        self.start_btn.setObjectName("PrimaryBtn")
        self.start_btn.clicked.connect(self.begin_task)
        
        self.stop_btn = QPushButton("STOP")
        self.stop_btn.setStyleSheet("color: #ef4444; font-weight: bold; background: transparent; border: 1px solid #450a0a; border-radius: 8px; padding: 10px;")
        self.stop_btn.clicked.connect(self.stop_task)
        
        ctrl_h.addWidget(self.start_btn, 4)
        ctrl_h.addWidget(self.stop_btn, 1)
        dl_layout.addLayout(ctrl_h)
        dl_layout.addStretch()

        # --- PAGE 2: LOGS ---
        log_page = QWidget()
        log_layout = QVBoxLayout(log_page)
        self.log_window = QTextEdit()
        self.log_window.setReadOnly(True)
        self.log_window.setStyleSheet("background-color: #09090b; border: 1px solid #27272a; color: #a1a1aa; font-family: 'Consolas';")
        log_layout.addWidget(QLabel("System Diagnostics"))
        log_layout.addWidget(self.log_window)

        # Add pages to stack
        self.stack.addWidget(dl_page)
        self.stack.addWidget(log_page)
        self.main_layout.addWidget(self.stack)

        # Wire up Nav
        self.btn_nav_dl.clicked.connect(lambda: self.switch_page(0))
        self.btn_nav_log.clicked.connect(lambda: self.switch_page(1))

    def toggle_mode(self):
        if self.download_mode == "playlist":
            self.download_mode = "single"
            self.mode_btn.setText("Current: Single Video")
        else:
            self.download_mode = "playlist"
            self.mode_btn.setText("Current: Playlist")

    def switch_page(self, index):
        self.stack.setCurrentIndex(index)
        self.btn_nav_dl.setProperty("active", "true" if index == 0 else "false")
        self.btn_nav_log.setProperty("active", "true" if index == 1 else "false")
        self.style().unpolish(self.btn_nav_dl) # Refresh style
        self.style().polish(self.btn_nav_dl)
        self.style().unpolish(self.btn_nav_log)
        self.style().polish(self.btn_nav_log)

    def select_folder(self):
        folder = QFileDialog.getExistingDirectory(self, "Select Save Location")
        if folder: self.path_input.setText(folder)

    def begin_task(self):
        url, path = self.url_input.text().strip(), self.path_input.text().strip()
        if not url or not path: return

        self.start_btn.setEnabled(False)
        self.start_btn.setText("INITIALIZING...")
        
        self.worker = DownloadWorker(url, path, mode=self.download_mode)
        self.worker.progress_signal.connect(self.on_progress)
        self.worker.finished_signal.connect(self.on_finished)
        self.worker.error_signal.connect(self.on_error)
        self.worker.log_signal.connect(self.log_window.append)
        self.worker.title_signal.connect(self.title_label.setText)
        self.worker.start()

    def on_progress(self, d):
        if d['status'] == 'downloading':
            p = d.get('_percent_str', '0%').replace('%','')
            try:
                self.progress_bar.setValue(int(float(p)))
                self.status_label.setText(f"Speed: {d.get('_speed_str','N/A')} | ETA: {d.get('_eta_str','N/A')}")
            except: pass

    def on_finished(self, msg):
        self.start_btn.setEnabled(True)
        self.start_btn.setText("START DOWNLOAD")
        self.status_label.setText(f"✅ {msg}")

    def on_error(self, err):
        self.start_btn.setEnabled(True)
        self.start_btn.setText("START DOWNLOAD")
        self.status_label.setText(f"❌ Error: {err[:50]}...")

    def stop_task(self):
        if hasattr(self, 'worker'): self.worker.stop()

if __name__ == "__main__":
    app = QApplication(sys.argv)
    window = TubeStreamPro()
    window.show()
    sys.exit(app.exec())