import openai
import os
from dotenv import load_dotenv

# Cargar variables de entorno
load_dotenv()

# Configurar la API Key de OpenAI
client = openai.OpenAI(api_key=os.getenv("OPENAI_API_KEY"))

def nombre_arreglado(nombre_entrada, retorno):
    prompt = f"""
    Devuelve solo el nombre de pila en formato título y el apellido:
    Solo si comienza con María, Juan, José o Luis, incluye el segundo nombre en caso contrario, deja solo un nombre.
    En otros casos, devuelve solo el primero.
    Si el texto es ofensivo o no válido, devuelve un valor vacío.
    Separado por | el nombre del apellido
    No añadas nada más.

    este es el nombre: {nombre_entrada}
    """

    try:
        response = client.chat.completions.create(
            model="gpt-4o-mini",
            messages=[{"role": "user", "content": prompt}]
        )
        
        resultado = response.choices[0].message.content.strip()
        
        if "|" in resultado:
            partes = resultado.split("|")
            nombre = partes[0].strip()
            apellido = partes[1].strip()
        else:
            nombre, apellido = "", ""

        return nombre if retorno == 0 else apellido

    except Exception as e:
        print(f"Error al procesar nombre: {e}")
        return ""

# Si el script es importado, no ejecuta nada
if __name__ == "__main__":
    print("Este script solo define la función `nombre_arreglado()`.")