feat: quiz endpoint second varient

.gitignore

@@ -222,7 +222,8 @@ test/1000-data-science-questions-answers.json
 test/test.ipynb
 vector_db
 .gitignore
-
+chroma_store
+Backend/chroma_store
 test.ipynb
 .gitignore
 test.json
@@ -230,3 +231,4 @@ test.json
 #db files
 .sqlite3
 .bin
+chroma_store/chroma.sqlite3

Backend/app/api/v1/endpoints/quiz.py

@@ -2,12 +2,15 @@ from fastapi import APIRouter, Depends, HTTPException, status
 from chromadb import AsyncHttpClient
 from app.models import User
 from app.api.deps import get_db, get_current_user, get_chroma_client
-from app.schema import Quiz_input, QuizOutput
+from app.schema import Quiz_input, QuizOutput, IngestRequest
 from .prompts import SYSTEM_PROMPT
 from fastapi import APIRouter, Depends, HTTPException
 from chromadb.api.models.Collection import Collection # Import Collection type
 from app.api.deps import get_chroma_collection
 from app.llm import call_llm
+import uuid
+
+
 router = APIRouter(prefix="/quiz")
 
 async def search_logic(query: str, collection: Collection):
@@ -28,8 +31,8 @@ async def search_documents(
         raise HTTPException(500, f"ChromaDB Query Error: {e}")
 
 
-@router.post("/", response_model=QuizOutput, status_code=status.HTTP_201_CREATED)
-async def generate_quiz(
+@router.post("/resume", response_model=QuizOutput, status_code=status.HTTP_201_CREATED)
+async def generate_quiz_resume(
     Input_model: Quiz_input, 
     collection: Collection = Depends(get_chroma_collection), 
     current_user: User = Depends(get_current_user)
@@ -53,6 +56,67 @@ async def generate_quiz(
             detail=f'Invalid Input: {str(e)}'
         )
 
+
+async def ingest_logic(input_data:IngestRequest , collection: Collection):
+    doc_id = input_data.id if input_data.id else str(uuid.uuid4())
+
+    await collection.add(
+        ids = [doc_id],
+        documents=[input_data.parsed_doc],
+        metadatas=[{"user_prompt": input_data.user_prompt}]
+    )
+
+    return {
+        "status": "success",
+        "id": doc_id,
+        "stored_prompt": input_data.user_prompt
+    }
+
+@router.post("/ingest", status_code=status.HTTP_201_CREATED)
+async def ingest_data(
+    input_data: IngestRequest,
+    collection: Collection = Depends(get_chroma_collection), 
+    current_user: User = Depends(get_current_user)
+):
+    try: 
+        return await ingest_logic(input_data, collection)
+    except Exception as e:
+        raise HTTPException(
+        status_code=status.HTTP_500_INTERNAL_SERVER_ERROR,
+        detail= f"Ingestion failed: {str(e)}"
+    )
+
+
+
+@router.post("/notes", response_model=QuizOutput, status_code=status.HTTP_201_CREATED)
+async def generate_quiz_notes(
+    Input_model: IngestRequest, 
+    collection: Collection = Depends(get_chroma_collection), 
+    current_user: User = Depends(get_current_user)
+):
+    try:
+        notes = Input_model
+        await ingest_logic(notes, collection)
+
+        query = Input_model.user_prompt
+        retrieved_context = await search_logic(query, collection)
+        
+
+        if not retrieved_context:
+            raise ValueError("No context available to generate quiz.")
+        prompt = await prompt_builder(Input_model.parsed_doc, Input_model.user_prompt, retrieved_context)
+        
+        quiz_data_obj = await call_llm(prompt)
+
+        return quiz_data_obj
+
+    except Exception as e:
+        raise HTTPException(
+            status_code=status.HTTP_400_BAD_REQUEST,
+            detail=f'Invalid Input: {str(e)}'
+        )
+
+
 # #--------Helper Functions--------#
 
 

Backend/app/schema/__init__.py

@@ -1,3 +1,3 @@
-from app.schema.models import StudentCreate, StudentUpdate, StudentResponse, UserCreate, Token, LoginRequest, Quiz_input, QuizOutput
+from app.schema.models import StudentCreate, StudentUpdate, StudentResponse, UserCreate, Token, LoginRequest, Quiz_input, QuizOutput, IngestRequest
 
-__all__ = ["StudentCreate", "StudentUpdate", "StudentResponse", "UserCreate", "Token", "LoginRequest", "Quiz_input", "QuizOutput"]
+__all__ = ["StudentCreate", "StudentUpdate", "StudentResponse", "UserCreate", "Token", "LoginRequest", "Quiz_input", "QuizOutput", "IngestRequest"]

Backend/app/schema/models.py

@@ -61,4 +61,9 @@ class QuizQuestion(BaseModel):
     User_response: str = Field("", alias="User_response")
 
 class QuizOutput(BaseModel):
-    quiz: List[QuizQuestion] = Field(..., description="A list of 10 generated MCQ questions.")
+    quiz: List[QuizQuestion] = Field(..., description="A list of 10 generated MCQ questions.")
+
+class IngestRequest(BaseModel):
+    parsed_doc: str = Field(..., description="The main document content to embed")
+    user_prompt: str = Field(..., description="The user prompt associated with this document")
+    id: Optional[str] = None

Backend/app/vector_db.py

@@ -1,4 +0,0 @@
-import chromadb
-from chromadb.config import Settings
-
-chroma_client = None

RUN.md

@@ -1,7 +1,7 @@
-npm run dev (frontend)
-
-python run.py (Backend)
-
 sudo service postgresql start (POstgresql)
 
 chroma run --host 0.0.0.0 --port 8080 --path ./chroma_store (chroma_db) (project root folder)
+
+npm run dev (frontend)
+
+python run.py (Backend)