Upload 2 files

app.py

@@ -4,7 +4,7 @@ import chess.pgn
 import matplotlib.pyplot as plt
 import numpy as np
 import io
-import base64
+import re
 from io import BytesIO
 
 class AnalizadorAjedrez:
@@ -99,19 +99,26 @@ class AnalizadorAjedrez:
         jugadas = [d['jugada'] for d in self.datos_jugadas]
         evaluaciones = [d['evaluacion'] for d in self.datos_jugadas]
         
-        ax1.plot(jugadas, evaluaciones, 'b-', linewidth=2, alpha=0.7)
-        ax1.fill_between(jugadas, evaluaciones, alpha=0.3)
-        ax1.axhline(y=0, color='black', linestyle='-', alpha=0.5)
-        ax1.set_title('📊 EVALUACIÓN DE POSICIÓN', fontweight='bold')
+        # Crear fondo coloreado para ventaja
+        ax1.axhspan(0.1, max(evaluaciones) if max(evaluaciones) > 0 else 0.1, 
+                   alpha=0.2, color='blue', label='Ventaja Blancas')
+        ax1.axhspan(min(evaluaciones) if min(evaluaciones) < 0 else -0.1, -0.1, 
+                   alpha=0.2, color='red', label='Ventaja Negras')
+        
+        ax1.plot(jugadas, evaluaciones, 'b-', linewidth=2.5, alpha=0.8, marker='o', markersize=3)
+        ax1.axhline(y=0, color='black', linestyle='-', alpha=0.5, linewidth=1)
+        ax1.set_title('📊 EVALUACIÓN DE POSICIÓN', fontweight='bold', fontsize=14)
         ax1.set_xlabel('Número de Jugada')
         ax1.set_ylabel('Ventaja para Blancas')
         ax1.grid(True, alpha=0.3)
+        ax1.legend()
 
         # 📈 2. ANÁLISIS DE MATERIAL
         material_data = [d['ventaja_material'] for d in self.datos_jugadas]
-        ax2.plot(jugadas, material_data, 'r-', linewidth=2, alpha=0.7)
-        ax2.axhline(y=0, color='black', linestyle='-', alpha=0.5)
-        ax2.set_title('📈 VENTAJA MATERIAL', fontweight='bold')
+        ax2.bar(jugadas, material_data, alpha=0.7, 
+               color=['blue' if x >= 0 else 'red' for x in material_data])
+        ax2.axhline(y=0, color='black', linestyle='-', alpha=0.5, linewidth=1)
+        ax2.set_title('📈 VENTAJA MATERIAL', fontweight='bold', fontsize=14)
         ax2.set_xlabel('Jugada')
         ax2.set_ylabel('Ventaja Material')
         ax2.grid(True, alpha=0.3)
@@ -120,9 +127,11 @@ class AnalizadorAjedrez:
         movilidad_b = [d['movilidad_blancas'] for d in self.datos_jugadas]
         movilidad_n = [d['movilidad_negras'] for d in self.datos_jugadas]
         
-        ax3.plot(jugadas, movilidad_b, 'b-', label=f'{blancas}', linewidth=2)
-        ax3.plot(jugadas, movilidad_n, 'r-', label=f'{negras}', linewidth=2)
-        ax3.set_title('🎯 MOVILIDAD DE PIEZAS', fontweight='bold')
+        ax3.fill_between(jugadas, movilidad_b, alpha=0.3, color='blue')
+        ax3.fill_between(jugadas, movilidad_n, alpha=0.3, color='red')
+        ax3.plot(jugadas, movilidad_b, 'b-', label=f'{blancas}', linewidth=2.5)
+        ax3.plot(jugadas, movilidad_n, 'r-', label=f'{negras}', linewidth=2.5)
+        ax3.set_title('🎯 MOVILIDAD DE PIEZAS', fontweight='bold', fontsize=14)
         ax3.set_xlabel('Jugada')
         ax3.set_ylabel('Movimientos Legales')
         ax3.legend()
@@ -135,12 +144,24 @@ class AnalizadorAjedrez:
         valores_n = [ultimo['material_negras'], ultimo['movilidad_negras']]
         
         x = np.arange(len(categorias))
