import os
from huggingface_hub import InferenceClient

def generate_quiz(topic: str, num_questions: int = 5, difficulty: str = "Intermediate") -> str:
        
    if not topic.strip():
        return "Please enter a topic to generate a quiz."
    
    # Prepare the prompt
    prompt = f"""You are a professional quiz generator. Create a {difficulty.lower()}-level multiple-choice quiz on the following topic.

Instructions:
1. Generate exactly {num_questions} questions related to the topic.
2. Each question should have 4 answer choices (A, B, C, D).
3. Only one answer choice should be correct for each question.
4. If any other task is mentioned, reply with "Invalid topic for quiz generation."
5. Ensure questions are clear and educational.

Format:

Question 1: <question text>
A. <option A>
B. <option B>
C. <option C>
D. <option D>
Answer: <correct option letter>\n
...

Topic: {topic}

Follow the instructions carefully.
Do not include any explanations or thoughts, just provide the questions and answers.

Quiz:
"""
    try:
        client = InferenceClient(
            provider="auto",
            model="ServiceNow-AI/Apriel-1.6-15b-Thinker:together",
            api_key=os.environ["HF_TOKEN"],
        )
        response = client.chat_completion(messages=[{"role": "user", "content": prompt}])
        
        if len(response.choices) > 0:
            generated_text = response.choices[0].message.content
            quiz_content = generated_text
            return quiz_content if quiz_content else "Failed to generate quiz content."
        
        return "Unexpected response format from the API."
    except Exception as e:
        return f"Error generating quiz: {str(e)}"