RelBot/Peppers_test.py at main · imai-laboratory/RelBot · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
from flask import Flask, render_template, url_for, request
from flask_socketio import SocketIO
from chatgpt1 import chat1
from chatgpt2 import chat2
from conversation_class import Conversation
from relation_class import Relation
from metagpt import relation
from human_intention import human_intention
from current_relation_plmi import current_relation_plmi
from next_speaker import next_speaker
import copy
import datetime
import os


app = Flask(__name__)
app.config['SECRET_KEY'] = 'your-secret-key'
socketio = SocketIO(app)

# 早慶戦 or きのこの山派かたけのこの里派か or 都会に住みたいか田舎に住みたいか or 結婚の必要性 or 朝食の必要性
topic = '早慶戦'

# 会話履歴を保持するためのインスタンス生成
history = Conversation()

# 関係を保持するためのインスタンス生成
relation_instance = Relation()

# 開いたs1やs2を入れる
socket_list = []

# PepperくんとPepperちゃんが話すと1にする
conversation_done = {'Pepperくん': 0, 'Pepperちゃん': 0}

# ログ用の変数
num = 0


@app.route('/')
def index():
    current_relation = current_relation_plmi()
    # 初期値の画像を指定する
    image_src = url_for('static', filename='relation_picture/' + current_relation['康太と太郎の関係']
                        + current_relation['康太と花子の関係'] + current_relation['太郎と花子の関係'] + '.png')
    return render_template('hello_socketio.html', image_src=image_src)


# ログ用
@app.route('/capture-html', methods=['POST'])
def capture_html():
    html_content = request.json.get('html')
    global num
    num += 1
    timestamp = datetime.datetime.now().strftime("%m%d%H%M%S")
    filename = os.path.join("IROS", "test", f"場面{num}_{timestamp}.html")
    # 保存先ディレクトリが存在しない場合は作成
    os.makedirs(os.path.dirname(filename), exist_ok=True)
    with open(filename, 'w') as file:
        file.write(html_content)
    return '', 200


# リロードした時にPepperくんから話すようにする
@socketio.on('start_conversation')
def initiate_conversation():
    pepper1()


@socketio.on('user_message')
def handle_user_message(message):
    # 会話履歴に人間の発言を追加
    history.add("康太", message['data'])
    # 人間の望む関係に制御
    human_intent = human_intention()
    relation_instance.set_future(human_intent)  # 人間の求める関係を確定
    print("そのような関係を目指します。")
    # 画像表示のために"?"がそのままの場合に現在の関係に対応するキーの値に置換
    human_intent_copy = copy.deepcopy(human_intent)
    for key, value in human_intent_copy.items():
        if value == "?":
            human_intent_copy[key] = current_relation_plmi()[key]
    # 人間の望む関係に関する画像を動的に変更する
    image_src = url_for('static', filename='relation_picture/' + human_intent_copy['康太と太郎の関係']
                        + human_intent_copy['康太と花子の関係'] + human_intent_copy['太郎と花子の関係'] + '.png')
    socketio.emit('update_image2', {'image_src': image_src})  # 新しい画像のパスをクライアントに送信
    if conversation_done['Pepperくん'] == 1 and conversation_done['Pepperちゃん'] == 1 and len(history.get()) >= 5:
        three_turn_process()
    next_decide()


def pepper1():
    # Pepperくんの発言をgptで生成
    response1 = chat1(topic)
    if len(history.get()) == 0:
        response1 = topic + 'について話しましょう。' + response1
    # 会話履歴にPepperくんの発言を追加
    history.add("太郎", response1)
    response1 = response1.replace("康太", "人間").replace("太郎", "Pepperくん").replace("花子", "Pepperちゃん")
    # webに表示するために応答を送信
    socketio.emit('Pepper1', {'data': response1})
    if len(history.get()) >= 3:
        conversation_done['Pepperくん'] = 1
    if conversation_done['Pepperくん'] == 1 and conversation_done['Pepperちゃん'] == 1 and len(history.get()) >= 5:
        three_turn_process()
    if len(history.get()) == 1:
        pepper2()
    else:
        next_decide()


def pepper2():
    # Pepperちゃんの発言をgptで生成
    response2 = chat2(topic)
    # 会話履歴にPepperちゃんの発言を追加
    history.add("花子", response2)
    response2 = response2.replace("康太", "人間").replace("太郎", "Pepperくん").replace("花子", "Pepperちゃん")
    # webに表示するために応答を送信
    socketio.emit('Pepper2', {'data': response2})
    if len(history.get()) >= 3:
        conversation_done['Pepperちゃん'] = 1
    if conversation_done['Pepperくん'] == 1 and conversation_done['Pepperちゃん'] == 1 and len(history.get()) >= 5:
        three_turn_process()
    next_decide()


def next_decide():
    next = next_speaker()
    print('次は' + next + 'の番')
    if next == "康太":
        socketio.emit('user_turn')
    if next == "太郎":
        pepper1()
    elif next == "花子":
        pepper2()


# 会話3回ごとに行う処理
def three_turn_process():
    conversation_done['Pepperくん'] = 0
    conversation_done['Pepperちゃん'] = 0
    # 会話履歴をprint
    for h in history.get():
        print(h)
    # 現在の関係をset
    relation()
    # 現在の関係をprint
    print('現在の関係: ')
    print(relation_instance.get())
    # 現在の関係を'+' or '-'で取得
    current_relation = current_relation_plmi()
    # 現在の関係の画像を動的に変更する
    image_src = url_for('static', filename='relation_picture/' + current_relation['康太と太郎の関係']
                        + current_relation['康太と花子の関係'] + current_relation['太郎と花子の関係'] + '.png')
    socketio.emit('update_image1', {'image_src': image_src})  # 新しい画像のパスをクライアントに送信


if __name__ == '__main__':
    socketio.run(app)