Update app.py

app.py

@@ -1,71 +1,211 @@
+import gradio as gr
+import random
+import time
+import pandas as pd
+from PyPDF2 import PdfReader
+
 # ============================
-# MODERN GRADIO UI
+# GLOBALS
 # ============================
+quiz_history = []
 
-custom_css = """
-body {
-    background: linear-gradient(135deg, #0f172a, #1e293b);
-}
-.gradio-container {
-    max-width: 900px !important;
-    margin: auto;
-}
-h1, h2, h3 {
-    text-align: center;
-}
-"""
+# ============================
+# Extract sentences
+# ============================
+def extract_sentences(text):
+    sentences = [s.strip() for s in text.split(".") if len(s.strip()) > 20]
+    return sentences
 
-with gr.Blocks(title="Smart Quiz Generator", css=custom_css, theme=gr.themes.Soft()) as app:
+# ============================
+# Generate quiz
+# ============================
+def generate_quiz(text, num_questions):
+    try:
+        num_questions = int(num_questions)
+        if num_questions <= 0:
+            num_questions = 10
+    except:
+        num_questions = 10
+
+    sentences = extract_sentences(text)
+    if len(sentences) == 0:
+        return [], "❌ Not enough content to generate quiz."
+
+    selected = [random.choice(sentences) for _ in range(num_questions)]
+    quiz = []
+
+    for s in selected:
+        words = s.split()
+        if len(words) < 6:
+            continue
+        answer = random.choice(words)
+        question = s.replace(answer, "_____")
+        options = list(set(words))
+        random.shuffle(options)
+        options = options[:3]
+        options.append(answer)
+        random.shuffle(options)
+        quiz.append({
+            "question": question,
+            "options": options,
+            "answer": answer
+        })
+
+    return quiz, f"✅ Generated {len(quiz)} questions successfully!"
 
-    gr.Markdown("""
-    # 🧠 Smart Quiz Generator
-    ### AI-Powered Unlimited MCQ System
-    Turn your study material into interactive quizzes instantly.
-    """)
+# ============================
+# Display quiz
+# ============================
+def show_quiz(quiz):
+    output = ""
+    for i, q in enumerate(quiz):
+        output += f"\nQ{i+1}. {q['question']}\n"
+        for opt in q["options"]:
+            output += f"   • {opt}\n"
+    return output
 
-    with gr.Tabs():
+# ============================
+# Submit answer
+# ============================
+def submit_answer(selected, quiz, index, score, end_time):
+    if quiz is None or len(quiz) == 0:
+        return "❌ Generate quiz first!", None, None, None, None, None, None, ""
+    if index >= len(quiz):
+        return "✅ Quiz Finished!", None, None, None, None, None, None, ""
+    if selected == quiz[index]["answer"]:
+        score += 1
+    index += 1
+    return show_question(quiz, index, score, end_time)
 
-        # ================= QUIZ TAB =================
-        with gr.Tab("📝 Take Quiz"):
+# ============================
+# Show question
+# ============================
+def show_question(quiz, index, score, end_time):
+    if quiz is None or len(quiz) == 0:
+        return "❌ No quiz found.", None, None, None, None, None, None, ""
+    if index >= len(quiz) or time.time() > end_time:
+        percent = (score / len(quiz)) * 100
+        quiz_history.append({
+            "Score": score,
+            "Total": len(quiz),
+            "Percentage": round(percent, 2),
+            "Time": time.strftime("%H:%M:%S")
+        })
+        return (f"✅ Quiz Finished! Score: {score}/{len(quiz)} ({percent:.1f}%)",
+                None, None, None, None, None, None, "")
+    q = quiz[index]
+    remaining = int(end_time - time.time())
+    return (
+        f"### Question {index+1}:\n\n{q['question']}",
+        gr.update(choices=q["options"], value=None),
+        f"Score: {score}",
+        quiz,
+        index,
+        score,
+        end_time,
+        f"⏳ Time Left: {remaining} seconds"
+    )
 
-            with gr.Group():
-                text_input = gr.Textbox(
-                    lines=6,
-                    label="📚 Paste Study Material",
-                    placeholder="Paste your notes or study content here..."
-                )
+# ============================
+# Prepare quiz (text or PDF)
+# ============================
+def prepare_quiz(text, pdf_library, selected_pdf, num_questions, timer_minutes):
+    if selected_pdf and selected_pdf in pdf_library:
+        text = pdf_library[selected_pdf]
 
-                pdf_input = gr.File(
-                    file_types=[".pdf"],
-                    file_count="multiple",
-                    label="📂 Upload PDF Books"
-                )
+    if not text or text.strip() == "":
+        return "⚠️ Provide text or select a PDF.", None, None, None, None, None, None, ""
 
-                pdf_selector = gr.Dropdown(
-                    label="📖 Select Uploaded Book",
-                    choices=[]
-                )
+    try:
+        timer_minutes = float(timer_minutes)
+        if timer_minutes <= 0:
+            timer_minutes = 10
+    except:
+        timer_minutes = 10
 
-            with gr.Row():
-                num_questions = gr.Number(
-                    label="❓ Number of Questions",
-                    value=10,
-                    precision=0
-                )
+    quiz, msg = generate_quiz(text, num_questions)
+    end_time = time.time() + timer_minutes*60
+    return show_question(quiz, 0, 0, end_time)
 
