Update app.py

app.py

@@ -1,63 +1,9 @@
 import streamlit as st
-# from function import GetLLMResponse
+from function import GetLLMResponse
 
 from langchain_community.llms import OpenAI
 from langchain_google_genai import ChatGoogleGenerativeAI
 
-def get_answers(questions,model):
-    st.write("running get answers function answering following questions",questions)
-
-
-    answer_prompt = (f"Answer the following questions {questions}")
-
-    
-    if model == "Open AI":
-        llm = OpenAI(temperature=0.7, openai_api_key=st.secrets["OPENAI_API_KEY"])
-        answers = llm(answer_prompt)
-        # return questions
-        
-    elif model == "Gemini":
-        llm = ChatGoogleGenerativeAI(model="gemini-pro", google_api_key=st.secrets["GOOGLE_API_KEY"])
-        answers = llm.invoke(answer_prompt)
-        answers = answers.content
-        # return questions.content
-
-    return(answers)    
-
-
-
-
-def GetLLMResponse(selected_topic_level, selected_topic,num_quizzes, model):
-
-
-
-    st.write("Selected option:", selected_topic_level,selected_topic,num_quizzes)
-
-
-    
-    question_prompt = (f'I want you to just generate question with this specification: Generate a {selected_topic_level} math quiz on the topic of {selected_topic}. Generate only {num_quizzes} questions not more and without providing answers.')
-    
-    st.write("running get llm response and print question prompt",question_prompt)
-    if model == "Open AI":
-        llm = OpenAI(temperature=0.7, openai_api_key=st.secrets["OPENAI_API_KEY"])
-        questions = llm(question_prompt)
-        
-        
-    elif model == "Gemini":
-        llm = ChatGoogleGenerativeAI(model="gemini-pro", google_api_key=st.secrets["GOOGLE_API_KEY"])
-        questions = llm.invoke(question_prompt)
-        questions = questions.content
-        # return questions.content
-
-
-    st.write("print questions",questions)
-    answers = get_answers(questions,model)
-    
-    # st.write(questions,answers)
-    return(questions,answers)
-
-
-
     
 # Page configuration
 st.set_page_config(page_title="Generate Math Quizzes",
@@ -105,31 +51,31 @@ def main():
     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!")
-
-
-        #     st.write("printing the response",questions,answers)
-        #     # # Display questions and answers in a table
-        #     # if response:
-        #     #     st.subheader("Quiz Questions and Answers:")
-        #     #     # Prepare data for the table
-        #     #     # col1, col2 = st.columns(2)
-        #     #     # with col1:
-        #     #     #     st.subheader("Questions")
-        #     #     #     questions = response
-        #     #     #     st.write(questions)
+        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 response:
+                st.subheader("Quiz Questions and Answers:")
+                # Prepare data for the table
+                # col1, col2 = st.columns(2)
+                # with col1:
+                #     st.subheader("Questions")
+                #     questions = response
+                #     st.write(questions)
                 
-        #     #     # with col2:
-        #     #     #     st.subheader("Answers")
-        #     #     #     answers = response
-        #     #     #     st.write(answers)
+                # with col2:
+                #     st.subheader("Answers")
+                #     answers = response
+                #     st.write(answers)
 
-        #     #     st.write("printing the response",questions,answers)
+                st.write("printing the response",questions,answers)
     
-        #     # else:
-        #     #     st.warning("No Quiz Questions and Answers")
+            else:
+                st.warning("No Quiz Questions and Answers")
                 
     else:
         st.warning("Click the 'Generate Quizzes' button to create quizzes.")