Update main.py

main.py

@@ -1,15 +1,12 @@
-# fastapi_app.py
-
 from fastapi import FastAPI, HTTPException
 from pydantic import BaseModel
-from typing import Optional
 from langchain.document_loaders import WikipediaLoader
 from langchain_groq import ChatGroq
 import os
 
 app = FastAPI(title="Quiz Generator API")
 
-# in‑memory store for the last quiz + context
+# in-memory store for the last quiz + context
 STORE = {
     "quiz": None,      # str
     "context": None,   # str
@@ -17,10 +14,9 @@ STORE = {
 
 # Replace with your actual Groq API key
 GROQ_API_KEY = os.getenv('api_key')
+
 class QuizRequest(BaseModel):
     search_query: str
-    complexity: str = "Hard"
-    quiz_type: str = "MCQs"
 
 class GradeRequest(BaseModel):
     answers: str
@@ -67,11 +63,8 @@ Instructions:
 context:
 {context_text}
 
-quiz complexity:
-{request.complexity}
-
-quiz type:
-{request.quiz_type}
+question:
+generate quiz on {request.search_query}
 
 Your response:
 """