Update models.py

models.py

@@ -284,7 +284,7 @@ def add_question_to_db(theme, question, correct_option, difficulty):
       
         theme_check_query = "SELECT COUNT(*) FROM themes WHERE theme = %s"
         cursor.execute(theme_check_query, (theme,))
-        theme_exists = cursor.fetchone()[0]
+        theme_exists = cursor.fetchone()["COUNT(*)"]
 
         if theme_exists == 0:
             insert_query = "INSERT INTO themes (theme, theme_bank, theme_quiz_table) VALUES (%s, %s, %s)"
@@ -324,9 +324,9 @@ def add_question_to_db(theme, question, correct_option, difficulty):
         """
         cursor.execute(theme_table_creation_query)
 
-        check_query = f"SELECT COUNT(*) FROM '{safe_theme}_bank' WHERE question_by_llm = %s"
+        check_query = f"SELECT COUNT(*) FROM `{safe_theme}_bank` WHERE question_by_llm = %s"
         cursor.execute(check_query, (question,))
-        is_duplicate = cursor.fetchone()[0] > 0
+        is_duplicate = cursor.fetchone()["COUNT(*)"] > 0
 
         if is_duplicate:
             logging.warning("Duplicate question detected for theme %s: %s", safe_theme, question)