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 = ( "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 = ('I want you to just generate question with this specification: Generate a basic level math quiz on the topic of basic algebra. Generate only 5 questions not more and without providing answers.')
+    question_prompt = ('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,answers)
+    # st.write(questions,answers)
+    return(questions)
 
 
 
@@ -103,8 +103,8 @@ def main():
     
     # Final Response
     if submit:
-        questions,answers = GetLLMResponse(selected_topic_level, selected_topic, num_quizzes, model)
-        st.write("printing the response",questions,answers)
+        questions = 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!")