Update function.py

function.py

@@ -8,12 +8,12 @@ def GetLLMResponse(selected_topic_level, selected_topic,num_quizzes, model):
 
     if model == "Open AI":
         llm = OpenAI(temperature=0.7, openai_api_key=st.secrets["OPENAI_API_KEY"])
-        questions = llm(question_prompt)
+        questions = [llm(question_prompt)]
         # return questions
         
     elif model == "Gemini":
         llm = ChatGoogleGenerativeAI(model="gemini-pro", google_api_key=st.secrets["GOOGLE_API_KEY"])
-        questions = llm.invoke(question_prompt)
+        questions = [llm.invoke(question_prompt)]
         # return questions.content
 
 
@@ -22,12 +22,12 @@ def GetLLMResponse(selected_topic_level, selected_topic,num_quizzes, model):
     
     if model == "Open AI":
         llm = OpenAI(temperature=0.7, openai_api_key=st.secrets["OPENAI_API_KEY"])
-        answers = llm(answer_prompt)
+        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 = [llm.invoke(answer_prompt)]
         # return questions.content