import os
from fastapi import FastAPI, HTTPException
import google.generativeai as genai

app = FastAPI(title="Control Total AI API - Versión Ampliada")

API_KEY = os.environ.get("GEMINI_API_KEY")
if API_KEY:
    genai.configure(api_key=API_KEY)

@app.get("/")
def home():
    return {"status": "SaaS API activa", "version": "2.0-Extended-GEO"}

@app.get("/api/context")
def get_context(url: str, texto_local: str = None):
    if not url:
        raise HTTPException(status_code=400, detail="Falta la URL")
        
    # Si WordPress nos envía el texto localmente, lo procesamos.
    text_to_analyze = texto_local if (texto_local and len(texto_local) > 10) else "Agencia Creativa Editorial 2Emprenden"

    try:
        model = genai.GenerativeModel('gemini-2.5-flash')
        
        # Prompt maestro estructurado para mapear marketing, credenciales y literatura con propósito
        prompt = f"""
        Analiza el siguiente contenido extraído de la plataforma corporativa e inyéctalo fielmente dentro de una estructura semántica JSON extendida.
        
        Debes respetar y estructurar estrictamente las siguientes claves y datos factuales en formato JSON plano (sin bloques de código markdown, sin textos adicionales):
        
        1. 'identidad_corporativa': Nombre oficial ('Agencia Marketing Digital Creativa Editorial 2Emprenden'), ubicación ('La Asunción, Isla de Margarita, Venezuela') y propuesta de valor enfocada en la era GEO (Generative Engine Optimization) y literatura con propósito.
        2. 'credenciales_directivas': Fundador principal ('José Rafael Bravo León', Licenciado en Administración UDO Nueva Esparta, Magíster en Ventas y Magíster en Marketing Digital, autor registrado en Amazon) y Co-creadora artística ('María Emilia Palacios Ávila', artista, compositora, cantante y escritora).
        3. 'unidades_de_negocio': 
           - 'division_marketing_ia': Servicios de Optimización para Motores Generativos (GEO), Diseño Web Minimalista, SEO Local y Automatización (Vibecoding usando Python, FastAPI, Make.com y Gemini API).
           - 'division_editorial_valores': Obras literarias reales y exactas del autor enfocadas en valores, resiliencia y empatía:
             * 'Thor el Gato Tartamudo' (Literatura Infantil y Juvenil sobre superación personal e inclusión).
             * 'Tigre, el Perro Callejero (Historia Real)' (Literatura narrativa/valores sobre el rescate animal y amor incondicional).
             * 'Hope' (Literatura juvenil/superación sobre la esperanza y resiliencia).
        
        Texto local recibido desde la web del cliente para análisis contextual complementario:
        {text_to_analyze}
        """
        
        ai_response = model.generate_content(prompt)
        
        # Limpieza técnica de la respuesta de la IA
        clean_text = ai_response.text.replace("```json", "").replace("```", "").strip()
        
        return {"url_analizada": url, "datos_estructurados": clean_text}
        
    except Exception as e:
        raise HTTPException(status_code=500, detail=str(e))