-        ax4.bar(x - 0.2, valores_b, 0.4, label=blancas, color='blue', alpha=0.7)
-        ax4.bar(x + 0.2, valores_n, 0.4, label=negras, color='red', alpha=0.7)
-        ax4.set_title('📋 COMPARACIÓN FINAL', fontweight='bold')
+        bar_width = 0.35
+        
+        ax4.bar(x - bar_width/2, valores_b, bar_width, label=blancas, 
+               color='blue', alpha=0.7, edgecolor='black')
+        ax4.bar(x + bar_width/2, valores_n, bar_width, label=negras, 
+               color='red', alpha=0.7, edgecolor='black')
+        
+        # Añadir valores encima de las barras
+        for i, v in enumerate(valores_b):
+            ax4.text(i - bar_width/2, v + 0.5, str(v), ha='center', va='bottom', fontweight='bold')
+        for i, v in enumerate(valores_n):
+            ax4.text(i + bar_width/2, v + 0.5, str(v), ha='center', va='bottom', fontweight='bold')
+            
+        ax4.set_title('📋 COMPARACIÓN FINAL', fontweight='bold', fontsize=14)
         ax4.set_xticks(x)
         ax4.set_xticklabels(categorias)
         ax4.legend()
+        ax4.grid(True, alpha=0.3, axis='y')
 
         plt.tight_layout()
         return fig
@@ -151,113 +172,229 @@ class AnalizadorAjedrez:
             return "No hay datos para generar reporte"
             
         reporte = "🎯 ANÁLISIS DETALLADO DE LA PARTIDA\n"
-        reporte += "=" * 40 + "\n"
+        reporte += "=" * 50 + "\n\n"
         
         # Estadísticas básicas
         total_jugadas = len(self.datos_jugadas)
         eval_final = self.datos_jugadas[-1]['evaluacion']
         
-        reporte += f"• Total de jugadas: {total_jugadas}\n"
-        reporte += f"• Evaluación final: {eval_final:.2f}\n"
+        reporte += f"📊 **Estadísticas Básicas:**\n"
+        reporte += f"   • Total de jugadas: {total_jugadas}\n"
+        reporte += f"   • Evaluación final: {eval_final:.2f}\n"
         
         # Balance de material final
         mat_final = self.datos_jugadas[-1]['ventaja_material']
         if mat_final > 2:
-            reporte += f"• Ventaja material final: +{mat_final} para Blancas\n"
+            reporte += f"   • ⚖️ Ventaja material final: +{mat_final} para Blancas\n"
         elif mat_final < -2:
-            reporte += f"• Ventaja material final: +{abs(mat_final)} para Negras\n"
+            reporte += f"   • ⚖️ Ventaja material final: +{abs(mat_final)} para Negras\n"
         else:
-            reporte += "• Material equilibrado al final\n"
+            reporte += "   • ⚖️ Material equilibrado al final\n"
             
         # Jugador con mejor movilidad promedio
         mov_avg_b = np.mean([d['movilidad_blancas'] for d in self.datos_jugadas])
         mov_avg_n = np.mean([d['movilidad_negras'] for d in self.datos_jugadas])
         
+        reporte += f"\n🎯 **Análisis de Movilidad:**\n"
         if mov_avg_b > mov_avg_n:
-            reporte += f"• Mejor movilidad promedio: Blancas ({mov_avg_b:.1f} vs {mov_avg_n:.1f})\n"
+            reporte += f"   • Mejor movilidad promedio: 🏆 Blancas ({mov_avg_b:.1f} vs {mov_avg_n:.1f})\n"
+        else:
+            reporte += f"   • Mejor movilidad promedio: 🏆 Negras ({mov_avg_n:.1f} vs {mov_avg_b:.1f})\n"
+        
+        # Punto de mayor ventaja
+        max_ventaja_b = max([d['evaluacion'] for d in self.datos_jugadas])
+        min_ventaja_b = min([d['evaluacion'] for d in self.datos_jugadas])
+        
+        reporte += f"\n📈 **Momentos Clave:**\n"
+        reporte += f"   • Mayor ventaja de Blancas: +{max_ventaja_b:.1f}\n"
+        reporte += f"   • Mayor ventaja de Negras: +{abs(min_ventaja_b):.1f}\n"
+        
+        # Duración de la partida
+        if total_jugadas < 30:
+            reporte += f"   • Partida rápida ({total_jugadas} jugadas)\n"
+        elif total_jugadas < 60:
+            reporte += f"   • Partida de duración media ({total_jugadas} jugadas)\n"
         else:
