Update main.py

main.py

@@ -1,42 +1,50 @@
+import os
 import json
+import requests
 from lexer import lexer
 from parser import Parser
 from semantico import AnalizadorSemantico
 from codigo_intermedio import GeneradorIntermedio
 from sugerencias_nlp import procesar_comentarios
 
-# Función temporal para generar sugerencias tipo NLP (puede reemplazarse por API real)
-def sugerencia_dummy(mensaje):
-    return f"(sugerencia generada para: {mensaje})"
+HF_TOKEN = os.environ.get("HF_TOKEN", "")
+HEADERS = {"Authorization": f"Bearer {HF_TOKEN}"}
+API_URL = "https://api-inference.huggingface.co/models/gpt2"
+
+def sugerencia_nlp_error(error_msg):
+    payload = {
+        "inputs": f"ERROR: {error_msg}\nSUGERENCIA:",
+        "parameters": {
+            "max_new_tokens": 30,
+            "temperature": 0.8,
+            "return_full_text": False
+        }
+    }
+    response = requests.post(API_URL, headers=HEADERS, json=payload)
+    if response.status_code == 200:
+        return response.json()[0]["generated_text"].strip()
+    return f"(sin sugerencia: {response.status_code})"
 
 def main():
     with open("entrada.txt", "r", encoding="utf-8") as f:
         codigo = f.read()
 
-    # Fase léxica
     tokens = lexer(codigo)
-
-    # Fase sintáctica
     parser = Parser(tokens)
     ast = parser.parse()
-
-    # Fase semántica
     semantico = AnalizadorSemantico(ast)
     resultado = semantico.analizar()
 
-    # Agregar sugerencias a errores
     errores_ext = [
-        {"mensaje": err, "sugerencia": sugerencia_dummy(err)}
+        {"mensaje": err, "sugerencia": sugerencia_nlp_error(err)}
         for err in resultado["errores_semanticos"]
     ]
 
-    # Extraer comentarios y sugerencias NLP
     comentarios_ext = [
         {"comentario": c, "sugerencia": s}
         for c, s in procesar_comentarios(codigo)
     ]
 
-    # Análisis completo combinado
     analisis_completo = {
         "variables_declaradas": resultado["variables_declaradas"],
         "errores_semanticos": errores_ext,
@@ -46,7 +54,6 @@ def main():
     with open("analisis.json", "w", encoding="utf-8") as f:
         json.dump(analisis_completo, f, indent=2)
 
-    # Código intermedio
     generador = GeneradorIntermedio()
     intermedio = generador.generar(ast)