Update app.py

app.py

@@ -69,8 +69,12 @@ def generate_quiz(quiz_name, quiz_topic, num_questions, pdf_content):
             prompt = PromptTemplate(template=template, input_variables=['chat_history','human_input','name','context'])
 
 
-            # Store question data in vector database
-            st.session_state.vectorDB.add(prompt.generate(), embedding=None)
+            # 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])
+
 
     # Save button to store vector database
     if st.session_state.vectorDB: