Update app.py

app.py

@@ -1,31 +1,28 @@
 import gradio as gr
-from transformers import AutoTokenizer, AutoModel
+from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 import torch
 
-# Cargar modelo y tokenizer
-tokenizer = AutoTokenizer.from_pretrained("microsoft/codebert-base")
-model = AutoModel.from_pretrained("microsoft/codebert-base")
+# Cargar modelo y tokenizer de CodeT5
+tokenizer = AutoTokenizer.from_pretrained("Salesforce/codet5-base")
+model = AutoModelForSeq2SeqLM.from_pretrained("Salesforce/codet5-base")
 
-# Función para "analizar" código con CodeBERT
+# Función para generar explicación del código con CodeT5
 def analizar_codigo(codigo):
     errores = "⚠️ No se realizó análisis de sintaxis (solo compatible con Python)."
-    
     try:
-        # Obtener embeddings del código
-        inputs = tokenizer(codigo, return_tensors="pt", truncation=True, max_length=512)
+        # Preparamos el prompt para explicación (puedes ajustar el prompt según la tarea)
+        prompt = f"Explain this JavaScript code:\n{codigo}\n"
+        inputs = tokenizer(prompt, return_tensors="pt", truncation=True, max_length=512)
         with torch.no_grad():
-            outputs = model(**inputs)
-
-        # Simulación de análisis lógico: mostrar dimensiones del embedding
-        emb_shape = outputs.last_hidden_state.shape
-        explicacion = (
-            f"✅ El código fue procesado correctamente con CodeBERT.\n\n"
-            f"Embedding shape: {emb_shape}\n"
-            f"(usa estos embeddings para tareas de clasificación, detección de errores, etc.)"
-        )
+            summary_ids = model.generate(
+                inputs.input_ids,
+                max_length=128,
+                num_beams=4,
+                early_stopping=True
+            )
+        explicacion = tokenizer.decode(summary_ids[0], skip_special_tokens=True)
     except Exception as e:
         explicacion = f"❌ Error al analizar el código: {str(e)}"
-
     return errores, explicacion
 
 # Interfaz con Gradio
@@ -34,11 +31,11 @@ demo = gr.Interface(
     inputs=gr.Textbox(lines=15, label="Pega tu código JavaScript aquí"),
     outputs=[
         gr.Textbox(label="Estado de la sintaxis"),
-        gr.Textbox(label="Análisis semántico (embeddings de CodeBERT)")
+        gr.Textbox(label="Explicación generada por CodeT5")
     ],
-    title="🔍 Analizador básico de código JavaScript con CodeBERT",
+    title="🔍 Explicador de código JavaScript con CodeT5",
     description=(
-        "Este Space procesa tu código JavaScript con CodeBERT para obtener representaciones internas (embeddings). "
+        "Este Space utiliza CodeT5 para generar explicaciones automáticas de tu código JavaScript. "
         "No se realiza análisis de sintaxis ni generación textual automática."
     )
 )