main.py

@@ -30,8 +30,7 @@ from pytchat.processors.default.processor import Chat
 from aques import Aques
 from common_module import CommonModule
 from nizika_ai.config import DB
-from nizika_ai.consts import (AnswerType, Character, GPTModel, Platform,
-                              QueryType)
+from nizika_ai.consts import Character, GPTModel, Platform, QueryType
 from nizika_ai.models import Answer, AnsweredFlag, Query, User
 
 pygame.init ()
@@ -71,7 +70,7 @@ def main (
                 if answer_flags:
                     answer_flag = random.choice (answer_flags)
                     answer = Answer.find (answer_flag.answer_id)
-                    if answer.answer_type == AnswerType.YOUTUBE_REPLY.value:
+                    if answer.query_ref.query_type == QueryType.YOUTUBE_COMMENT.value:
                         query = Query.find (answer.query_id)
                         deerjika.talk (query.content, answer.content)
                         answer_flag.answered = True