import streamlit as st
import google.generativeai as genai
import os

# Suponiendo que la configuración de la API ya está correctamente realizada y funcionando.
gemini_api_key = os.getenv("GEMINI_API_KEY")
genai.configure(api_key=gemini_api_key)
model = genai.GenerativeModel('gemini-1.5-flash-latest')

def explicar_despeje_ecuaciones_lineales():
    prompt = "Explica de manera detallada y paso a paso cómo despejar una ecuación lineal."
    response = model.generate_content(prompt)
    st.write(response.text)

def explicar_ecuaciones_lineales_2x2():
    prompt = "Explica de manera detallada y paso a paso cómo resolver un sistema de ecuaciones lineales con dos variables."
    response = model.generate_content(prompt)
    st.write(response.text)

def explicar_ecuaciones_lineales_3x3():
    prompt = "Explica de manera detallada y paso a paso cómo resolver un sistema de ecuaciones lineales con tres variables."
    response = model.generate_content(prompt)
    st.write(response.text)

def main():
    st.set_page_config(page_title="Módulo de Aprendizaje")
    st.title("Bienvenid@ al Módulo de Aprendizaje")
    st.write("Aquí podrás revisar y aprender acerca de despeje de ecuaciones lineales y sistemas de ecuaciones lineales 2x2 y 3x3. ¿Qué deseas profundizar?")

    # Crear un objeto para almacenar el estado de la selección
    if 'seleccion' not in st.session_state:
        st.session_state.seleccion = None

    col1, col2, col3 = st.columns(3)
    with col1:
        if st.button("Despeje de Ecuaciones Lineales"):
            st.session_state.seleccion = "Despeje"
    with col2:
        if st.button("Ecuaciones Lineales 2x2"):
            st.session_state.seleccion = "2x2"
    with col3:
        if st.button("Ecuaciones Lineales 3x3"):
            st.session_state.seleccion = "3x3"

    if st.session_state.seleccion == "Despeje":
        explicar_despeje_ecuaciones_lineales()
    elif st.session_state.seleccion == "2x2":
        explicar_ecuaciones_lineales_2x2()
    elif st.session_state.seleccion == "3x3":
        explicar_ecuaciones_lineales_3x3()

if __name__ == "__main__":
    main()