-            reporte += f"• Mejor movilidad promedio: Negras ({mov_avg_n:.1f} vs {mov_avg_b:.1f})\n"
+            reporte += f"   • Partida larga ({total_jugadas} jugadas)\n"
             
         return reporte
 
+def limpiar_y_corregir_pgn(pgn_text):
+    """Limpia y corrige errores comunes en PGN automáticamente"""
+    correcciones = {
+        'N75': 'Nf3', 'N76': 'Nf6', 'N15': 'Nf5', 'N16': 'Nf6',
+        'Re6': 'Nc6', 'Re3': 'Ne3', 'Re4': 'Ne4',
+        'Rel': 'Re1', 'Ral': 'Ra1', 'Rbl': 'Rb1', 'Rcl': 'Rc1', 'Rdl': 'Rd1',
+        'Nba': 'Nb8', 'Nbc': 'Nbc6', 'Nbd': 'Nbd7',
+        'Of8': 'Qf8', 'Oc7': 'Qc7', 'Od8': 'Qd8',
+        'B6': 'Bf6', 'B7': 'Bb7', 'B5': 'Bb5',
+        'ae': 'a6', 'af': 'a5'
+    }
+    
+    # Aplicar correcciones
+    for error, correccion in correcciones.items():
+        pgn_text = pgn_text.replace(error, correccion)
+    
+    # Corregir números de jugada mal formados
+    pgn_text = re.sub(r'(\d+)\.([A-Z])', r'\1. \2', pgn_text)
+    
+    # Asegurar espacios después de puntos
+    pgn_text = re.sub(r'(\d+)\.(\S)', r'\1. \2', pgn_text)
+    
+    return pgn_text
+
 def analizar_y_mostrar(pgn_text):
     """Función que Gradio usará para la interfaz"""
     try:
         if not pgn_text.strip():
             return None, "❌ Por favor, pega una partida PGN válida"
             
+        # Limpiar y corregir PGN automáticamente
+        pgn_limpio = limpiar_y_corregir_pgn(pgn_text)
+        
         analizador = AnalizadorAjedrez()
-        blancas, negras, resultado, error = analizador.analizar_partida(pgn_text)
+        blancas, negras, resultado, error = analizador.analizar_partida(pgn_limpio)
         
-        if error:
-            return None, f"❌ {error}"
+        if "Error" in blancas:
+            # Intentar análisis con PGN original como fallback
+            try:
+                analizador = AnalizadorAjedrez()
+                blancas, negras, resultado, error = analizador.analizar_partida(pgn_text)
+                if "Error" in blancas:
+                    return None, f"❌ No se pudo analizar el PGN.\n\n💡 **Sugerencias:**\n• Verifica que la notación sea correcta\n• Asegúrate de que los movimientos sigan formato estándar\n• Revisa que no haya caracteres especiales extraños\n\nError técnico: {error}"
+            except:
+                return None, f"❌ Error crítico en el PGN.\n\n📝 **Formato esperado:**\n```\n[Event \"Nombre\"]\n[White \"JugadorB\"]\n[Black \"JugadorN\"]\n\n1. e4 e5 2. Nf3 Nc6 3. Bb5 a6\n```"
         
         # Generar gráficos
         fig = analizador.generar_graficos_gradio(blancas, negras)
         
         # Convertir figura a imagen para Gradio
         buf = BytesIO()
-        fig.savefig(buf, format='png', dpi=100, bbox_inches='tight')
+        fig.savefig(buf, format='png', dpi=120, bbox_inches='tight', facecolor='white')
         buf.seek(0)
         plt.close(fig)
         
         # Generar reporte textual
         reporte = analizador.generar_reporte_textual()