-                timer_minutes = gr.Number(
-                    label="⏱ Duration (Minutes)",
-                    value=10
-                )
+# ============================
+# Load multiple PDFs
+# ============================
+def load_multiple_pdfs(files):
+    if files is None:
+        return {}, gr.update(choices=[])
+
+    pdf_texts = {}
+    names = []
+
+    for file in files:
+        try:
+            text = ""
+            with open(file.name, "rb") as f:
+                pdf = PdfReader(f)
+                for page in pdf.pages:
+                    page_text = page.extract_text()
+                    if page_text:
+                        text += page_text + " "
+            name = file.name.split("/")[-1]
+            pdf_texts[name] = text.strip()
+            names.append(name)
+        except Exception as e:
+            print("PDF Error:", e)
+
+    return pdf_texts, gr.update(choices=names, value=names[0] if names else None)
 
-            start_btn = gr.Button("🚀 Start Quiz", variant="primary")
+# ============================
+# Dashboard summary
+# ============================
+def get_dashboard():
+    if not quiz_history:
+        return "No attempts yet.", None
+    df = pd.DataFrame(quiz_history)
+    summary = f"""
+### 📊 Performance Summary
+Attempts: {len(df)}  
+Best Score: {df['Percentage'].max():.1f}%  
+Average Score: {df['Percentage'].mean():.1f}%  
+"""
+    return summary, df
 
-            gr.Markdown("---")
+# ============================
+# Save CSV
+# ============================
+def save_results():
+    if not quiz_history:
+        return None
+    df = pd.DataFrame(quiz_history)
+    file = "quiz_results.csv"
+    df.to_csv(file, index=False)
+    return file
 
-            question_box = gr.Markdown()
-            options_radio = gr.Radio(label="Choose Your Answer")
-            submit_btn = gr.Button("✅ Submit Answer", variant="secondary")
+# ============================
+# GRADIO UI
+# ============================
+with gr.Blocks(title="Smart Quiz Generator") as app:
+
+    gr.Markdown("# 🧠 Smart Quiz Generator")
+
+    with gr.Tabs():
+        with gr.Tab("📝 Take Quiz"):
 
+            text_input = gr.Textbox(lines=6, label="Paste Study Material")
+
+            pdf_input = gr.File(file_types=[".pdf"], file_count="multiple", label="Upload PDFs")
+            pdf_selector = gr.Dropdown(label="Select Book", choices=[])
+
+            with gr.Tabs():
+                with gr.Tab("❓ Select Objectives"):
+                    num_questions = gr.Number(label="Number of Questions (No Limit)", value=10, precision=0)
+                with gr.Tab("⏱ Select Time"):
+                    timer_minutes = gr.Number(label="Quiz Duration in Minutes (No Limit)", value=10)
+
+            start_btn = gr.Button("Start Quiz")
+
+            question_box = gr.Markdown()
+            options_radio = gr.Radio(label="Choose Answer")
+            submit_btn = gr.Button("Submit Answer")
             score_display = gr.Markdown()
             timer_display = gr.Markdown()
 
@@ -75,43 +215,26 @@ with gr.Blocks(title="Smart Quiz Generator", css=custom_css, theme=gr.themes.Sof
             endtime_state = gr.State()
             pdf_library_state = gr.State({})
 
-        # ================= DASHBOARD TAB =================
-        with gr.Tab("📊 Performance Dashboard"):
-
-            dash_btn = gr.Button("🔄 Refresh Dashboard", variant="primary")
-
+        with gr.Tab("📊 Dashboard"):
+            dash_btn = gr.Button("Refresh Dashboard")
             dash_summary = gr.Markdown()
             dash_table = gr.Dataframe()
-
-            download_btn = gr.Button("⬇ Download Results")
+            download_btn = gr.Button("Download Results")
             download_file = gr.File()
 
-    # ================= FOOTER =================
-    gr.Markdown("""
-    ---
-    🔥 Built by Muhammad Raza | Generative AI Engineer  
-    Transforming Education with AI
-    """)
-
     # ---------------- EVENTS ----------------
-    pdf_input.change(
-        load_multiple_pdfs,
-        inputs=pdf_input,
-        outputs=[pdf_library_state, pdf_selector]
-    )
+    pdf_input.change(load_multiple_pdfs, inputs=pdf_input, outputs=[pdf_library_state, pdf_selector])
 
     start_btn.click(
         prepare_quiz,
         inputs=[text_input, pdf_library_state, pdf_selector, num_questions, timer_minutes],
-        outputs=[question_box, options_radio, score_display,
-                 quiz_state, index_state, score_state, endtime_state, timer_display]
+        outputs=[question_box, options_radio, score_display, quiz_state, index_state, score_state, endtime_state, timer_display]
     )
 
     submit_btn.click(
         submit_answer,
         inputs=[options_radio, quiz_state, index_state, score_state, endtime_state],
-        outputs=[question_box, options_radio, score_display,
-                 quiz_state, index_state, score_state, endtime_state, timer_display]
+        outputs=[question_box, options_radio, score_display, quiz_state, index_state, score_state, endtime_state, timer_display]
     )
 
     dash_btn.click(get_dashboard, outputs=[dash_summary, dash_table])