Update main.py

main.py

@@ -25,19 +25,14 @@ embeddings = None
 vectorstore = None
 retriever = None
 quiz_chain = None
-grade_chain = None
 
-# Request schemas
+# Request schema
 class QuizRequest(BaseModel):
     topic: str
 
-class GradeRequest(BaseModel):
-    quiz: str
-    answers: str  # user-provided answers corresponding to quiz questions
-
 @app.on_event("startup")
 def load_components():
-    global llm, embeddings, vectorstore, retriever, quiz_chain, grade_chain
+    global llm, embeddings, vectorstore, retriever, quiz_chain
     try:
         api_key = os.getenv("API_KEY")
         if not api_key:
@@ -45,7 +40,7 @@ def load_components():
             raise RuntimeError("API_KEY environment variable is not set or empty.")
         logger.info("API_KEY is set.")
 
-        # 1) Init LLM & Embeddings
+        # Initialize LLM & Embeddings
         llm = ChatGroq(
             model="meta-llama/llama-4-scout-17b-16e-instruct",
             temperature=0,
@@ -58,7 +53,7 @@ def load_components():
             encode_kwargs={"normalize_embeddings": True},
         )
 
-        # 2) Load FAISS indexes
+        # Load FAISS indexes
         for zip_name, dir_name in [("faiss_index.zip", "faiss_index"), ("faiss_index(1).zip", "faiss_index_extra")]:
             if not os.path.exists(dir_name):
                 with zipfile.ZipFile(zip_name, 'r') as z:
@@ -78,11 +73,11 @@ def load_components():
 
         retriever = vectorstore.as_retriever(search_kwargs={"k": 3})
 
-        # Quiz generation chain (no answers provided)
+        # Quiz generation chain (with answers)
         quiz_prompt = PromptTemplate(
             template="""
 Generate a quiz on the topic "{topic}" using **only** the information in the "Retrieved context".
-Include clear questions and multiple-choice options (A, B, C, D), but do NOT provide the correct answers.
+Include clear questions, multiple-choice options (A, B, C, D), and **provide the correct answers** for each question.
 If context is insufficient, reply with "I don't know".
 
 Retrieved context:
@@ -91,7 +86,7 @@ Retrieved context:
 Quiz topic:
 {topic}
 
-Quiz questions:
+Quiz with answers:
 """,
             input_variables=["context", "topic"],
         )
@@ -104,33 +99,13 @@ Quiz questions:
         )
         logger.info("Quiz chain ready.")
 
-        # Grade quiz chain (combines quiz and answers into single question)
-        grade_prompt = PromptTemplate(
-            template="""
-Grade the following quiz based on the "Retrieved context":
-
-{question}
-
-Provide a score and brief feedback for each question. If context is insufficient to grade, reply "I don't know" for that question.
-""",
-            input_variables=["context", "question"],
-        )
-        grade_chain = RetrievalQA.from_chain_type(
-            llm=llm,
-            chain_type="stuff",
-            retriever=retriever,
-            return_source_documents=False,
-            chain_type_kwargs={"prompt": grade_prompt},
-        )
-        logger.info("Grading chain ready.")
-
     except Exception as e:
         logger.error("Error loading components", exc_info=True)
         raise
 
 @app.get("/")
 def root():
-    return {"message": "API is up and running!"}
+    return {"message": "Quiz API is up and running!"}
 
 @app.post("/quiz")
 def create_quiz(request: QuizRequest):
@@ -142,16 +117,3 @@ def create_quiz(request: QuizRequest):
     except Exception as e:
         logger.error("Error generating quiz", exc_info=True)
         raise HTTPException(status_code=500, detail=str(e))
-
-@app.post("/grade")
-def grade_quiz(request: GradeRequest):
-    try:
-        # Combine quiz questions and answers into one input
-        combined = f"Quiz questions:\n{request.quiz}\n\nUser answers:\n{request.answers}"
-        logger.info("Grading quiz. Combined input: %s", combined)
-        result = grade_chain.invoke({"question": combined})
-        logger.info("Quiz graded successfully.")
-        return {"grading": result.get("result")}
-    except Exception as e:
-        logger.error("Error grading quiz", exc_info=True)
-        raise HTTPException(status_code=500, detail=str(e))