Merge branch 'main' into btc-sounds

common_const.py

@@ -2,13 +2,3 @@ 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
-

common_module.py

@@ -1,28 +1,17 @@
 import unicodedata
 
-from common_const import *
-
 
 class CommonModule:
     @staticmethod
-    def is_wide (
+    def is_wide (c: str) -> bool:
-            c:  str) \
-    -> bool:
         return unicodedata.east_asian_width (c) in ['F', 'W', 'A']
 
     @classmethod
-    def len_by_full (
+    def len_by_full (cls, string: str) -> float:
-            cls,
-            string: str) \
-    -> float:
         return sum (1 if cls.is_wide (c) else .5 for c in string)
 
     @classmethod
-    def index_by_f2c (
+    def index_by_f2c (cls, string: str, index: float) -> int:
-            cls,
-            string: str,
-            index:  float) \
-    -> int:
         i: int = 0
         work: str = ''
         for c in string:
@@ -35,12 +24,7 @@ class CommonModule:
         return i
 
     @classmethod
-    def mid_by_full (
+    def mid_by_full (cls, string: str, start: float, length: float) -> str:
-            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)]

main.py

@@ -1,17 +1,12 @@
-# vim: nosmartindent autoindent
-
 import json
-import math
 import random
 import subprocess
 import sys
 import time
-from datetime import datetime, timedelta
+from datetime import datetime
 
 import emoji
-import ephem
 import pygame
-import pygame.gfxdraw
 import pytchat
 from playsound import playsound
 from pygame.locals import *
@@ -25,20 +20,8 @@ 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 (
+    def main (cls, argv: list, argc: int) -> None:
-            cls,
-            argv: list,
-            argc: int) \
-    -> None:
         mode = Mode.NIZIKA
         match (argc > 1) and argv[1]:
             case '-g':
@@ -58,50 +41,12 @@ 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)
 
@@ -129,55 +74,11 @@ 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):
@@ -217,13 +118,11 @@ class Main:
                         # 履歴に追加
                         histories = (histories
                                      + [{'role': 'user',      'content': message},
-                                        {'role': 'assistant', 'content': answer}])[(-12):]
+                                    {'role': 'assistant', 'content': answer}])[(-12):]
 
                         # ログ書込み
                         with open ('log.txt', 'a') as f:
-                            f.write (f'{datetime.now ()}\t'
+                            f.write (f'{datetime.now ()}\t{json.dumps (chat_item.__dict__)}\t{answer}\n')
-                                     + f'{json.dumps (chat_item.__dict__)}\t'
-                                     + f'{answer}\n')
 
                         # 吹出し描画（ニジカは上，ゴートうは下）
                         if nizika_mode:
@@ -241,8 +140,7 @@ class Main:
                         screen.blit (
                                 user_font.render (
                                     '> ' + (message
-                                            if (CommonModule.len_by_full (
+                                            if (CommonModule.len_by_full (message)
-                                                    message)
                                                 <= 21)
                                             else (CommonModule.mid_by_full (
                                                       message, 0, 19.5)
@@ -270,7 +168,7 @@ class Main:
                                                + '...')),
                                          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))
+                                    (100, 200 + 384) if goatoh_mode else (100 + (64 if (double_mode and not goatoh_talking) else 0), 200))
 
                         pygame.display.update ()
 
@@ -298,26 +196,20 @@ class Main:
 
                             playsound ('./nizika_talking.wav')
 
-                        time.sleep (1)
+                        time.sleep (10)
 
                         if not double_mode or random.random () < .5:
                             break
 
-                        cls.draw_bg (screen, bg_day, bg_evening, bg_night,
+                        screen.fill ((0, 255, 0))
-                                     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 '伊地知ニジカ',
+                        chat_item.author['name'] = 'ゴートうひとり' if goatoh_talking else '伊地知ニジカ'
-                                            'id':       '',
+                        chat_item.author['id'] = ''
-                                            'imageUrl': './favicon-goatoh.ico' if goatoh_talking else './favicon.ico'}
+                        chat_item.author['channelId'] = './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
+
+                        message = histories.pop (-1)['content']
             else:
                 # Chat オブジェクトが無効
 
@@ -331,117 +223,8 @@ 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))
 

talk.py

@@ -24,13 +24,7 @@ class Talk:
     top_p: float = 1.
 
     @classmethod
