Update app.py

Fix classify_text

app.py

@@ -60,21 +60,54 @@ def classify_text(text):
     Función para clasificar texto y detectar coloquialismos cubanos
     """
     if not text or text.strip() == "":
-        return "Por favor, ingresa un texto para clasificar."
+        return "⚠️ Por favor, ingresa un texto para clasificar."
     
     try:
+        # Obtener resultados del modelo
         results = classifier(text)
-        # Ordenar por score (de mayor a menor)
-        results.sort(key=lambda x: x['score'], reverse=True)
         
-        # Mostrar los 3 mejores resultados
-        output = "### 📊 Resultados de clasificación:\n\n"
-        for i, result in enumerate(results[:0], 1):
-            label = result['coloquial']
-            score = result['score']
-            output += f"{i}. **{label}**: {score:.2%}\n"
+        # El resultado es una lista de diccionarios con 'label' y 'score'
+        # Ejemplo: [{'label': 'LABEL_1', 'score': 0.99}, {'label': 'LABEL_0', 'score': 0.01}]
+        
+        # Obtener el resultado con mayor confianza
+        top_result = results[0]  # Ya viene ordenado por score (mayor a menor)
+        label = top_result['label']
+        score = top_result['score']
+        
+        # Determinar si es coloquial (LABEL_1 = coloquial, LABEL_0 = no coloquial)
+        if label == "LABEL_1":
+            es_coloquial = True
+            emoji = "🇨🇺"
+            tipo = "**COLOQUIAL CUBANO**"
+            descripcion = "✅ Este texto contiene expresiones típicas del dialecto cubano."
+        else:  # LABEL_0
+            es_coloquial = False
+            emoji = "📚"
+            tipo = "**NO COLOQUIAL / ESTÁNDAR**"
+            descripcion = "ℹ️ Este texto usa lenguaje estándar, sin expresiones coloquiales cubanas."
+        
+        # Construir la salida formateada
+        output = f"""
+### {emoji} Resultado: {tipo}
+
+**Confianza:** {score:.2%}
+
+{descripcion}
+
+---
+### 📊 Detalle de clasificación:
+
+| Etiqueta | Probabilidad |
+|----------|--------------|
+| 🇨🇺 Coloquial | {results[0]['score']:.2%} (LABEL_1) |
+| 📚 No coloquial | {results[1]['score']:.2%} (LABEL_0) |
+
+---
+💡 **Nota:** El modelo analiza si el texto contiene expresiones típicas del dialecto cubano.
+        """
         
         return output
+        
     except Exception as e:
         return f"❌ Error en clasificación: {str(e)}"