import tkinter as tk
from tkinter import ttk
import tkinter.simpledialog as sd
import time
import os
from utils import search_youtube, download_audio

class UIEngine:
    def __init__(self, db):
        self.db = db
        self.root = tk.Tk()
        self.root.title("celaigia")
        self.root.geometry("400x300")  # Set a fixed size

        self.status_label = tk.Label(self.root, text="Ready")
        self.status_label.pack()

        self.search_frame = ttk.Frame(self.root)
        self.search_frame.pack()

        self.search_status_label = tk.Label(self.search_frame, text="Enter search query:")
        self.search_status_label.grid(row=0, column=0)
        self.search_entry = tk.Entry(self.search_frame)
        self.search_entry.grid(row=0, column=1)
        self.search_button = tk.Button(self.search_frame, text="Search", command=lambda: self.search_and_display())
        self.search_button.grid(row=0, column=2)

        self.results_frame = ttk.Frame(self.root)
        self.results_frame.pack()

        self.download_status_label = tk.Label(self.root, text="Enter URL to download:")
        self.download_status_label.pack()
        self.download_entry = tk.Entry(self.root)
        self.download_entry.pack()
        self.download_button = tk.Button(self.root, text="Download", command=lambda: self.handle_download(self.download_entry.get()))
        self.download_button.pack()

    def search_and_display(self):
        self.set_status("searching")
        query = self.search_entry.get()
        # Clear the existing results
        for widget in self.results_frame.winfo_children():
            widget.destroy()
        results = search_youtube(query)
        self.display_results(results)
        self.set_status("Enter search query or select item to download")

    def set_status(self, status):
        self.status_label.config(text=status)
        self.root.update()  # Force GUI update

    def display_results(self, results):
        for i, result in enumerate(results):
            result_button = ttk.Button(self.results_frame, text=result['title'], command=lambda url=result['webpage_url'], filename=result['title']: self.handle_download(url, filename))
            result_button.pack()

    def handle_download(self, url, filename):

        if self.db.downloadable(url, filename):
            self.set_status("Downloading...")
            success = download_audio(url, self.db.base_path)
            if success:
                self.set_status("Download completed, adding to db")
                self.prompt_user_for_details(url, filename)
                self.set_status("Enter search query or select item to download")

    # Modify the prompt_user_for_details method
    def prompt_user_for_details(self,url, filename):
        title = sd.askstring("Enter Title", "Enter Title:", initialvalue=filename)
        artist = sd.askstring("Enter Artist", "Enter Artist:")
        tags = sd.askstring("Enter Tags", "Enter Tags (comma separated):")
        entry = {
            "id": url.split('https://www.youtube.com/watch?v=')[-1].strip(),
            "title": title.strip(),
            "artist": artist.strip(),
            "filename": filename.strip(),
            "tags": [tag.strip() for tag in tags.split(",")] if tags else [],
            "timestamp": time.time()
        }
        self.add_to_db(entry)

    def add_to_db(self, entry):
        self.db.add_entry(entry)

    def run(self):
        self.root.mainloop()