-        reporte = f"Partida: {blancas} vs {negras}\nResultado: {resultado}\n\n" + reporte
+        reporte_header = f"**♟️ PARTIDA ANALIZADA:**\n"
+        reporte_header += f"**Blancas:** {blancas}\n"
+        reporte_header += f"**Negras:** {negras}\n" 
+        reporte_header += f"**Resultado:** {resultado}\n\n"
+        
+        full_reporte = reporte_header + reporte
         
-        return buf, reporte
+        return buf, full_reporte
         
     except Exception as e:
-        return None, f"❌ Error inesperado: {str(e)}"
+        return None, f"❌ Error inesperado: {str(e)}\n\nPor favor, verifica el formato de tu PGN."
 
-# Interfaz de Gradio
-with gr.Blocks(theme=gr.themes.Soft()) as demo:
-    gr.Markdown("# ♟️ Analizador de Partidas de Ajedrez")
-    gr.Markdown("Pega cualquier partida en formato PGN y obtén un análisis completo con gráficos y estadísticas.")
+# Interfaz de Gradio mejorada
+with gr.Blocks(theme=gr.themes.Soft(primary_hue="blue", secondary_hue="slate")) as demo:
+    gr.Markdown("""
+    # ♟️ Analizador Avanzado de Partidas de Ajedrez
+    
+    **Pega cualquier partida en formato PGN y obtén un análisis completo con gráficos y estadísticas detalladas.**
+    
+    *El sistema corrige automáticamente errores comunes de notación.*
+    """)
     
     with gr.Row():
