Update utils/quiz_generator.py

utils/quiz_generator.py

@@ -1,23 +1,17 @@
 import os
-from openai import OpenAI
-from dotenv import load_dotenv
+import requests
 
-# Load your API key from .env
-load_dotenv()
-api_key = os.getenv("OPENAI_API_KEY")
-client = OpenAI(api_key=api_key)
+HF_TOKEN = os.getenv("HF_TOKEN")
+API_URL = "https://api-inference.huggingface.co/models/google/flan-t5-base"
+headers = {"Authorization": f"Bearer {HF_TOKEN}"}
 
-def generate_quiz(topic, num_questions=5):
-    prompt = f"Generate {num_questions} multiple-choice questions from the following content. Include 4 options and indicate the correct answer:\n\n{topic}"
+def generate_quiz(topic, num_questions):
+    prompt = f"Generate {num_questions} MCQ questions based on the following content:\n{topic}"
+    
+    payload = {"inputs": prompt}
+    response = requests.post(API_URL, headers=headers, json=payload)
 
-    try:
-        response = client.chat.completions.create(
-            model="gpt-3.5-turbo",
-            messages=[
-                {"role": "system", "content": "You are a quiz-generating assistant."},
-                {"role": "user", "content": prompt}
-            ]
-        )
-        return response.choices[0].message.content.strip()
-    except Exception as e:
-        return f"Error generating quiz:\n{str(e)}"
+    if response.status_code != 200:
+        return f"❌ Error: {response.status_code} - {response.json()}"
+
+    return response.json()[0]["generated_text"]