import streamlit as st
import os
from streamlit_chat import message
import openai

# Configura la clave de API de OpenAI
openai.api_key = os.getenv("OPENAI_API_KEY")

def resolver_ecuaciones_lineales(modelo, mensaje):
    respuesta = openai.Completion.create(
        engine=modelo,
        prompt=mensaje,
        max_tokens=2048,
        n=1,
        stop=None,
        temperature=0.7,
    )
    return respuesta.choices[0].text

def explicar_resolucion_ecuaciones_lineales():
    st.title("Resolver Ecuaciones Lineales")
    st.write("Esta aplicación proporciona explicaciones paso a paso para resolver ecuaciones lineales.")

    opcion = st.selectbox(
        "Selecciona un tema",
        ("Resolver Ecuaciones Lineales (1 variable)", "Resolver Ecuaciones Lineales (2 variables)", "Resolver Ecuaciones Lineales (3 variables)"),
    )

    if opcion == "Resolver Ecuaciones Lineales (1 variable)":
        mensaje = "Explica los pasos para resolver una ecuación lineal con una variable de manera detallada y paso a paso."
        respuesta = resolver_ecuaciones_lineales("text-davinci-002", mensaje)
        st.write(respuesta)
    elif opcion == "Resolver Ecuaciones Lineales (2 variables)":
        mensaje = "Explica los pasos para resolver un sistema de ecuaciones lineales con dos variables de manera detallada y paso a paso."
        respuesta = resolver_ecuaciones_lineales("text-davinci-002", mensaje)
        st.write(respuesta)
    elif opcion == "Resolver Ecuaciones Lineales (3 variables)":
        mensaje = "Explica los pasos para resolver un sistema de ecuaciones lineales con tres variables de manera detallada y paso a paso."
        respuesta = resolver_ecuaciones_lineales("text-davinci-002", mensaje)
        st.write(respuesta)

def main():
    st.set_page_config(page_title="Resolvedor de Ecuaciones Lineales")
    st.sidebar.title("Menú")
    opciones = ["Explicar Resolución de Ecuaciones Lineales", "Resolver Ecuaciones Lineales"]
    seleccion = st.sidebar.selectbox("Elige una opción", opciones)

    if seleccion == "Explicar Resolución de Ecuaciones Lineales":
        explicar_resolucion_ecuaciones_lineales()
    elif seleccion == "Resolver Ecuaciones Lineales":
        st.write("¡Próximamente!")

if __name__ == "__main__":
    main()