ChatThinker/app.py at main · Rice9547/ChatThinker · 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
import os
from flask import Flask, request, abort
from linebot import LineBotApi, WebhookHandler
from linebot.exceptions import InvalidSignatureError
from linebot.models import MessageEvent, TextMessage, TextSendMessage
from dotenv import load_dotenv
from session_manager import SessionManager
from chat_processor_final import ChatProcessor

load_dotenv()

app = Flask(__name__)

line_bot_api = LineBotApi(os.getenv('LINE_CHANNEL_ACCESS_TOKEN'))
handler = WebhookHandler(os.getenv('LINE_CHANNEL_SECRET'))

session_manager = SessionManager()
chat_processor = ChatProcessor(session_manager)

@app.route("/")
def index():
    return """
    <!DOCTYPE html>
    <html>
    <head>
        <title>ChatThinker - LINE Bot 聊天優化機器人</title>
        <style>
            body { font-family: Arial, sans-serif; margin: 40px; }
            .container { max-width: 800px; margin: 0 auto; }
            .status { color: green; font-weight: bold; }
        </style>
    </head>
    <body>
        <div class="container">
            <h1>🤖 ChatThinker</h1>
            <p class="status">✅ 應用程式正在運行中</p>
            <h2>功能說明</h2>
            <p>這是一個 LINE Bot 聊天優化機器人，可以幫助你：</p>
            <ul>
                <li>生成更好的對話內容</li>
                <li>潤飾現有的對話草稿</li>
                <li>根據不同情境提供建議</li>
            </ul>
            <h2>使用方式</h2>
            <p>請在 LINE 中與機器人對話，支援以下指令：</p>
            <ul>
                <li><code>/new</code> - 開始新對話</li>
                <li><code>/more</code> - 生成更多內容</li>
            </ul>
            <h2>API 端點</h2>
            <ul>
                <li><code>/callback</code> - LINE Bot Webhook 端點</li>
            </ul>
        </div>
    </body>
    </html>
    """

@app.route("/callback", methods=['POST'])
def callback():
    signature = request.headers['X-Line-Signature']
    body = request.get_data(as_text=True)

    try:
        handler.handle(body, signature)
    except InvalidSignatureError:
        abort(400)

    return 'OK'

@handler.add(MessageEvent, message=TextMessage)
def handle_message(event):
    user_id = event.source.user_id
    user_message = event.message.text

    # 記錄狀態以便除錯
    current_state = session_manager.get_state(user_id)
    print(f"[用戶: {user_id[:8]}...] 狀態: {current_state}, 訊息: {user_message}")

    if user_message == '/new':
        session_manager.clear_session(user_id)
        reply_text = "開始新的對話！請告訴我：\n1. 你是誰？（例如：我是一個大學生）"
        session_manager.set_state(user_id, 'awaiting_user_identity')

    elif user_message == '/more':
        last_prompt = session_manager.get_last_prompt(user_id)
        if last_prompt:
            reply_text = chat_processor.generate_more(last_prompt)
        else:
            reply_text = "沒有找到之前的對話內容。請先開始一個新的對話（輸入 /new）"

    else:
        current_state = session_manager.get_state(user_id)

        if current_state is None:
            reply_text = "歡迎使用聊天優化機器人！\n\n請輸入 /new 開始新對話\n或輸入 /more 生成更多內容"

        elif current_state == 'awaiting_user_identity':
            session_manager.set_user_identity(user_id, user_message)
            session_manager.set_state(user_id, 'awaiting_target_identity')
            reply_text = f"了解，你是：{user_message}\n\n2. 請告訴我對話對象是誰？（例如：我的教授）"

        elif current_state == 'awaiting_target_identity':
            session_manager.set_target_identity(user_id, user_message)
            session_manager.set_state(user_id, 'awaiting_context')
            reply_text = f"了解，對象是：{user_message}\n\n3. 請描述對話情境（例如：請教課業問題）"

        elif current_state == 'awaiting_context':
            session_manager.set_context(user_id, user_message)
            session_manager.set_state(user_id, 'awaiting_past_conversation')
            reply_text = f"了解，情境是：{user_message}\n\n4. 請提供過去的對話紀錄（如果沒有，請輸入「無」）"

        elif current_state == 'awaiting_past_conversation':
            # 限制過去對話的長度，避免超過 LINE 訊息限制
            if len(user_message) > 500:
                user_message = user_message[:500] + "...(已截斷)"
            session_manager.set_past_conversation(user_id, user_message)
            session_manager.set_state(user_id, 'awaiting_mode_selection')
            reply_text = "資料收集完成！\n\n請選擇模式：\n1. 輸入「生成」- 我會直接為你生成對話內容\n2. 輸入「潤飾」- 請提供你的對話草稿，我會幫你優化"

        elif current_state == 'awaiting_mode_selection':
            if user_message.strip() == '生成':
                try:
                    session_data = session_manager.get_session_data(user_id)
                    reply_text = chat_processor.generate_conversation(session_data, user_id)
                    session_manager.set_state(user_id, 'conversation_complete')
                except Exception as e:
                    print(f"Error generating conversation: {e}")
                    reply_text = f"抱歉，生成對話時發生錯誤。請確認已設定 OpenAI API 金鑰。\n\n錯誤訊息：{str(e)[:100]}...\n\n請輸入 /new 重新開始"
            elif user_message.strip() == '潤飾':
                session_manager.set_state(user_id, 'awaiting_draft')
                reply_text = "請提供你的對話草稿："
            else:
                reply_text = f"請輸入「生成」或「潤飾」來選擇模式\n(你輸入的是：'{user_message}')"

        elif current_state == 'awaiting_draft':
            session_data = session_manager.get_session_data(user_id)
            reply_text = chat_processor.polish_conversation(session_data, user_message, user_id)
            session_manager.set_state(user_id, 'conversation_complete')

        elif current_state == 'conversation_complete':
            reply_text = "對話已完成！\n\n你可以：\n- 輸入 /more 生成更多內容\n- 輸入 /new 開始新對話"

        else:
            reply_text = "系統錯誤，請輸入 /new 重新開始"

    line_bot_api.reply_message(
        event.reply_token,
        TextSendMessage(text=reply_text)
    )

if __name__ == "__main__":
    app.run(debug=False, port=5000)