feat: implement dynamic topic management in question generation

Author: dcloud347

main.py

@@ -6,6 +6,14 @@
 from app.ai.questionGenerator import QuestionGenerator
 
 
+class TopicFrame(ttk.Frame):
+    def __init__(self, parent, remove_callback):
+        super().__init__(parent)
+        self.entry = ttk.Entry(self)
+        self.entry.pack(side=tk.LEFT, padx=(0, 5))
+        self.remove_btn = ttk.Button(self, text="X", width=2, command=lambda: remove_callback(self))
+        self.remove_btn.pack(side=tk.LEFT)
+
 def generate_questions():
     try:
         number = int(num_questions.get())
@@ -21,77 +29,87 @@ def generate_questions():
         messagebox.showerror("Error", "Please enter a valid subject.")
         return
 
-    topic = topic_entry.get().strip()
+    topics = [frame.entry.get().strip() for frame in topic_frames if frame.entry.get().strip()]
     age_group = age_group_entry.get()
 
     if not age_group:
         messagebox.showerror("Error", "Age Group cannot be empty.")
         return
 
-    # Initialize the question generator
     result_text.set("Generating questions...\nIt may take a minute to generate quiz questions.")
     root.update_idletasks()
 
+    all_questions = []
     try:
-        questions = genAi.generateQuestions(number, subject, age_group, topic if topic else None)
+        if topics:
+            for topic in topics:
+                questions = genAi.generateQuestions(number, subject, age_group, topic)
+                all_questions.extend(questions)
+        else:
+            questions = genAi.generateQuestions(number, subject, age_group)
+            all_questions.extend(questions)
         result_text.set("Finished generating questions!")
     except Exception as e:
         messagebox.showerror("Error", f"An error occurred: {e}")
         return
 
-    # Save to JSON
     with open("output.json", "w") as file:
-        file.write(json.dumps([question.__dict__ for question in questions], indent=4))
+        file.write(json.dumps([question.__dict__ for question in all_questions], indent=4))
 
     messagebox.showinfo("Success", "Questions generated and saved to output.json!")
 
 
+def add_topic_field():
+    frame = TopicFrame(topics_container, remove_topic_field)
+    frame.pack(pady=2)
+    topic_frames.append(frame)
+
+
+def remove_topic_field(frame):
+    frame.destroy()
+    topic_frames.remove(frame)
+
 if __name__ == "__main__":
-    # Initialize the Question Generator
     genAi = QuestionGenerator()
+    topic_frames = []
 
-    # Create main window
     root = tk.Tk()
     root.title("Quiz Question Generator")
-    root.geometry("450x350")
+    root.geometry("450x500")
     root.configure(bg="#f0f0f0")
 
-    # Styling
     style = ttk.Style()
     style.configure("TButton", padding=6, relief="flat", background="#4CAF50", font=("Arial", 10, "bold"))
     style.configure("TLabel", font=("Arial", 10))
     style.configure("TEntry", padding=5)
     style.configure("TCombobox", padding=5)
 
-    # Number of Questions
     ttk.Label(root, text="Number of questions (1-5):", background="#f0f0f0").pack(pady=5)
     num_questions = ttk.Entry(root)
     num_questions.pack()
 
-    # Subject
     ttk.Label(root, text="Subject:", background="#f0f0f0").pack(pady=5)
     subject_var = tk.StringVar()
     subject_menu = ttk.Combobox(root, textvariable=subject_var, values=[s.name for s in Subject])
     subject_menu.pack()
 
-    # topic (Optional)
-    ttk.Label(root, text="Topic (Optional):", background="#f0f0f0").pack(pady=5)
-    topic_entry = ttk.Entry(root)
-    topic_entry.pack()
+    # Topics section
+    ttk.Label(root, text="Topics (Optional):", background="#f0f0f0").pack(pady=5)
+    topics_container = ttk.Frame(root)
+    topics_container.pack(pady=5)
+
+    add_topic_btn = ttk.Button(root, text="Add Topic", command=add_topic_field)
+    add_topic_btn.pack(pady=5)
 
-    # Age Group
     ttk.Label(root, text="Age Group:", background="#f0f0f0").pack(pady=5)
     age_group_entry = ttk.Entry(root)
     age_group_entry.pack()
 
-    # Generate Button
     generate_button = ttk.Button(root, text="Generate Questions", command=generate_questions)
     generate_button.pack(pady=15)
 
-    # Status Label
     result_text = tk.StringVar()
     status_label = ttk.Label(root, textvariable=result_text, background="#f0f0f0", font=("Arial", 10, "italic"))
     status_label.pack()
 
-    # Run the application
-    root.mainloop()
+    root.mainloop()