Update app.py

app.py

@@ -6,53 +6,70 @@ import os
 from groq import Groq
 
 # Initialize Groq client
-GROQ_API_KEY= os.environ.get('GroqApi')
+GROQ_API_KEY = os.environ.get('GroqApi')
 client = Groq(api_key=GROQ_API_KEY)
 
 # Initialize embedding model
 embedding_model = SentenceTransformer('all-MiniLM-L6-v2')
 
-# Upload PDF
-st.title("PDF to Quiz Generator")
+# Streamlit app title and description
+st.title("📘 PDF to Quiz Generator")
+st.write("Upload a PDF document, and we'll generate a multiple-choice quiz for you!")
+
+# File uploader for PDF
 uploaded_file = st.file_uploader("Upload a PDF file", type=["pdf"])
 
 if uploaded_file:
-    # Extract text from PDF
-    st.write("Processing PDF...")
+    # Step 1: Extract text from PDF
+    st.write("🔄 Processing PDF...")
     reader = PdfReader(uploaded_file)
     pdf_text = ""
     for page in reader.pages:
         pdf_text += page.extract_text()
 
-    # Split text into chunks
+    # Step 2: Split text into chunks
     chunk_size = 512  # Adjust as needed
     text_chunks = [pdf_text[i:i + chunk_size] for i in range(0, len(pdf_text), chunk_size)]
 
-    # Generate embeddings
-    st.write("Generating embeddings...")
+    # Step 3: Generate embeddings
+    st.write("🔍 Generating embeddings...")
     embeddings = embedding_model.encode(text_chunks)
 
-    # Store embeddings in FAISS
+    # Step 4: Store embeddings in FAISS
     dimension = embeddings.shape[1]
     index = faiss.IndexFlatL2(dimension)
     index.add(embeddings)
 
-    # Generate questions using Groq API
-    def generate_question(content):
+    # Step 5: Generate questions and options using Groq API
+    def generate_question_and_options(content):
         response = client.chat.completions.create(
-            messages=[{"role": "user", "content": f"Generate a multiple-choice question from: {content}"}],
+            messages=[{"role": "user", "content": f"Generate a multiple-choice question with options from: {content}"}],
             model="llama3-8b-8192",
         )
         return response.choices[0].message.content
 
-    # Generate quiz
-    st.write("Generating quiz...")
+    # Generate the quiz
+    st.write("📝 Generating quiz...")
     quiz = []
     for chunk in text_chunks:
-        question = generate_question(chunk)
-        quiz.append(question)
+        question_data = generate_question_and_options(chunk)
+        # Expecting response in format: "Question: ...\nOptions: A)... B)... C)...\nAnswer: ..."
+        question_lines = question_data.split("\n")
+        question = question_lines[0].replace("Question: ", "")
+        options = [line.split(")")[1].strip() for line in question_lines if line.startswith(("A", "B", "C", "D"))]
+        answer = question_lines[-1].replace("Answer: ", "").strip()
+        quiz.append({"question": question, "options": options, "answer": answer})
 
     # Display the quiz
-    st.write("Here is your quiz:")
+    st.write("🎯 **Your Quiz:**")
     for i, q in enumerate(quiz, 1):
-        st.markdown(f"**Question {i}:** {q}")
+        st.markdown(f"### Question {i}: {q['question']}")
+        selected_option = st.radio(f"Select your answer for Question {i}:", q['options'], key=f"q{i}")
+
+        if st.button(f"Submit Answer for Question {i}", key=f"submit_q{i}"):
+            if selected_option == q['answer']:
+                st.success(f"🎉 Correct! The answer is: {q['answer']}", icon="✅")
+            else:
+                st.error(f"❌ Incorrect! The correct answer is: {q['answer']}", icon="🚫")
+
+    st.write("Thank you for taking the quiz! 🎓")