Update app.py

app.py

@@ -4,25 +4,25 @@ import streamlit as st
 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)
+def get_answers(questions,model):
+    st.write("running get answers function answering following questions",questions)
 
 
-#     answer_prompt = ( "Answer the 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
+    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
+    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)    
+    return(answers)    
 
 
 
@@ -35,7 +35,7 @@ def GetLLMResponse(selected_topic_level, selected_topic,num_quizzes, model):
 
 
     
-    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.')
+    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":
@@ -51,10 +51,10 @@ def GetLLMResponse(selected_topic_level, selected_topic,num_quizzes, model):
 
 
     st.write("print questions",questions)
-    # answers = get_answers(questions,model)
+    answers = get_answers(questions,model)
     
     # st.write(questions,answers)
-    return(questions)
+    return(questions,answers)
 
 
 
@@ -103,7 +103,7 @@ def main():
     
     # Final Response
     if submit:
-        questions = GetLLMResponse(selected_topic_level, selected_topic, num_quizzes, model)
+        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)