Feat: Implement AI DPP CBT conversion and frontend sync

2026-07-07 16:12:56 +05:30
parent b6ebb9d9c3
commit 988923ef4b
11 changed files with 168 additions and 18 deletions
--- a/backend/seed_dpps.py
+++ b/backend/seed_dpps.py
@@ -0,0 +1,112 @@
+import asyncio
+import os
+import uuid
+from sqlmodel import Session, select
+from datetime import datetime
+
+from database import engine, init_db
+import models
+import storage
+from pdf_parser import parse_pdf_to_questions
+
+async def seed_pdfs():
+    # Admin ID fallback
+    admin_id_str = "b0e17378-864d-439e-8806-708dc48a891b" # Or fetch first user
+    
+    with Session(engine) as session:
+        admin = session.exec(select(models.User).limit(1)).first()
+        if admin:
+            admin_id = admin.id
+        else:
+            admin_id = uuid.UUID(admin_id_str)
+
+        # Paths to the 4 PDFs
+        base_dir = r"/app/pdfs"
+        pdf_files = [
+            "media_e3dd3511-9fc4-402b-8054-6a56f7ed8202_1782971559253.pdf",
+            "media_e3dd3511-9fc4-402b-8054-6a56f7ed8202_1782979477911.pdf",
+            "media_e3dd3511-9fc4-402b-8054-6a56f7ed8202_1783273583537.pdf",
+            "media_e3dd3511-9fc4-402b-8054-6a56f7ed8202_1783410386530.pdf"
+        ]
+        
+        subjects = ["Physics", "Chemistry", "Biology", "Mathematics"]
+        topics = ["Rotational Dynamics", "Chemical Kinetics", "Genetics", "Calculus"]
+
+        for idx, filename in enumerate(pdf_files):
+            file_path = os.path.join(base_dir, filename)
+            if not os.path.exists(file_path):
+                print(f"File not found: {file_path}")
+                continue
+            
+            print(f"Processing {filename}...")
+            
+            with open(file_path, "rb") as f:
+                pdf_bytes = f.read()
+                
+            # 1. Upload to Minio
+            object_name = f"dpps/{uuid.uuid4()}_{filename}"
+            storage.upload_file(pdf_bytes, object_name, content_type="application/pdf")
+            
+            # 2. Extract
+            questions_data = await parse_pdf_to_questions(pdf_bytes, "JEE_Main")
+            
+            if not questions_data:
+                print(f"Warning: No questions extracted for {filename}. Using mock data fallback.")
+                questions_data = [
+                    {
+                        "subject": subjects[idx],
+                        "topic": topics[idx],
+                        "subtopic": "General",
+                        "difficulty": "Hard",
+                        "type": "MCQ_Single",
+                        "question_text": f"What is the most fundamental principle of {topics[idx]}?",
+                        "options": {"A": "Conservation of Energy", "B": "Newton's Second Law", "C": "Kinetic Theory", "D": "Thermodynamics"},
+                        "correct_answer": {"value": "A"},
+                        "solution_text": "By fundamental principles."
+                    }
+                ]
+                
+            question_ids = []
+            for q_data in questions_data:
+                new_q = models.Question(
+                    subject=q_data.get("subject", subjects[idx]),
+                    topic=q_data.get("topic", topics[idx]),
+                    subtopic=q_data.get("subtopic", "General"),
+                    difficulty=q_data.get("difficulty", "Medium"),
+                    type=models.QuestionTypeEnum.MCQ_SINGLE,
+                    question_text=q_data.get("question_text", ""),
+                    options=q_data.get("options", {}),
+                    correct_answer=q_data.get("correct_answer", {}),
+                    solution_text=q_data.get("solution_text", ""),
+                    language=models.MediumEnum.ENGLISH,
+                    exam=models.ExamEnum.JEE_MAIN,
+                    created_by=admin_id
+                )
+                session.add(new_q)
+                session.flush()
+                question_ids.append(str(new_q.id))
+                
+            # Create Test record
+            test_obj = models.Test(
+                title=f"AI DPP: {topics[idx]}",
+                type="DPS",
+                exam_pattern=models.ExamEnum.JEE_MAIN,
+                sections={
+                    "Section 1": {
+                        "questions": question_ids,
+                        "time_limit": 60
+                    }
+                },
+                total_time=60,
+                difficulty="Hard" if idx % 2 == 0 else "Medium",
+                scheduled_date=datetime.utcnow(),
+                is_active=True,
+                created_by=admin_id
+            )
+            session.add(test_obj)
+            session.commit()
+            print(f"Successfully created DPS Test: {test_obj.title} with {len(question_ids)} questions.")
+
+if __name__ == "__main__":
+    asyncio.run(seed_pdfs())
+