Update utils/quiz_generator.py

utils/quiz_generator.py

@@ -1,42 +1,22 @@
-import openai
+from openai import OpenAI
 import os
 from dotenv import load_dotenv
 
 load_dotenv()
-openai.api_key = os.getenv("OPENAI_API_KEY")
-
-
-def generate_quiz(text, num_questions=5, engine="gpt-3.5-turbo"):
-    prompt = f"""
-You are a quiz creator.
-Generate {num_questions} multiple choice questions from the following content.
-Each question should have 4 options (a, b, c, d), and the correct answer should be indicated clearly.
-
-Content:
-{text}
-
-Format:
-Q1. Question text?
-a) Option A
-b) Option B
-c) Option C
-d) Option D
-Answer: a
-
-Q2. ...
-    """
+client = OpenAI(api_key=os.getenv("OPENAI_API_KEY"))
 
+def generate_quiz(topic, num_questions=5):
+    prompt = f"Generate {num_questions} multiple choice questions with 4 options and answers from this content:\n\n{topic}"
+    
     try:
-        response = openai.ChatCompletion.create(
-            model=engine,
+        response = client.chat.completions.create(
+            model="gpt-3.5-turbo",
             messages=[
                 {"role": "system", "content": "You are a helpful quiz generator."},
                 {"role": "user", "content": prompt}
-            ],
-            temperature=0.7,
-            max_tokens=800
+            ]
         )
-        quiz_text = response['choices'][0]['message']['content']
-        return quiz_text
+        quiz = response.choices[0].message.content.strip()
+        return quiz
     except Exception as e:
-        return f"Error generating quiz: {str(e)}"
+        return f"Error generating quiz:\n{e}"