import streamlit as st
from function import GetLLMResponse

from langchain_community.llms import OpenAI
from langchain_google_genai import ChatGoogleGenerativeAI

    
# Page configuration
st.set_page_config(page_title="Generate Math Quizzes",
                   page_icon="🧮",
                   layout="wide",
                   initial_sidebar_state="collapsed")




def main():        
    math_topics = {
        "Elementary School Level": ["Basic Arithmetic", "Place Value", "Fraction", "Decimals", "Geomerty"],
        "Middle School Level": ["Algebra", "Ratio and Proportion", "Percentages", "Geometry", "Integers and Rational Numbers"],
        "High School Level": ["Algebra II", "Trigonometry", "Pre-Calculus", "Calculus", "Statistics and Probability"]
    }

    st.header("Select AI:")
    model = st.radio("Model", [ "Gemini","Open AI",])
    st.write("Selected option:", model)
    
    
    
    # Header and description
    st.title("Generate Math Quizzes 🧮")
    st.text("Choose the difficulty level and topic for your math quizzes.")
    
    # User input for quiz generation
    ## Layout in columns
    col1, col2, col3 = st.columns([1, 1, 1])
    
    with col1:
        selected_topic_level = st.selectbox('Select Topic Level', list(math_topics.keys()))
    
    with col2:
        selected_topic = st.selectbox('Select Topic', math_topics[selected_topic_level])
    
    with col3:
        num_quizzes = st.slider('Number Quizzes', min_value=1, max_value= 5, value=1)
    
    submit = st.button('Generate Quizzes')
    st.write("Selected option:", selected_topic_level,selected_topic,num_quizzes)
    
    # Final Response
    if submit:
        questions,answers = GetLLMResponse(selected_topic_level, selected_topic, num_quizzes, model)
        st.write("printing the response",questions,answers)
        with st.spinner("Generating Quizzes..."):
            questions,answers = GetLLMResponse(selected_topic_level, selected_topic, num_quizzes, model)
            st.success("Quizzes Generated!")


            
            # Display questions and answers in a table
            if questions:
                st.subheader("Quiz Questions and Answers:")
                # Prepare data for the table
                col1, col2 = st.columns(2)
                with col1:
                    st.subheader("Questions")
                    
                    st.write(questions)
                
                with col2:
                    st.subheader("Answers")

                    st.write(answers)

               
    
            else:
                st.warning("No Quiz Questions and Answers")
                
    else:
        st.warning("Click the 'Generate Quizzes' button to create quizzes.")





    
    


if __name__ == "__main__":
    main()