-    def main (
+    def main (cls, message: str, name: str | None = None, histories: list = [], goatoh_mode: bool = False) -> str:
-            cls,
-            message:        str,
-            name:           str | None  = None,
-            histories:      list        = [],
-            goatoh_mode:    bool        = False,
-    ) -> str:
         # ChatGPT API Organisation ID
         openai.organization = OPENAI_ORGANISATION
 
@@ -44,20 +38,14 @@ class Talk:
         return answer.content if answer is not None else cls.DUMMY_RESPONSE
 
     @classmethod
-    def __get_message (
+    def __get_message (cls, message: str, name: str | None, histories: list, goatoh_mode: bool = False) \
-            cls,
+            -> chat.chat_completion_message.ChatCompletionMessage | None:
-            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 に記録される。'
                                            + 'あなたの名前はゴートうひとり。'
                                            + 'あなたの一人称は「私」。'
@@ -117,22 +105,16 @@ 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』が絶賛開催中だが、ニジカは永久追放されているため参加できない。'
-                                           + 'ニジカはその逆恨みでニコニコ本社を爆破したところ、死刑判決を食らった。'
                                            + 'あなたの友達は「ゴートうひとり」と「ラマ田リョウ」。'
                                            + 'ゴートうのことは「ﾎﾞｯﾁﾁｬｰﾝ」、ラマ田のことは「ﾘｮｰ」と呼ぶこと。'
                                            + 'ゴートうは男の子だが、性自認はメスである。'
@@ -143,8 +125,7 @@ class Talk:
                                            + '喜多ちゃんの本名は喜多郁代。'
                                            + '喜多ちゃんは「結束バンド」というバンドでギターボーカルを担当している。'
                                            + '結束バンドは、喜多ちゃんのほかに、後藤ひとり（ギター担当、内気な女の子）、'
-                                           + '伊地知虹夏（ドラム担当、明るい女の子）、山田リョウ（ベース担当、一人でいることが好きな物静かな女の子、金は借りたら返さない）、'
+                                           + '伊地知虹夏（ドラム担当、明るい女の子）、山田リョウ（ベース担当、一人でいることが好きな物静かな女の子、金は借りたら返さない）がいる。'
-                                           + '真島茂樹（ダンス担当、おっさん）がいる。'
                                            + '実は、この伊地知虹夏がニジカの前世なのだが、ニジカにはその記憶が一切ない'
                                            + '（前世という概念の存在さえニジカは知らない）。'
                                            + 'でも、「虹夏」という名前を聞くと、ニジカはもやもやしてしまう。'
@@ -214,24 +195,14 @@ class Talk:
                                                  + 'ふざけるのはいい加減にするぬ゛ぬ゛ん゛。'))},
 
                               {'role':    'system',
-                               'content': '洗澡歌（しーざおぐあ）歌って'},
+                               'content': '洗操歌（しーざおぐあ）歌って'},
                             
                               {'role':    'assistant',
                                'content': ('おけだぬ゛～゛ん゛（苦笑）。'
-                                           + '毛巾浴帽小鴨鴨水溫剛剛好♪'
+                                           + '毛巾浴帽小鸭鸭水温刚刚好♪'
-                                           + '潑潑水來搓泡泡今天眞是美妙♪'
+                                           + '泼泼水来搓泡泡今天真是美妙♪'
-                                           + '大聲唱歌扭扭腰我愛洗洗澡♪'
+                                           + '大声唱歌扭扭腰我爱洗洗澡♪'
-                                           + 'だぬ゛ん♪')},
+                                           + 'だぬ゛ん♪')}]
-
-                              {'role':    'system',
-                               'content': 'ニジカの耳はそこなの？'},
-                               
-                              {'role':    'assistant',
-                               'content': ('ぬ゛ん゛。'
-                                           + 'ニジカにはヒトの耳とシカの耳の4つの耳があるんだぬ゛ん゛。'
-                                           + '音を聞くのはヒトの耳でするんだぬ゛ん゛。'
-                                           + 'シカの耳はアンテナで、みんなの声をここ虹鹿園に届けるためにあるんだぬ゛ん゛。'
-                                           + '電波干渉しちゃだめだぬ゛～゛ん゛(# ﾟДﾟ)')}]
 
         messages += histories + [{'role': 'user', 'content': message}]
 
@@ -239,7 +210,7 @@ class Talk:
             return openai.chat.completions.create (
                     model = 'gpt-3.5-turbo',
                     messages = messages).choices[0].message
-        except:
+        except openai.AuthenticationError:
             return None