import os
import requests

HUGGINGFACE_API_TOKEN = os.environ.get("HF_TOKEN", "")
API_URL = "https://api-inference.huggingface.co/models/gpt2"

headers = {
    "Authorization": f"Bearer {HUGGINGFACE_API_TOKEN}"
}

def obtener_sugerencia_nlp(texto_comentario):
    payload = {
        "inputs": f"// {texto_comentario}\n# sugerido:",
        "parameters": {
            "max_new_tokens": 20,
            "temperature": 0.7,
            "return_full_text": False
        }
    }

    response = requests.post(API_URL, headers=headers, json=payload)
    if response.status_code == 200:
        resultado = response.json()
        return resultado[0]["generated_text"].strip()
    else:
        return f"(sin sugerencia: {response.status_code})"

def procesar_comentarios(codigo: str):
    lineas = codigo.splitlines()
    sugerencias = []
    for linea in lineas:
        if "//" in linea:
            comentario = linea.split("//", 1)[1].strip()
            sugerencia = obtener_sugerencia_nlp(comentario)
            sugerencias.append((comentario, sugerencia))
    return sugerencias