Update models.py

models.py

@@ -236,7 +236,7 @@ def add_ques_llm(theme, question, options, correct_option, difficulty):
                 logging.error("Invalid theme provided: %s", theme)
                 return None
 
-            theme_bank = theme_bank_result[0]
+            theme_bank = theme_bank_result["theme_bank"]
             user_id_creator = get_jwt_identity()
         
             question_json = json.dumps({
@@ -246,7 +246,7 @@ def add_ques_llm(theme, question, options, correct_option, difficulty):
 
             check_query = f"SELECT COUNT(*) FROM {theme_bank} WHERE question_by_llm = %s"
             cursor.execute(check_query, (question_json,))
-            is_duplicate = cursor.fetchone()[0] > 0
+            is_duplicate = cursor.fetchone()["COUNT(*)"] > 0
 
             if is_duplicate:
                 logging.warning("Duplicate question detected for theme %s: %s", theme, question)
@@ -729,7 +729,7 @@ def add_theme_if_not_exists(theme):
         
         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: