Update app.py

app.py

@@ -11,32 +11,6 @@ if gemini_api_key is None:
 genai.configure(api_key=gemini_api_key)
 model = genai.GenerativeModel('gemini-pro')
 
-def get_game_state():
-    """Obtiene el estado actual del juego"""
-    if 'aciertos' not in st.session_state:
-        st.session_state.aciertos = 0
-    if 'errores' not in st.session_state:
-        st.session_state.errores = 0
-    if 'nivel' not in st.session_state:
-        st.session_state.nivel = 0  # 0 = Principiante, 1 = Intermedio, 2 = Difícil
-    if 'problema' not in st.session_state:
-        st.session_state.problema = None
-    if 'respuestas' not in st.session_state:
-        st.session_state.respuestas = None
-    if 'respuesta_correcta' not in st.session_state:
-        st.session_state.respuesta_correcta = None
-    return (st.session_state.aciertos, st.session_state.errores, st.session_state.nivel,
-            st.session_state.problema, st.session_state.respuestas, st.session_state.respuesta_correcta)
-
-def update_game_state(aciertos, errores, nivel, problema, respuestas, respuesta_correcta):
-    """Actualiza el estado del juego"""
-    st.session_state.aciertos = aciertos
-    st.session_state.errores = errores
-    st.session_state.nivel = nivel
-    st.session_state.problema = problema
-    st.session_state.respuestas = respuestas
-    st.session_state.respuesta_correcta = respuesta_correcta
-
 def chat_with_model(user_input):
     """Envía una pregunta al modelo de IA y obtiene una respuesta."""
     try:
@@ -78,8 +52,8 @@ def generar_sistema_ecuaciones_2x2_y_respuestas():
     matriz_coeficientes = np.array([[a, b], [c, d]])
     matriz_resultados = np.array([e, f])
     soluciones = np.linalg.solve(matriz_coeficientes, matriz_resultados)
-    respuestas_incorrectas = {(round(soluciones[0] + random.choice([-2, -1, 1, 2]) * random.random(), 2), 
-                               round(soluciones[1] + random.choice([-2, -1, 1, 2]) * random.random(), 2)) 
+    respuestas_incorrectas = {(round(soluciones[0] + random.choice([-2, -1, 1, 2]) * random.random(), 2),
+                               round(soluciones[1] + random.choice([-2, -1, 1, 2]) * random.random(), 2))
                               for _ in range(3)}
     respuestas_correctas = (round(soluciones[0], 2), round(soluciones[1], 2))
     respuestas = list(respuestas_incorrectas) + [respuestas_correctas]
@@ -100,7 +74,7 @@ def generar_sistema_ecuaciones_3x3_y_respuestas():
     soluciones = np.linalg.solve(matriz_coeficientes, matriz_resultados)
     respuestas_incorrectas = {(round(soluciones[0] + random.choice([-2, -1, 1, 2]) * random.random(), 2),
                                round(soluciones[1] + random.choice([-2, -1, 1, 2]) * random.random(), 2),
-                               round(soluciones[2] + random.choice([-2, -1, 1, 2]) * random.random(), 2)) 
+                               round(soluciones[2] + random.choice([-2, -1, 1, 2]) * random.random(), 2))
                               for _ in range(3)}
     respuestas_correctas = (round(soluciones[0], 2), round(soluciones[1], 2), round(soluciones[2], 2))
     respuestas = list(respuestas_incorrectas) + [respuestas_correctas]
@@ -108,71 +82,76 @@ def generar_sistema_ecuaciones_3x3_y_respuestas():
     sistema = f"{a}x + {b}y + {c}z = {e}\n\n\n{d}x + {f}y + {g}z = {h}\n\n\n{i}x + {j}y + {k}z = {l}"
     return sistema, respuestas, respuestas_correctas
 
-def get_ui_elements():
-    """Genera los elementos de la interfaz de usuario según el nivel actual"""
-    nivel = st.session_state.nivel
-    if nivel == 2:  # Sistema 3x3
-        problema, respuestas, respuesta_correcta = generar_sistema_ecuaciones_3x3_y_respuestas()
-    elif nivel == 1:  # Sistema 2x2
-        problema, respuestas, respuesta_correcta = generar_sistema_ecuaciones_2x2_y_respuestas()
-    else:  # Ecuación lineal
-        problema, respuestas, respuesta_correcta = generar_ecuacion_y_respuestas()
-    opciones = [", ".join(map(str, r)) if isinstance(r, tuple) else str(r) for r in respuestas]
-    return problema, opciones, respuesta_correcta
-
-def manejar_respuesta(respuesta_elegida, respuesta_correcta, nivel):
+def manejar_respuesta(problema, respuesta_elegida, respuesta_correcta, nivel):
     """Evalúa la respuesta del usuario, proporcionando retroalimentación y una explicación."""
     tolerancia = 1e-9
     correcto = False
     if nivel == 2:  # Sistema 3x3
         correcto = all(abs(e - c) < tolerancia for e, c in zip(respuesta_elegida, respuesta_correcta))
+        explicacion = chat_with_model(f"Explica cómo resolver un sistema de ecuaciones lineales 3x3 y por qué la respuesta {respuesta_elegida} es {'correcta' if correcto else 'incorrecta'}.")
     elif nivel == 1:  # Sistema 2x2
         correcto = all(abs(e - c) < tolerancia for e, c in zip(respuesta_elegida, respuesta_correcta))
