src/streamlit_app.py

"""
Streamlit web interface for the Logo Downloader
"""

import os
import logging
import streamlit as st
from pathlib import Path
from typing import Optional

from services.logo_downloader import LogoDownloader
from services.appconfig import GEMINI_API_KEY, DEFAULT_LOGOS_PER_ENTITY, MAX_LOGOS_PER_ENTITY

# Setup logging
logging.basicConfig(level=logging.INFO)
logger = logging.getLogger(__name__)


def process_text_request(text: str, api_key: Optional[str], num_logos: int = DEFAULT_LOGOS_PER_ENTITY):
    """
    Process text and download logos through Streamlit interface
    """
    try:
        if not text or not text.strip():
            return "❌ Please provide some text to analyze.", None, "No text provided."

        if num_logos < 1 or num_logos > MAX_LOGOS_PER_ENTITY:
            return f"❌ Number of logos must be between 1 and {MAX_LOGOS_PER_ENTITY}.", None, f"Invalid number: {num_logos}"

        final_api_key = api_key.strip() if api_key and api_key.strip() else GEMINI_API_KEY

        downloader = LogoDownloader(gemini_api_key=final_api_key)
        results = downloader.process_text(text, num_logos)

        if results['status'] == 'success' and results['stats']['total_downloads'] > 0:
            status_msg = f"✅ {downloader.get_stats_summary()}"
            zip_path = results.get('zip_path')
            detailed_results = _format_detailed_results(results)
            return status_msg, zip_path, detailed_results

        elif results['status'] == 'warning':
            return f"⚠️ {results['message']}", None, results.get('message', 'No details available')

        else:
            return f"❌ Processing failed: {results['message']}", None, results.get('message', 'Unknown error')

    except Exception as e:
        logger.error(f"Error in process_text_request: {e}")
        return f"❌ An error occurred: {str(e)}", None, f"Error details: {str(e)}"


def _format_detailed_results(results):
    if not results.get('results'):
        return "No detailed results available."

    details = []
    details.append(f"📊 **Processing Summary:**")
    details.append(f"- Total entities found: {results['stats']['total_entities']}")
    details.append(f"- Total logos downloaded: {results['stats']['total_downloads']}")
    details.append(f"- Successful entities: {results['stats']['successful_entities']}")
    details.append(f"- Failed entities: {results['stats']['failed_entities']}")
    details.append("")
    details.append("📋 **Entity Details:**")

    for result in results['results']:
        entity = result['entity']
        count = result['downloaded_count']
        if count > 0:
            details.append(f"✅ **{entity}**: {count} logos downloaded")
        else:
            error_msg = result.get('error', 'No logos found')
            details.append(f"❌ **{entity}**: Failed ({error_msg})")

    return "\n".join(details)


def main():
    st.set_page_config(page_title="🎨 Logo Downloader", layout="centered")
    st.title("🎨 Logo Downloader")
    st.markdown("Extract entities from text and download their logos automatically.")

    with st.form(key="logo_form"):
        text_input = st.text_area(
            "📝 Enter text containing company names, products, or brands:",
            placeholder="e.g., We use AWS, Docker, React, and Adobe Photoshop for our projects",
            height=150
        )

        api_key_input = st.text_input(
            "🔑 Gemini API Key (optional)",
            type="password",
            placeholder="Enter your Gemini API key for enhanced extraction"
        )

        num_logos_input = st.slider(
            "🖼️ Logos per entity",
            min_value=1,
            max_value=MAX_LOGOS_PER_ENTITY,
            value=DEFAULT_LOGOS_PER_ENTITY,
            step=1
        )

        submit_btn = st.form_submit_button("🚀 Download Logos")

    if submit_btn:
        with st.spinner("Processing logos..."):
            status_msg, zip_path, detailed_results = process_text_request(
                text_input,
                api_key_input,
                num_logos_input
            )
        st.markdown(status_msg)

        if zip_path and Path(zip_path).exists():
            with open(zip_path, "rb") as f:
                st.download_button(
                    label="📥 Download Logos ZIP",
                    data=f,
                    file_name=Path(zip_path).name,
                    mime="application/zip"
                )

        st.markdown("## 📊 Detailed Results")
        st.markdown(detailed_results)

    st.markdown("---")
    st.info("💡 Tip: Get a free Gemini API key at [Google AI Studio](https://makersuite.google.com/app/apikey) for better extraction accuracy.")

    st.markdown("## 💡 Examples")
    examples = [
        "Our tech stack includes React, Node.js, MongoDB, Docker, AWS, and we use Figma for design, along with GitHub for version control.",
        "The team uses Microsoft Office, Adobe Creative Suite, Slack for communication, Zoom for meetings, and Salesforce for CRM.",
        "Popular social media platforms like Instagram, TikTok, Twitter, LinkedIn, and YouTube are essential for digital marketing."
    ]

    for ex in examples:
        if st.button(f"Use example: {ex[:50]}..."):
            st.session_state["text_input"] = ex


if __name__ == "__main__":
    main()