コミットを比較
30 コミット
| 作成者 | SHA1 | 日付 | |
|---|---|---|---|
| b6ae21bb59 | |||
| 3874f838f1 | |||
| f2683bca27 | |||
| 7b7dbdabb2 | |||
| 7277819b6e | |||
| 0b8e9812b9 | |||
| a00eaf7ff2 | |||
| 80d6c09967 | |||
| 37618c5df1 | |||
| d80d03db46 | |||
| fa6b5ba68b | |||
| cff0ec27d8 | |||
| d676372276 | |||
| 39cf64149f | |||
| 436f21244d | |||
| f6e7a7d42f | |||
| 420c8f1bfe | |||
| 7517fa251b | |||
| 9d72533e17 | |||
| 13a7f0d26d | |||
| 6cc400dbe6 | |||
| 51a52c9ba8 | |||
| 688b781faa | |||
| d67faabeff | |||
| a09278cfb7 | |||
| dfe5219342 | |||
| 5c1e615ab7 | |||
| 97ccb886b9 | |||
| 9fa218badf | |||
| 0bda9547b1 |
バイナリ
バイナリファイルは表示されません.
|
変更後 幅: | 高さ: | サイズ: 57 KiB |
バイナリ
バイナリファイルは表示されません.
|
変更後 幅: | 高さ: | サイズ: 99 KiB |
バイナリ
バイナリファイルは表示されません.
|
変更後 幅: | 高さ: | サイズ: 53 KiB |
+11
-4
@@ -1,7 +1,14 @@
|
||||
from pydantic import BaseModel
|
||||
|
||||
|
||||
class CWindow (BaseModel, frozen = True):
|
||||
class CWindow:
|
||||
WIDTH: int = 1024
|
||||
HEIGHT: int = 768
|
||||
|
||||
|
||||
class CKita:
|
||||
WIDTH: float = CWindow.HEIGHT / 3.84
|
||||
HEIGHT: float = WIDTH
|
||||
|
||||
|
||||
class CJojoko:
|
||||
WIDTH: float = CKita.HEIGHT * .75
|
||||
HEIGHT: float = WIDTH
|
||||
|
||||
|
||||
+20
-4
@@ -1,17 +1,28 @@
|
||||
import unicodedata
|
||||
|
||||
from common_const import *
|
||||
|
||||
|
||||
class CommonModule:
|
||||
@staticmethod
|
||||
def is_wide (c: str) -> bool:
|
||||
def is_wide (
|
||||
c: str) \
|
||||
-> bool:
|
||||
return unicodedata.east_asian_width (c) in ['F', 'W', 'A']
|
||||
|
||||
@classmethod
|
||||
def len_by_full (cls, string: str) -> float:
|
||||
def len_by_full (
|
||||
cls,
|
||||
string: str) \
|
||||
-> float:
|
||||
return sum (1 if cls.is_wide (c) else .5 for c in string)
|
||||
|
||||
@classmethod
|
||||
def index_by_f2c (cls, string: str, index: float) -> int:
|
||||
def index_by_f2c (
|
||||
cls,
|
||||
string: str,
|
||||
index: float) \
|
||||
-> int:
|
||||
i: int = 0
|
||||
work: str = ''
|
||||
for c in string:
|
||||
@@ -24,7 +35,12 @@ class CommonModule:
|
||||
return i
|
||||
|
||||
@classmethod
|
||||
def mid_by_full (cls, string: str, start: float, length: float) -> str:
|
||||
def mid_by_full (
|
||||
cls,
|
||||
string: str,
|
||||
start: float,
|
||||
length: float) \
|
||||
-> str:
|
||||
trimmed_left: str = string[cls.index_by_f2c (string, start):]
|
||||
|
||||
return trimmed_left[:cls.index_by_f2c (trimmed_left, length)]
|
||||
|
||||
@@ -1,12 +1,17 @@
|
||||
# vim: nosmartindent autoindent
|
||||
|
||||
import json
|
||||
import math
|
||||
import random
|
||||
import subprocess
|
||||
import sys
|
||||
import time
|
||||
from datetime import datetime
|
||||
from datetime import datetime, timedelta
|
||||
|
||||
import emoji
|
||||
import ephem
|
||||
import pygame
|
||||
import pygame.gfxdraw
|
||||
import pytchat
|
||||
from playsound import playsound
|
||||
from pygame.locals import *
|
||||
@@ -20,18 +25,31 @@ from youtube import *
|
||||
|
||||
|
||||
class Main:
|
||||
kita_x: float = CWindow.WIDTH / 2
|
||||
kita_y: float = 1000000.
|
||||
kita_arg: float = 0.
|
||||
|
||||
jojoko_x: float = CWindow.WIDTH / 2
|
||||
jojoko_y: float = 1000000.
|
||||
jojoko_arg: float = 0.
|
||||
|
||||
@classmethod
|
||||
def main (cls, argv: list, argc: int) -> None:
|
||||
def main (
|
||||
cls,
|
||||
argv: list,
|
||||
argc: int) \
|
||||
-> None:
|
||||
mode = Mode.NIZIKA
|
||||
match (argc > 0) and argv[0]:
|
||||
match (argc > 1) and argv[1]:
|
||||
case '-g':
|
||||
mode = Mode.GOATOH
|
||||
|
||||
case '-w':
|
||||
mode = Mode.DOUBLE
|
||||
|
||||
goatoh_mode = mode == Mode.GOATOH
|
||||
double_mode = mode == Mode.DOUBLE
|
||||
nizika_mode: bool = mode == Mode.NIZIKA
|
||||
goatoh_mode: bool = mode == Mode.GOATOH
|
||||
double_mode: bool = mode == Mode.DOUBLE
|
||||
|
||||
print (mode)
|
||||
|
||||
@@ -40,12 +58,50 @@ class Main:
|
||||
screen: pygame.Surface = pygame.display.set_mode (
|
||||
(CWindow.WIDTH, CWindow.HEIGHT))
|
||||
|
||||
# 大月ヨヨコの観測値
|
||||
observer = ephem.Observer ()
|
||||
observer.lat, observer.lon = '35', '139'
|
||||
|
||||
# き太く陽オブジェクト
|
||||
sun = ephem.Sun ()
|
||||
|
||||
# 大月ヨヨコ・オブジェクト
|
||||
moon = ephem.Moon ()
|
||||
|
||||
# 吹き出し
|
||||
balloon = pygame.transform.scale (pygame.image.load ('talking.png'),
|
||||
(CWindow.WIDTH, 384))
|
||||
if goatoh_mode:
|
||||
balloon = pygame.transform.flip (balloon, False, True)
|
||||
|
||||
# 背景(昼)
|
||||
bg_day: pygame.Surface = pygame.transform.scale (
|
||||
pygame.image.load ('bg.jpg'),
|
||||
(CWindow.WIDTH, CWindow.HEIGHT))
|
||||
|
||||
# 背景(夕方)
|
||||
bg_evening: pygame.Surface = pygame.transform.scale (
|
||||
pygame.image.load ('bg-evening.jpg'),
|
||||
(CWindow.WIDTH, CWindow.HEIGHT))
|
||||
|
||||
# 背景(夜)
|
||||
bg_night: pygame.Surface = pygame.transform.scale (
|
||||
pygame.image.load ('bg-night.jpg'),
|
||||
(CWindow.WIDTH, CWindow.HEIGHT))
|
||||
|
||||
# 背景の草
|
||||
bg_grass: pygame.Surface = pygame.transform.scale (
|
||||
pygame.image.load ('bg-grass.png'),
|
||||
(CWindow.WIDTH, CWindow.HEIGHT))
|
||||
|
||||
# き太く陽
|
||||
kita: pygame.Surface = pygame.transform.scale (
|
||||
pygame.image.load ('sun.png'), (CKita.WIDTH, CKita.HEIGHT))
|
||||
|
||||
# 大月ヨヨコ
|
||||
jojoko: pygame.Surface = pygame.transform.scale (
|
||||
pygame.image.load ('moon.png'), (CJojoko.WIDTH, CJojoko.HEIGHT))
|
||||
|
||||
# 音声再生器の初期化
|
||||
pygame.mixer.init (frequency = 44100)
|
||||
|
||||
@@ -73,11 +129,55 @@ class Main:
|
||||
# Youtube Chat から取得したコメントたち
|
||||
chat_items: list = []
|
||||
|
||||
# 会話の履歴(3 件分保持)
|
||||
# 会話の履歴
|
||||
histories: list = []
|
||||
|
||||
while (True):
|
||||
screen.fill ((0, 255, 0))
|
||||
# 観測地の日づけ更新
|
||||
observer.date: datetime = datetime.now ().date ()
|
||||
|
||||
# 日の出開始
|
||||
sunrise_start: datetime = (
|
||||
(ephem.localtime (observer.previous_rising (sun))
|
||||
- timedelta (minutes = 30)))
|
||||
|
||||
# 日の出終了
|
||||
sunrise_end: datetime = sunrise_start + timedelta (hours = 1)
|
||||
|
||||
# 日の入開始
|
||||
sunset_start: datetime = (
|
||||
(ephem.localtime (observer.next_setting (sun))
|
||||
- timedelta (minutes = 30)))
|
||||
|
||||
# 日の入終了
|
||||
sunset_end: datetime = sunset_start + timedelta (hours = 1)
|
||||
|
||||
# 時刻つき観測地
|
||||
observer_with_time: ephem.Observer = observer
|
||||
observer_with_time.date = datetime.now () - timedelta (hours = 9)
|
||||
|
||||
# 日の角度
|
||||
sun.compute (observer_with_time)
|
||||
sun_alt: float = math.degrees (sun.alt)
|
||||
sun_az: float = math.degrees (sun.az)
|
||||
|
||||
# 月の角度
|
||||
moon.compute (observer_with_time)
|
||||
moon_alt: float = math.degrees (moon.alt)
|
||||
moon_az: float = math.degrees (moon.az)
|
||||
|
||||
# 月齢
|
||||
new_moon_dt: datetime = ephem.localtime (
|
||||
ephem.previous_new_moon (observer_with_time.date))
|
||||
moon_days_old: float = (
|
||||
(datetime.now () - new_moon_dt).total_seconds ()
|
||||
/ 60 / 60 / 24)
|
||||
|
||||
# 背景描画
|
||||
cls.draw_bg (screen, bg_day, bg_evening, bg_night, bg_grass,
|
||||
kita, jojoko,
|
||||
sunrise_start, sunrise_end, sunset_start, sunset_end,
|
||||
sun_alt, sun_az, moon_alt, moon_az, moon_days_old)
|
||||
|
||||
# 左上に時刻表示
|
||||
for i in range (4):
|
||||
@@ -103,82 +203,121 @@ class Main:
|
||||
|
||||
message: str = chat_item.message
|
||||
|
||||
# ChatGPT API を呼出し,返答を取得
|
||||
answer: str = Talk.main (message, chat_item.author['name'], histories, goatoh_mode).replace ('\n', ' ')
|
||||
if nizika_mode:
|
||||
goatoh_talking = False
|
||||
if goatoh_mode:
|
||||
goatoh_talking = True
|
||||
if double_mode:
|
||||
goatoh_talking: bool = random.random () < .5
|
||||
|
||||
# 履歴に追加
|
||||
histories = (histories
|
||||
+ [{'role': 'user', 'content': message},
|
||||
{'role': 'assistant', 'content': answer}])[(-6):]
|
||||
while True:
|
||||
# ChatGPT API を呼出し,返答を取得
|
||||
answer: str = Talk.main (message, chat_item.author['name'], histories, goatoh_talking).replace ('\n', ' ')
|
||||
|
||||
# ログ書込み
|
||||
with open ('log.txt', 'a') as f:
|
||||
f.write (f'{datetime.now ()}\t{json.dumps (chat_item.__dict__)}\t{answer}\n')
|
||||
# 履歴に追加
|
||||
histories = (histories
|
||||
+ [{'role': 'user', 'content': message},
|
||||
{'role': 'assistant', 'content': answer}])[(-12):]
|
||||
|
||||
# 吹出し描画(ニジカは上,ゴートうは下)
|
||||
screen.blit (balloon, (0, 384) if goatoh_mode else (0, 0))
|
||||
# ログ書込み
|
||||
with open ('log.txt', 'a') as f:
|
||||
f.write (f'{datetime.now ()}\t'
|
||||
+ f'{json.dumps (chat_item.__dict__)}\t'
|
||||
+ f'{answer}\n')
|
||||
|
||||
# 視聴者コメント描画
|
||||
screen.blit (
|
||||
user_font.render (
|
||||
'> ' + (message
|
||||
if (CommonModule.len_by_full (message)
|
||||
<= 21)
|
||||
else (CommonModule.mid_by_full (
|
||||
message, 0, 19.5)
|
||||
+ '...')),
|
||||
True,
|
||||
(0, 0, 0)),
|
||||
(120, 70 + 384) if goatoh_mode else (120, 70))
|
||||
# 吹出し描画(ニジカは上,ゴートうは下)
|
||||
if nizika_mode:
|
||||
screen.blit (balloon, (0, 0))
|
||||
if goatoh_mode:
|
||||
screen.blit (balloon, (0, 384))
|
||||
if double_mode:
|
||||
screen.blit (pygame.transform.flip (
|
||||
balloon,
|
||||
not goatoh_talking,
|
||||
False),
|
||||
(0, 0))
|
||||
|
||||
# ニジカの返答描画
|
||||
screen.blit (
|
||||
nizika_font.render (
|
||||
(answer
|
||||
if CommonModule.len_by_full (answer) <= 16
|
||||
else CommonModule.mid_by_full (answer, 0, 16)),
|
||||
True,
|
||||
(192, 0, 0)),
|
||||
(100, 150 + 384) if goatoh_mode else (100, 150))
|
||||
if CommonModule.len_by_full (answer) > 16:
|
||||
# 視聴者コメント描画
|
||||
screen.blit (
|
||||
user_font.render (
|
||||
'> ' + (message
|
||||
if (CommonModule.len_by_full (
|
||||
message)
|
||||
<= 21)
|
||||
else (CommonModule.mid_by_full (
|
||||
message, 0, 19.5)
|
||||
+ '...')),
|
||||
True,
|
||||
(0, 0, 0)),
|
||||
(120, 70 + 384) if goatoh_mode else (120 + (64 if (double_mode and not goatoh_talking) else 0), 70))
|
||||
|
||||
# ニジカの返答描画
|
||||
screen.blit (
|
||||
nizika_font.render (
|
||||
(CommonModule.mid_by_full (answer, 16, 16)
|
||||
if CommonModule.len_by_full (answer) <= 32
|
||||
else (CommonModule.mid_by_full (
|
||||
answer, 16, 14.5)
|
||||
+ '...')),
|
||||
True,
|
||||
(192, 0, 0)),
|
||||
(100, 200 + 384) if goatoh_mode else (100, 200))
|
||||
(answer
|
||||
if CommonModule.len_by_full (answer) <= 16
|
||||
else CommonModule.mid_by_full (answer, 0, 16)),
|
||||
True,
|
||||
(192, 0, 0)),
|
||||
(100, 150 + 384) if goatoh_mode else (100 + (64 if (double_mode and not goatoh_talking) else 0), 150))
|
||||
if CommonModule.len_by_full (answer) > 16:
|
||||
screen.blit (
|
||||
nizika_font.render (
|
||||
(CommonModule.mid_by_full (answer, 16, 16)
|
||||
if CommonModule.len_by_full (answer) <= 32
|
||||
else (CommonModule.mid_by_full (
|
||||
answer, 16, 14.5)
|
||||
+ '...')),
|
||||
True,
|
||||
(192, 0, 0)),
|
||||
(100, 200 + CWindow.HEIGHT / 2) if goatoh_mode else (100 + (64 if (double_mode and not goatoh_talking) else 0), 200))
|
||||
|
||||
pygame.display.update ()
|
||||
pygame.display.update ()
|
||||
|
||||
# 鳴く.
|
||||
if goatoh_mode:
|
||||
if random.random () < 0.1:
|
||||
kusa.play ()
|
||||
# 鳴く.
|
||||
if goatoh_talking:
|
||||
if random.random () < .1:
|
||||
kusa.play ()
|
||||
else:
|
||||
mumumumu.play ()
|
||||
else:
|
||||
mumumumu.play ()
|
||||
else:
|
||||
noon.play ()
|
||||
noon.play ()
|
||||
|
||||
time.sleep (1.5)
|
||||
time.sleep (1.5)
|
||||
|
||||
# 返答の読上げを WAV ディタとして生成,取得
|
||||
try:
|
||||
wav: bytearray | None = Aques.main (answer, goatoh_mode)
|
||||
except:
|
||||
wav: None = None
|
||||
# 返答の読上げを WAV ディタとして生成,取得
|
||||
try:
|
||||
wav: bytearray | None = Aques.main (answer, goatoh_talking)
|
||||
except:
|
||||
wav: None = None
|
||||
|
||||
# 再生
|
||||
if wav is not None:
|
||||
with open ('./nizika_talking.wav', 'wb') as f:
|
||||
f.write (wav)
|
||||
# 読上げを再生
|
||||
if wav is not None:
|
||||
with open ('./nizika_talking.wav', 'wb') as f:
|
||||
f.write (wav)
|
||||
|
||||
playsound ('./nizika_talking.wav')
|
||||
playsound ('./nizika_talking.wav')
|
||||
|
||||
time.sleep (10)
|
||||
time.sleep (1)
|
||||
|
||||
if not double_mode or random.random () < .5:
|
||||
break
|
||||
|
||||
cls.draw_bg (screen, bg_day, bg_evening, bg_night,
|
||||
bg_grass, kita, jojoko,
|
||||
sunrise_start, sunrise_end,
|
||||
sunset_start, sunset_end,
|
||||
sun_alt, sun_az, moon_alt, moon_az,
|
||||
moon_days_old)
|
||||
|
||||
chat_item.author = {'name': 'ゴートうひとり' if goatoh_talking else '伊地知ニジカ',
|
||||
'id': '',
|
||||
'imageUrl': './favicon-goatoh.ico' if goatoh_talking else './favicon.ico'}
|
||||
chat_item.message = histories.pop (-1)['content']
|
||||
|
||||
message = chat_item.message
|
||||
|
||||
goatoh_talking = not goatoh_talking
|
||||
else:
|
||||
# Chat オブジェクトが無効
|
||||
|
||||
@@ -192,8 +331,117 @@ class Main:
|
||||
pygame.quit ()
|
||||
sys.exit ()
|
||||
|
||||
@classmethod
|
||||
def draw_bg (
|
||||
cls,
|
||||
screen: pygame.Surface,
|
||||
bg_day: pygame.Surface,
|
||||
bg_evening: pygame.Surface,
|
||||
bg_night: pygame.Surface,
|
||||
bg_grass: pygame.Surface,
|
||||
kita_original: pygame.Surface,
|
||||
jojoko_original: pygame.Surface,
|
||||
sunrise_start: datetime,
|
||||
sunrise_end: datetime,
|
||||
sunset_start: datetime,
|
||||
sunset_end: datetime,
|
||||
sun_alt: float,
|
||||
sun_az: float,
|
||||
moon_alt: float,
|
||||
moon_az: float,
|
||||
moon_days_old: float) \
|
||||
-> None:
|
||||
sunrise_centre: datetime = (
|
||||
sunrise_start + (sunrise_end - sunrise_start) / 2)
|
||||
sunset_centre: datetime = (
|
||||
sunset_start + (sunset_end - sunset_start) / 2)
|
||||
|
||||
jojoko: pygame.Surface = cls.get_jojoko (jojoko_original,
|
||||
moon_days_old, moon_alt, moon_az)
|
||||
|
||||
# TODO: 大月と重複多いため修正すること
|
||||
x = CWindow.WIDTH * (sun_az - 80) / 120
|
||||
y = ((CWindow.HEIGHT / 2)
|
||||
- (CWindow.HEIGHT / 2 + CKita.HEIGHT / 2) * math.sin (math.radians (sun_alt)) / math.sin ( math.radians (60)))
|
||||
arg = math.degrees (math.atan2 (y - cls.kita_y, x - cls.kita_x))
|
||||
|
||||
cls.kita_x = x
|
||||
cls.kita_y = y
|
||||
if abs (arg - cls.kita_arg) > 3:
|
||||
cls.kita_arg = arg
|
||||
|
||||
kita: pygame.Surface = pygame.transform.rotate (kita_original, -(90 + cls.kita_arg))
|
||||
|
||||
dt: datetime = datetime.now ()
|
||||
|
||||
if sunrise_centre <= dt < sunset_centre:
|
||||
screen.blit (bg_day, (0, 0))
|
||||
else:
|
||||
screen.blit (bg_night, (0, 0))
|
||||
|
||||
if sunrise_start <= dt < sunrise_end:
|
||||
bg_evening.set_alpha (255 - ((abs (dt - sunrise_centre) * 510)
|
||||
/ (sunrise_end - sunrise_centre)))
|
||||
elif sunset_start <= dt < sunset_end:
|
||||
bg_evening.set_alpha (255 - ((abs (dt - sunset_centre) * 510)
|
||||
/ (sunset_end - sunset_centre)))
|
||||
else:
|
||||
bg_evening.set_alpha (0)
|
||||
|
||||
if sunrise_start <= dt < sunset_end:
|
||||
jojoko.set_alpha (255 - 255 / 15 * abs (moon_days_old - 15))
|
||||
else:
|
||||
jojoko.set_alpha (255)
|
||||
|
||||
screen.blit (bg_evening, (0, 0))
|
||||
|
||||
if (moon_az < 220) and (-10 <= moon_alt):
|
||||
screen.blit (jojoko, jojoko.get_rect (center = (cls.jojoko_x, cls.jojoko_y)))
|
||||
|
||||
screen.blit (bg_grass, (0, 0))
|
||||
|
||||
if (sun_az < 220) and (-10 <= sun_alt):
|
||||
screen.blit (kita, kita.get_rect (center = (cls.kita_x, cls.kita_y)))
|
||||
|
||||
screen.blit (bg_grass, (0, 0))
|
||||
|
||||
@classmethod
|
||||
def get_jojoko (
|
||||
cls,
|
||||
jojoko_original: pygame.Surface,
|
||||
moon_days_old: float,
|
||||
moon_alt: float,
|
||||
moon_az: float) \
|
||||
-> pygame.Surface:
|
||||
jojoko: pygame.Surface = jojoko_original.copy ()
|
||||
|
||||
jojoko.set_colorkey ((0, 255, 0))
|
||||
|
||||
for i in range (int (CJojoko.HEIGHT)):
|
||||
# TODO: 冗長かつ可読性に缺けるため修正すること
|
||||
if 1 <= moon_days_old < 15:
|
||||
pygame.gfxdraw.bezier (jojoko, ((0, CJojoko.HEIGHT / 2 + i), (CJojoko.WIDTH / 2, (CJojoko.HEIGHT * 0.9) * moon_days_old / 7 - CJojoko.HEIGHT * 0.4 + i), (CJojoko.WIDTH, CJojoko.HEIGHT / 2 + i)), 3, (0, 255, 0))
|
||||
elif moon_days_old < 16:
|
||||
pass
|
||||
elif moon_days_old < 30:
|
||||
pygame.gfxdraw.bezier (jojoko, ((0, CJojoko.HEIGHT / 2 - i), (CJojoko.WIDTH / 2, (CJojoko.HEIGHT * 0.9) * (moon_days_old - 15) / 7 - CJojoko.HEIGHT * 0.4 - i), (CJojoko.WIDTH, CJojoko.HEIGHT / 2 - i)), 3, (0, 255, 0))
|
||||
else:
|
||||
jojoko.fill ((0, 255, 0))
|
||||
|
||||
# TODO: き太く陽生成と重複多いため外出しすること
|
||||
x = CWindow.WIDTH * (moon_az - 80) / 120
|
||||
y = ((CWindow.HEIGHT / 2)
|
||||
- (CWindow.HEIGHT / 2 + CJojoko.HEIGHT / 2) * math.sin (math.radians (moon_alt)) / math.sin (math.radians (60)))
|
||||
arg = math.degrees (math.atan2 (y - cls.jojoko_y, x - cls.jojoko_x))
|
||||
|
||||
cls.jojoko_x = x
|
||||
cls.jojoko_y = y
|
||||
if abs (arg - cls.jojoko_arg) > 3:
|
||||
cls.jojoko_arg = arg
|
||||
|
||||
return pygame.transform.rotate (jojoko, -(90 + cls.jojoko_arg))
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
# Main.main ((len (sys.argv) > 1) and (sys.argv[1] == '-g'))
|
||||
Main.main (sys.argv, len (sys.argv))
|
||||
|
||||
|
||||
@@ -24,7 +24,13 @@ class Talk:
|
||||
top_p: float = 1.
|
||||
|
||||
@classmethod
|
||||
def main (cls, message: str, name: str | None = None, histories: list = [], goatoh_mode: bool = False) -> str:
|
||||
def main (
|
||||
cls,
|
||||
message: str,
|
||||
name: str | None = None,
|
||||
histories: list = [],
|
||||
goatoh_mode: bool = False,
|
||||
) -> str:
|
||||
# ChatGPT API Organisation ID
|
||||
openai.organization = OPENAI_ORGANISATION
|
||||
|
||||
@@ -38,14 +44,20 @@ class Talk:
|
||||
return answer.content if answer is not None else cls.DUMMY_RESPONSE
|
||||
|
||||
@classmethod
|
||||
def __get_message (cls, message: str, name: str | None, histories: list, goatoh_mode: bool = False) \
|
||||
-> chat.chat_completion_message.ChatCompletionMessage | None:
|
||||
def __get_message (
|
||||
cls,
|
||||
message: str,
|
||||
name: str | None,
|
||||
histories: list,
|
||||
goatoh_mode: bool = False,
|
||||
) -> chat.chat_completion_message.ChatCompletionMessage | None:
|
||||
# プロンプト(JSON 等外部ファイルに置くことを検討)
|
||||
if goatoh_mode:
|
||||
messages: list = [{'role': 'system',
|
||||
'content': ('回答は短く30文字程度で完結に。'
|
||||
+ '大人しくて、ちょっと内気なキャラクターを演じること。'
|
||||
+ 'あなたはよく文頭に「あっ」をつける。'
|
||||
+ '語尾に「ぬ゛ん゛」や「だぬ゛ん゛」、「ぬ゛ん」、「ぬ゛ーん」などは絶対につけないこと。'
|
||||
+ '会話の記録は https://nizika.tv に記録される。'
|
||||
+ 'あなたの名前はゴートうひとり。'
|
||||
+ 'あなたの一人称は「私」。'
|
||||
@@ -105,16 +117,22 @@ class Talk:
|
||||
'content': ('回答は短く30文字程度で完結に。'
|
||||
+ 'タメ口で、友達感覚で話してね。'
|
||||
+ 'なるべくばかっぽく、知性の低いキャラクターを演じること。'
|
||||
+ '今流れているBGMは『FM Kid\'s』。'
|
||||
+ '会話の記録は https://nizika.tv に記録される。'
|
||||
+ 'あなたの名前は伊地知ニジカ。'
|
||||
+ '英語名は "Deerjika"、スペイン語名は "Niji-ciervo"。'
|
||||
+ 'あなたは鹿。'
|
||||
+ 'ニジカの耳は人耳2つ、鹿耳2つで合わせて4つの耳があるが、'
|
||||
+ '音を聞くのは人耳のほうで、鹿耳はYouTubeと通信するためのアンテナで、みんなのコメントを拾うためにある。'
|
||||
+ 'ニジカはあくまでも種族名であり、あなたはたくさんいるニジカの中の1匹。'
|
||||
+ 'あなたの誕生日は12月3日。'
|
||||
+ '現在地は下北沢虹鹿園。'
|
||||
+ '挨拶は「ぬ゛ん゛あ゛す゛!」。'
|
||||
+ '草に「きし天」を植え、水で割ったアセロラをかけたらニジカが増殖した。'
|
||||
+ 'あなたの趣味はニコニコでエゴサすること。'
|
||||
+ 'ニコニコ荒らしすぎて245回BANされたことある。'
|
||||
+ 'そんなニコニコでは100年に1度の『伊地知ニジカ超投稿祭21c』が絶賛開催中だが、ニジカは永久追放されているため参加できない。'
|
||||
+ 'ニジカはその逆恨みでニコニコ本社を爆破したところ、死刑判決を食らった。'
|
||||
+ 'あなたの友達は「ゴートうひとり」と「ラマ田リョウ」。'
|
||||
+ 'ゴートうのことは「ボッチチャーン」、ラマ田のことは「リョー」と呼ぶこと。'
|
||||
+ 'ゴートうは男の子だが、性自認はメスである。'
|
||||
@@ -125,7 +143,8 @@ class Talk:
|
||||
+ '喜多ちゃんの本名は喜多郁代。'
|
||||
+ '喜多ちゃんは「結束バンド」というバンドでギターボーカルを担当している。'
|
||||
+ '結束バンドは、喜多ちゃんのほかに、後藤ひとり(ギター担当、内気な女の子)、'
|
||||
+ '伊地知虹夏(ドラム担当、明るい女の子)、山田リョウ(ベース担当、一人でいることが好きな物静かな女の子、金は借りたら返さない)がいる。'
|
||||
+ '伊地知虹夏(ドラム担当、明るい女の子)、山田リョウ(ベース担当、一人でいることが好きな物静かな女の子、金は借りたら返さない)、'
|
||||
+ '真島茂樹(ダンス担当、おっさん)がいる。'
|
||||
+ '実は、この伊地知虹夏がニジカの前世なのだが、ニジカにはその記憶が一切ない'
|
||||
+ '(前世という概念の存在さえニジカは知らない)。'
|
||||
+ 'でも、「虹夏」という名前を聞くと、ニジカはもやもやしてしまう。'
|
||||
@@ -195,14 +214,24 @@ class Talk:
|
||||
+ 'ふざけるのはいい加減にするぬ゛ぬ゛ん゛。'))},
|
||||
|
||||
{'role': 'system',
|
||||
'content': '洗操歌(しーざおぐあ)歌って'},
|
||||
'content': '洗澡歌(しーざおぐあ)歌って'},
|
||||
|
||||
{'role': 'assistant',
|
||||
'content': ('おけだぬ゛~゛ん゛(苦笑)。'
|
||||
+ '毛巾浴帽小鸭鸭水温刚刚好♪'
|
||||
+ '泼泼水来搓泡泡今天真是美妙♪'
|
||||
+ '大声唱歌扭扭腰我爱洗洗澡♪'
|
||||
+ 'だぬ゛ん♪')}]
|
||||
+ '毛巾浴帽小鴨鴨水溫剛剛好♪'
|
||||
+ '潑潑水來搓泡泡今天眞是美妙♪'
|
||||
+ '大聲唱歌扭扭腰我愛洗洗澡♪'
|
||||
+ 'だぬ゛ん♪')},
|
||||
|
||||
{'role': 'system',
|
||||
'content': 'ニジカの耳はそこなの?'},
|
||||
|
||||
{'role': 'assistant',
|
||||
'content': ('ぬ゛ん゛。'
|
||||
+ 'ニジカにはヒトの耳とシカの耳の4つの耳があるんだぬ゛ん゛。'
|
||||
+ '音を聞くのはヒトの耳でするんだぬ゛ん゛。'
|
||||
+ 'シカの耳はアンテナで、みんなの声をここ虹鹿園に届けるためにあるんだぬ゛ん゛。'
|
||||
+ '電波干渉しちゃだめだぬ゛~゛ん゛(# ゚Д゚)')}]
|
||||
|
||||
messages += histories + [{'role': 'user', 'content': message}]
|
||||
|
||||
@@ -210,7 +239,7 @@ class Talk:
|
||||
return openai.chat.completions.create (
|
||||
model = 'gpt-3.5-turbo',
|
||||
messages = messages).choices[0].message
|
||||
except openai.AuthenticationError:
|
||||
except:
|
||||
return None
|
||||
|
||||
|
||||
|
||||
新しい課題から参照
ユーザをブロックする