Create quiz_generator.py

utils/quiz_generator.py

@@ -5,29 +5,38 @@ from dotenv import load_dotenv
 load_dotenv()
 openai.api_key = os.getenv("OPENAI_API_KEY")
 
-def generate_quiz(topic_text, num_questions=5):
+
+def generate_quiz(text, num_questions=5, engine="gpt-3.5-turbo"):
     prompt = f"""
-    Generate {num_questions} multiple-choice questions with 4 options each and mention the correct answer for the topic below.
-    
-    Topic:
-    {topic_text}
-    
-    Format:
-    Q1. Question?
-    a)
-    b)
-    c)
-    d)
-    Answer: <correct option>
+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. ...
     """
 
     try:
         response = openai.ChatCompletion.create(
-            model="gpt-3.5-turbo",  # You can also use text-davinci-003 if preferred
-            messages=[{"role": "user", "content": prompt}],
+            model=engine,
+            messages=[
+                {"role": "system", "content": "You are a helpful quiz generator."},
+                {"role": "user", "content": prompt}
+            ],
             temperature=0.7,
             max_tokens=800
         )
-        return response['choices'][0]['message']['content']
+        quiz_text = response['choices'][0]['message']['content']
+        return quiz_text
     except Exception as e:
-        return f"Error: {str(e)}"
+        return f"Error generating quiz: {str(e)}"