From 960ce4b3cfbd3f749ef550010622df3815ad2ec8 Mon Sep 17 00:00:00 2001
From: miteruzo <miteruzo@naver.com>
Date: Tue, 24 Dec 2024 23:39:41 +0900
Subject: [PATCH] =?UTF-8?q?=E3=81=AF=E3=81=98=E3=81=BE=E3=82=8A=E3=81=AE?=
 =?UTF-8?q?=E5=A4=A7=E5=9C=B0?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .gitmodules           |  3 ++
 nizika_ai             |  1 +
 queries_to_answers.py | 91 +++++++++++++++++++++++++++++++++++++++++++
 3 files changed, 95 insertions(+)
 create mode 100644 .gitmodules
 create mode 160000 nizika_ai
 create mode 100644 queries_to_answers.py

diff --git a/.gitmodules b/.gitmodules
new file mode 100644
index 0000000..6cd7edd
--- /dev/null
+++ b/.gitmodules
@@ -0,0 +1,3 @@
+[submodule "nizika_ai"]
+	path = nizika_ai
+	url = https://git.miteruzo.com/miteruzo/nizika_ai
diff --git a/nizika_ai b/nizika_ai
new file mode 160000
index 0000000..9e136a7
--- /dev/null
+++ b/nizika_ai
@@ -0,0 +1 @@
+Subproject commit 9e136a7cb3baf97b384a07c4af31205d745e7200
diff --git a/queries_to_answers.py b/queries_to_answers.py
new file mode 100644
index 0000000..68c8670
--- /dev/null
+++ b/queries_to_answers.py
@@ -0,0 +1,91 @@
+from __future__ import annotations
+
+import random
+from datetime import datetime
+from typing import TypedDict
+
+from nizika_ai.config import DB
+from nizika_ai.consts import AnswerType, Character, Platform
+from nizika_ai.models import Answer, AnsweredFlag, Query, User
+from nizika_ai.talk import Talk
+
+
+def main (
+) -> None:
+    DB.begin_transaction ()
+    queries: list[Query] = Query.where ('answered', False).get ()
+    if not queries:
+        return
+    query: Query = random.choice (queries)
+    user: User = query.user
+    user_name: str | None = None
+    if query.user_id is not None:
+        user_name = user.name
+    histories: list[History] = []
+    for history in query.answer_histories:
+        if history.query is not None:
+            histories.append ({ 'role': 'user', 'content': history.query.content })
+        histories.append ({ 'role': 'assistant', 'content': history.content })
+    for character in [Character.DEERJIKA, Character.GOATOH]:
+        if query.target_character & character.value:
+            add_answer (query, character, user_name, histories)
+    query.answered = True
+    query.save ()
+    DB.commit ()
+
+
+def add_answer (
+        query:      Query,
+        character:  Character,
+        user_name:  str | None,
+        histories:  list[History],
+) -> None:
+    message: str | list[dict[str, str | dict[str, str]]]
+    if query.image_url is None:
+        message = query.content
+    else:
+        message = [{ 'type': 'text', 'text': query.content },
+                   { 'type': 'image_url', 'image_url': query.image_url }]
+    answer = Answer ()
+    answer.query_id = query.id
+    answer.character = character.value
+    answer.content = Talk.main (message, user_name, histories,
+                                goatoh_mode = character == Character.GOATOH)
+    answer.answer_type = query.query_type
+    answer.sent_at = datetime.now ()
+    answer.save ()
+    add_answered_flags (answer)
+
+
+def add_answered_flags (
+        answer: Answer,
+) -> None:
+    answer_type:    AnswerType
+    try:
+        answer_type = AnswerType (answer.answer_type)
+    except Exception:
+        return
+    if answer_type in [AnswerType.YOUTUBE_REPLY]:
+        add_answered_flag (answer, Platform.YOUTUBE)
+    if answer_type in [AnswerType.BLUESKY_REPLY]:
+        add_answered_flag (answer, Platform.BLUESKY)
+
+
+def add_answered_flag (
+        answer:     Answer,
+        platform:   Platform,
+) -> None:
+    answered_flag = AnsweredFlag ()
+    answered_flag.answer_id = answer.id
+    answered_flag.platform = platform.value
+    answered_flag.answered = False
+    answered_flag.save ()
+
+
+class History (TypedDict):
+    role:       str
+    content:    str
+
+
+if __name__ == '__main__':
+    main ()