Update app.py

app.py

@@ -36,13 +36,46 @@ def get_text_chunks(text):
     text_chunks = text.split('\n\n')  # Modify this based on your text splitting requirements
     return text_chunks
 
+# Function to process PDF and create vector database
+def processing(pdf):
+    raw_text = get_pdf_text(pdf)
+    text_chunks = get_text_chunks(raw_text)
+    vectorDB = get_vectorstore(text_chunks)
+    return vectorDB
+
+# Function to generate quiz questions, options, and correct answers
+def generate_quiz_content(text, num_questions):
+    messages = [{'role': 'system', 'content': 'You are a helpful assistant.'}]
+    for i in range(num_questions):
+        user_message = {'role': 'user', 'content': f'Generate question {i + 1} from the given text:\n{text}'}
+        messages.append(user_message)
+
+    response = openai.ChatCompletion.create(
+        model='gpt-3.5-turbo-16k',
+        messages=messages,
+        max_tokens=200,
+        temperature=0.7
+    )
+    quiz_data = []
+
+    for choice in response.get('choices', []):
+        content = choice.get('content', '').strip()
+        options_start = content.find("Options:") + len("Options:")
+        correct_answer_start = content.find("Correct Answer:") + len("Correct Answer:")
+
+        question = content[:options_start].strip()
+        options_str = content[options_start:correct_answer_start].strip()
+        correct_answer = content[correct_answer_start:].strip()
+
+        options = [opt.strip() for opt in options_str.split(',')]
+        quiz_data.append((question, options, correct_answer))
+
+    return quiz_data
+
 # Function to retrieve quiz data from the vector database
 def retrieve_quiz_data(vectorDB, num_questions):
-    # Retrieve stored quiz data from the vector database
-    # You need to implement the logic to query the vector database and get quiz data
-    # For illustration purposes, assuming you have a function named 'query_vector_database'
-    quiz_data = query_vector_database(vectorDB, num_questions)
-    return quiz_data
+    # Replace this with your actual logic to query the vector database and retrieve quiz data
+    return [(f"Question {i + 1}", [f"Option {j}" for j in range(1, 5)], f"Option {i % 4 + 1}") for i in range(num_questions)]
 
 # Function to generate quiz questions
 def generate_quiz(quiz_name, quiz_topic, num_questions, pdf_content):
@@ -71,12 +104,6 @@ def generate_quiz(quiz_name, quiz_topic, num_questions, pdf_content):
         if st.button('Save Vector Database'):
             st.success('Vector Database Saved')
 
-# Replace this function with your logic to query the vector database and get quiz data
-def query_vector_database(vectorDB, num_questions):
-    # Implement your logic to query the vector database and retrieve quiz data
-    # This is a placeholder, replace it with your actual implementation
-    return [(f"Question {i + 1}", [f"Option {j}" for j in range(1, 5)], f"Option {i % 4 + 1}") for i in range(num_questions)]
-
 if __name__ =='__main__':
     st.set_page_config(page_title="CB Quiz Generator", page_icon="📝")
     st.title('CB Quiz Generator')