Update app.py

app.py

@@ -28,7 +28,7 @@ def get_pdf_text(pdf):
     return text
 
 def get_text_chunks(text: str):
-    """ This function will split the text into smaller chunks"""
+    # This function will split the text into smaller chunks
     text_splitter = RecursiveCharacterTextSplitter(
         chunk_size=1000,
         chunk_overlap=100,
@@ -39,29 +39,32 @@ def get_text_chunks(text: str):
     return chunks
 
 def get_vectorstore(text_chunks):
-    """ This function will create a vector database as well as create and store the embedding of the text chunks into the VectorDB """
+    # This function will create a vector database as well as create and store the embedding of the text chunks into the VectorDB 
     embeddings = OpenAIEmbeddings()
     vectorstore = FAISS.from_texts(texts=text_chunks, embedding=embeddings)
     return vectorstore
 
 def search_vectorstore(vectorDB, query):
-    """ This function searches for specific data or content in the vector database"""
+    # This function searches for specific data or content in the vector database
     # Specify the search type
     search_type = 'similarity'
     search_results = vectorDB.search(query, search_type=search_type)
     return search_results
 
 def generate_quiz_questions(search_results, num_questions):
-    """ Generate quiz questions with options using GPT-3.5-turbo-16k based on the search results """
+    # Generate quiz questions with options using GPT-3.5-turbo-16k based on the search results
     st.header(f"Quiz Generator: {quiz_name}")
     st.subheader(f"Topic: {quiz_topic}")
+    # Process PDF and create vector database
+    if st.button('Generate Quiz'):
+        st.session_state['vectorDB'] = (pdf_content)
 
     # Placeholder for quiz questions
     quiz_questions = []
 
     # Generate questions using GPT-3.5-turbo-16k with the correct endpoint
     for i in range(num_questions):
-        prompt = f"Generate a multiple-choice question  and correct answer for the same related to:\n- {random.choice(search_results)}"
+        prompt = f"As a highly intelligent and powerful quiz generator, your task is to Generate a different multiple-choice question and correct answer in a comprehensive and accurate format for the same related to:\n- {random.choice(search_results)} and the response generated by you should be comprehensive and relevant to the topic specified, you should not generate the same content multiple times in a single response, generate the explaination for each correct answer"
 
         # Prepare messages
         messages = [
@@ -108,7 +111,7 @@ if __name__ == '__main__':
     # User inputs
     quiz_name = st.text_input('Enter Quiz Name:')
     quiz_topic = st.text_input('Enter Quiz Topic:')
-    num_questions = st.number_input('Enter Number of Questions:', min_value=1, value=1, step=1)
+    num_questions = st.number_input('Enter Number of Questions:', min_value=1, value=5, step=1)
     pdf_content = st.file_uploader("Upload PDF Content for Questions:", type='pdf')
 
     # Generate quiz if all inputs are provided