import gradio as gr
import logging
from app.services import sticker_service
from app.gradio_formatter import gradio_formatter

# Configure logging
logging.basicConfig(
    level=logging.INFO,
    format='%(asctime)s %(levelname)s %(message)s'
)
logger = logging.getLogger(__name__)


class StickerUI:
    """Main class for the Sticker Search UI application."""
    
    def __init__(self):
        self.demo = None
        self._initialize_components()
    
    def _initialize_components(self):
        """Initialize all UI components."""
        # Search Tab
        self.search_input = gr.Textbox(label="Search keywords")
        self.limit_input = gr.Slider(1, 10, value=4, step=1, label="Max results")
        self.search_button = gr.Button("Search")
        self.deepseek_button = gr.Button("AI Search")
        self.search_results = gr.Dataframe(
            headers=["Preview", "Relevance", "Description", "Filename"],
            datatype=["markdown", "number", "str", "str"],
            row_count=("dynamic")
        )

        self.ai_search_results = gr.Dataframe(
            headers=["Preview", "AI-Score", "AI-Reason", "Description", "Filename"],
            datatype=["markdown", "number", "str", "str", "str"],
            row_count=("dynamic")
        )
        
        # Upload Tab
        self.image_input = gr.Image(type="filepath")
        self.title_input = gr.Textbox(label="Title")
        self.tags_input = gr.Textbox(label="Tags (comma separated)")
        self.desc_input = gr.Textbox(label="Description")
        self.upload_button = gr.Button("Upload")
        self.upload_output = gr.Textbox(label="Status")
        
        # All Stickers Tab
        self.refresh_button = gr.Button("Refresh")
        self.stickers_table = gr.Dataframe(
            headers=["ID", "Preview", "Title", "Description", "Tags", "Filename", "Hash"],
            datatype=["str", "markdown", "str", "str", "str", "str", "str"]
        )
    
    def _setup_search_tab(self):
        """Configure the search tab layout and functionality."""
        with gr.Tab("Search"):
            with gr.Row():
                self.search_input.render()
                self.limit_input.render()
            
            with gr.Row():
                self.search_button.render()
                self.deepseek_button.render()
            
            self.search_results.render()
            self.ai_search_results.render()
            
            # Event handlers
            self.search_button.click(
                fn=lambda q, l: gradio_formatter.format_search_results(
                    sticker_service.search_stickers(q, l)
                ),
                inputs=[self.search_input, self.limit_input],
                outputs=self.search_results
            )
            
            self.deepseek_button.click(
                fn=lambda q, l: gradio_formatter.format_ai_search_results(
                    sticker_service.search_stickers(q, l, reranking=True)
                ),
                inputs=[self.search_input, self.limit_input],
                outputs=self.ai_search_results
            )
    
    def _setup_upload_tab(self):
        """Configure the upload tab layout and functionality."""
        with gr.Tab("Upload"):
            with gr.Row():
                self.image_input.render()
            
            with gr.Row():
                self.title_input.render()
                self.tags_input.render()
            
            with gr.Row():
                self.desc_input.render()
                self.upload_button.render()
            
            self.upload_output.render()
            
            self.upload_button.click(
                fn=sticker_service.upload_sticker,
                inputs=[
                    self.image_input,
                    self.title_input,
                    self.desc_input,
                    self.tags_input
                ],
                outputs=self.upload_output
            )
    
    def _setup_all_stickers_tab(self):
        """Configure the all stickers tab layout and functionality."""
        with gr.Tab("All Stickers"):
            self.refresh_button.render()
            self.stickers_table.render()
            
            self.refresh_button.click(
                fn=sticker_service.get_all_stickers,
                outputs=self.stickers_table
            )
    
    def create_ui(self):
        """Create and configure the complete UI."""
        with gr.Blocks(title="Neko Sticker Search 🔍") as self.demo:
            self._setup_search_tab()
            self._setup_upload_tab()
            self._setup_all_stickers_tab()
        
        return self.demo