-        with gr.Column():
+        with gr.Column(scale=1):
+            gr.Markdown("### 📝 Ingresa tu partida")
             pgn_input = gr.Textbox(
                 label="Pega tu partida PGN aquí",
-                lines=15,
+                lines=18,
                 placeholder="""[Event \"Mi Torneo\"]
 [White \"Jugador Blanco\"]
 [Black \"Jugador Negro\"]
 [Result \"1-0\"]
 
-1. e4 e5 2. Nf3 Nc6 3. Bb5 a6 4. Ba4 Nf6 5. O-O Be7 6. Re1 b5 7. Bb3 d6 8. c3 O-O"""
+1. e4 e5 2. Nf3 Nc6 3. Bb5 a6 4. Ba4 Nf6 5. O-O Be7 6. Re1 b5 7. Bb3 d6 8. c3 O-O
+9. h3 Nb8 10. d4 Nbd7 11. Nbd2 Bb7 12. Bc2 Re8 13. Nf1 Bf8 14. Ng3 g6 15. a4 c5
+16. d5 c4 17. Bg5 h6 18. Be3 Nc5 19. Qd2 h5 20. Bh6 Nh7 21. g3 Bd7 22. Rf1 Qc7
+23. Ne2 Rec8 24. Ng5 Nxg5 25. Bxg5 Bg7 26. f4 exf4 27. gxf4 f6 28. Bf3 Be5
+29. fxe5 fxe5 30. Bg5 Qd7 31. e6 Qe7 32. Bf6 Qf8 33. e7 1-0""",
+                info="Puedes pegar PGNs con pequeños errores, el sistema los corregirá automáticamente."
             )
-            btn_analizar = gr.Button("Analizar Partida", variant="primary")
             
-        with gr.Column():
-            image_output = gr.Image(label="Análisis Gráfico de la Partida", type="filepath")
-            text_output = gr.Textbox(label="Reporte de Análisis", lines=8)
-    
-    # Ejemplos de PGN
-    gr.Markdown("### 📋 Ejemplo de formato PGN:")
-    gr.Examples(
-        examples=[[
-"""[Event "Ejemplo"]
-[Site "?"]
-[Date "2023.??.??"]
-[Round "?"]
-[White "Carlsen"]
-[Black "Caruana"]
+            with gr.Row():
+                btn_analizar = gr.Button("🧠 Analizar Partida", variant="primary", size="lg")
+                btn_limpiar = gr.Button("🗑️ Limpiar", variant="secondary")
+            
+            gr.Markdown("### 💡 Ejemplos rápidos:")
+            with gr.Row():
+                gr.Examples(
+                    examples=[[
+"""[Event "Partida Rápida"]
+[White "Aman"]
+[Black "Boris"]
 [Result "1-0"]
 
-1. e4 e5 2. Nf3 Nc6 3. Bb5 a6 4. Ba4 Nf6 5. O-O Be7 6. Re1 b5 7. Bb3 d6 8. c3 O-O 
-9. h3 Nb8 10. d4 Nbd7 11. Nbd2 Bb7 12. Bc2 Re8 13. Nf1 Bf8 14. Ng3 g6 15. a4 c5 
-16. d5 c4 17. Bg5 h6 18. Be3 Nc5 19. Qd2 h5 20. Bh6 Nh7 21. g3 Bd7 22. Rf1 Qc7 
-23. Ne2 Rec8 24. Ng5 Nxg5 25. Bxg5 Bg7 26. f4 exf4 27. gxf4 f6 28. Bf3 Be5 
-29. fxe5 fxe5 30. Bg5 Qd7 31. e6 Qe7 32. Bf6 Qf8 33. e7 1-0"""]],
-        inputs=pgn_input
-    )
+1. e4 e5 2. Nf3 Nc6 3. Bc4 Nf6 4. Ng5 d5 5. exd5 Nxd5 6. Nxf7 Kxf7 7. Qf3+ Ke6 
+8. Nc3 Ne7 9. d4 c6 10. Bg5 h6 11. Bxe7 Bxe7 12. O-O-O Rf8 13. Qe4 Rxf2 14. d5+ Kd6 
+15. Nxd5 cxd5 16. Rxd5+ Kc6 17. Qc4+ Kb6 18. Rb5+ Ka6 19. Qc6+ 1-0"""]],
+                    inputs=pgn_input,
+                    label="Ataque Legal"
+                )
+                
+        with gr.Column(scale=1):
+            gr.Markdown("### 📊 Resultados del Análisis")
+            image_output = gr.Image(
+                label="Análisis Gráfico de la Partida", 
+                type="filepath",
+                height=400
+            )
+            text_output = gr.Markdown(
+                label="Reporte de Análisis Detallado"
+            )
+    
+    # Funcionalidades adicionales
+    gr.Markdown("---")
+    with gr.Accordion("📚 Guía de Formato PGN", open=False):
+        gr.Markdown("""
+        **Formato PGN Correcto:**
+        ```
+        [Event "Nombre del Torneo"]
+        [Site "Ciudad, País"]
+        [Date "2024.01.01"]
+        [Round "1"]
+        [White "Apellido, Nombre"]
+        [Black "Apellido, Nombre"]
+        [Result "1-0"]
+        [WhiteElo "2500"]
+        [BlackElo "2400"]
+
+        1. e4 e5 2. Nf3 Nc6 3. Bb5 a6 4. Ba4 Nf6 5. O-O Be7 6. Re1 b5 7. Bb3 d6
+        8. c3 O-O 9. h3 Nb8 10. d4 Nbd7 11. Nbd2 Bb7 12. Bc2 Re8 13. Nf1 Bf8
+        14. Ng3 g6 15. a4 c5 16. d5 c4 17. Bg5 h6 18. Be3 Nc5 19. Qd2 h5
+        ...
+        33. e7 1-0
+        ```
+
+        **Notación de Piezas:**
+        - Rey: **K** (King)
+        - Reina: **Q** (Queen) 
+        - Torre: **R** (Rook)
+        - Alfil: **B** (Bishop)
+        - Caballo: **N** (Knight)
+        - Peón: (sin letra, ej: e4, d5)
+        """)
     
+    # Conectores de eventos
     btn_analizar.click(
         fn=analizar_y_mostrar,
         inputs=pgn_input,
         outputs=[image_output, text_output]
     )
+    
+    btn_limpiar.click(
+        fn=lambda: [None, "### 📊 Reporte de Análisis\n\n*Los resultados aparecerán aquí después del análisis...*"],
+        inputs=[],
+        outputs=[image_output, text_output]
+    )
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch(
+        share=True,
+        show_error=True
+    )

requirements.txt

@@ -1,5 +1,5 @@
-chess
-matplotlib
-numpy
-gradio
-pillow
+chess==1.10.0
+matplotlib==3.7.1
+numpy==1.24.3
+gradio==4.19.1
+pillow==10.0.0