Update function.py

function.py

@@ -10,15 +10,12 @@ def get_answers(questions,model):
     answer_prompt = (f"I want you to become a teacher answer this specific Question: {questions}. You should gave me a straightforward and consise explanation and answer to each one of them")
 
     
-    if model == "Open AI":
-        llm = OpenAI(temperature=0.8, 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
+   
+    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)    
@@ -41,9 +38,8 @@ def GetLLMResponse(selected_topic_level, selected_topic,num_quizzes, model):
         questions = questions.content
         # return questions.content
 
-    answers = "testing"
 
-    # answers = get_answers(questions,model)
+    answers = get_answers(questions,model)
     
 
     return(questions,answers)