from langchain_groq import ChatGroq
from langchain.prompts import PromptTemplate

class QuizGenerator:
    def __init__(self, api_key):
        self.llm = ChatGroq(
            temperature=0.7,
            groq_api_key=api_key,
            model_name="llama3-8b-8192"  # Groq currently supports Llama2, not Llama3
        )
        
        self.quiz_prompt = PromptTemplate(
            input_variables=["content", "num_questions"],
            template="""
            Create {num_questions} multiple-choice questions based on this content:
            {content}
            
            For each question:
            1. Provide the question
            2. List 4 possible answers
            3. Indicate the correct answer
            4. Add a brief explanation
            """
        )
        
        self.chain = self.quiz_prompt | self.llm
    
    def generate_quiz(self, content, num_questions=5):
        return self.chain.invoke({
            "content": content,
            "num_questions": num_questions
        }).content