+        explicacion = chat_with_model(f"Explica cómo resolver un sistema de ecuaciones lineales 2x2 y por qué la respuesta {respuesta_elegida} es {'correcta' if correcto else 'incorrecta'}.")
     else:  # Ecuación lineal
         correcto = abs(respuesta_elegida - respuesta_correcta) < tolerancia
+        explicacion = chat_with_model(f"Explica cómo resolver una ecuación lineal y por qué la respuesta {respuesta_elegida} es {'correcta' if correcto else 'incorrecta'}.")
+
     if correcto:
         resultado = "¡Correcto! +1 punto."
-        aciertos, errores, nivel, _, _, _ = get_game_state()
-        if aciertos >= 5 and nivel == 0:
+        if st.session_state.aciertos >= 5 and nivel == 0:
             nivel = 1
             st.success("¡Felicitaciones, has pasado al nivel Intermedio: Sistemas de ecuaciones lineales 2x2!")
-        elif aciertos >= 10 and nivel == 1:
+        elif st.session_state.aciertos >= 10 and nivel == 1:
             nivel = 2
             st.success("¡Felicitaciones, has pasado al nivel Difícil: Sistemas de ecuaciones lineales 3x3!")
-        update_game_state(aciertos + 1, errores, nivel, None, None, None)
+        st.session_state.aciertos += 1
     else:
         resultado = "Incorrecto, sigue practicando."
-        aciertos, errores, nivel, _, _, _ = get_game_state()
-        update_game_state(aciertos, errores + 1, nivel, None, None, None)
-    explicacion = chat_with_model(f"Explica el problema y por qué la respuesta {respuesta_elegida} es {'correcta' if correcto else 'incorrecta'}.")
-    return resultado + f" Escogiste la respuesta: {respuesta_elegida}. La respuesta correcta es: {respuesta_correcta}.\n{explicacion}"
+        st.session_state.errores += 1
+
+    return resultado + f" Escogiste la respuesta: {respuesta_elegida}. La respuesta correcta es: {respuesta_correcta}.\n{explicacion}", nivel
 
 # Interfaz de usuario y lógica de la aplicación
 st.title("Desafío de Matemáticas")
 st.markdown("Intenta resolver el problema y selecciona tu respuesta.")
 
-aciertos, errores, nivel, problema, respuestas, respuesta_correcta = get_game_state()
+if 'aciertos' not in st.session_state:
+    st.session_state.aciertos = 0
+if 'errores' not in st.session_state:
+    st.session_state.errores = 0
+if 'nivel' not in st.session_state:
+    st.session_state.nivel = 0
+
+if 'nuevo_problema' not in st.session_state or st.session_state.nuevo_problema:
+    st.session_state.nuevo_problema = False
+    if st.session_state.nivel == 2:
+        problema_actual, respuestas, respuesta_correcta = generar_sistema_ecuaciones_3x3_y_respuestas()
+    elif st.session_state.nivel == 1:
+        problema_actual, respuestas, respuesta_correcta = generar_sistema_ecuaciones_2x2_y_respuestas()
+    else:
+        problema_actual, respuestas, respuesta_correcta = generar_ecuacion_y_respuestas()
+    st.session_state.problema = problema_actual
+    st.session_state.respuestas = respuestas
+    st.session_state.respuesta_correcta = respuesta_correcta
 
 niveles = ["Principiante", "Intermedio", "Difícil"]
-nivel_actual = niveles[nivel]
+nivel_actual = niveles[st.session_state.nivel]
 st.sidebar.markdown(f"Nivel: {nivel_actual}")
-st.sidebar.write(f"Aciertos: {aciertos}")
-st.sidebar.write(f"Errores: {errores}")
-
-if problema is None or respuestas is None or respuesta_correcta is None:
-    problema, respuestas, respuesta_correcta = get_ui_elements()
-    update_game_state(aciertos, errores, nivel, problema, respuestas, respuesta_correcta)
+st.sidebar.write(f"Aciertos: {st.session_state.aciertos}")
+st.sidebar.write(f"Errores: {st.session_state.errores}")
 
-st.write(problema)
-seleccion = st.radio("Elige tu respuesta", respuestas, key="opciones")
+st.write(st.session_state.problema)
+opciones = [f"{r[0]}, {r[1]}" if isinstance(r, tuple) else str(r) for r in st.session_state.respuestas]
+seleccion = st.radio("Elige tu respuesta", opciones, key="opciones")
 respuesta_elegida = tuple(map(float, seleccion.split(', '))) if "," in seleccion else float(seleccion)
 
 if st.button("Enviar"):
-    resultado = manejar_respuesta(respuesta_elegida, respuesta_correcta, nivel)
+    resultado, nuevo_nivel = manejar_respuesta(st.session_state.problema, respuesta_elegida, st.session_state.respuesta_correcta, st.session_state.nivel)
     st.write(resultado)
+    st.session_state.nivel = nuevo_nivel
+    st.session_state.nuevo_problema = True
 
 boton_nuevo_problema = st.button("Generar Nuevo Problema")
 if boton_nuevo_problema:
-    problema, respuestas, respuesta_correcta = get_ui_elements()
-    update_game_state(aciertos, errores, nivel, problema, respuestas, respuesta_correcta)
+    st.session_state.nuevo_problema = True
     st.experimental_rerun()