from flask import Flask, request, jsonify, render_template_string

app = Flask(__name__)

# 主页
@app.route('/')
def home():
    return "AstrBot Web 端已启动！"

# 聊天接口
@app.route('/chat', methods=['POST'])
def chat():
    data = request.json
    user_input = data.get("message", "")
    response = f"AstrBot: {user_input}"  # 这里可以调用 AstrBot 的核心逻辑
    return jsonify({"response": response})  # 删除重复的 return 语句

# 预设用户名和密码
USERS = {"admin": "123456"}

# 登录页面
@app.route('/login', methods=['GET', 'POST'])
def login():
    if request.method == 'POST':
        username = request.form.get('username')
        password = request.form.get('password')
        if username in USERS and USERS[username] == password:
            return "登录成功！"
        else:
            return "登录失败，用户名或密码错误！"
    
    # 简单的 HTML 登录表单
    return render_template_string('''
        <form method="POST">
            用户名: <input type="text" name="username"><br>
            密码: <input type="password" name="password"><br>
            <input type="submit" value="登录">
        </form>
    ''')

if __name__ == '__main__':
    app.run(host='0.0.0.0', port=7860)