Update app.py

app.py

@@ -69,12 +69,8 @@ def generate_quiz(quiz_name, quiz_topic, num_questions, pdf_content):
             prompt = PromptTemplate(template=template, input_variables=['chat_history','human_input','name','context'])
 
 
-            # Get the vector from the prompt
-            vector = st.session_state.vectorDB.embed(prompt.generate())
-
-            # Add the vector to the FAISS index
-            st.session_state.vectorDB.index([vector])
-
+            # Store question data in vector database
+            st.session_state.vectorDB.add(prompt.generate(), embedding=None)
 
     # Save button to store vector database
     if st.session_state.vectorDB:
@@ -95,3 +91,4 @@ if __name__ =='__main__':
     # Generate quiz if all inputs are provided
     if quiz_name and quiz_topic and num_questions and pdf_content:
         generate_quiz(quiz_name, quiz_topic, num_questions, pdf_content)
+