mirror of
https://github.com/OpenBMB/ChatDev.git
synced 2024-09-16 04:37:11 +03:00
69 lines
1.8 KiB
Python
69 lines
1.8 KiB
Python
import logging
|
|
import requests
|
|
import os
|
|
from flask import Flask, send_from_directory, request, jsonify
|
|
import argparse
|
|
|
|
app = Flask(__name__, static_folder='static')
|
|
app.logger.setLevel(logging.ERROR)
|
|
log = logging.getLogger('werkzeug')
|
|
log.setLevel(logging.ERROR)
|
|
messages = []
|
|
port = [8000]
|
|
|
|
def send_msg(role, text):
|
|
try:
|
|
data = {"role": role, "text": text}
|
|
response = requests.post(f"http://127.0.0.1:{port[-1]}/send_message", json=data)
|
|
except:
|
|
logging.info("flask app.py did not start for online log")
|
|
|
|
|
|
@app.route("/")
|
|
def index():
|
|
return send_from_directory("static", "index.html")
|
|
|
|
|
|
@app.route("/chain_visualizer")
|
|
def chain_visualizer():
|
|
return send_from_directory("static", "chain_visualizer.html")
|
|
|
|
|
|
@app.route("/replay")
|
|
def replay():
|
|
return send_from_directory("static", "replay.html")
|
|
|
|
|
|
@app.route("/get_messages")
|
|
def get_messages():
|
|
return jsonify(messages)
|
|
|
|
|
|
@app.route("/send_message", methods=["POST"])
|
|
def send_message():
|
|
data = request.get_json()
|
|
role = data.get("role")
|
|
text = data.get("text")
|
|
|
|
avatarUrl = find_avatar_url(role)
|
|
|
|
message = {"role": role, "text": text, "avatarUrl": avatarUrl}
|
|
messages.append(message)
|
|
return jsonify(message)
|
|
|
|
|
|
def find_avatar_url(role):
|
|
role = role.replace(" ", "%20")
|
|
avatar_filename = f"avatars/{role}.png"
|
|
avatar_url = f"/static/{avatar_filename}"
|
|
return avatar_url
|
|
|
|
|
|
if __name__ == "__main__":
|
|
parser = argparse.ArgumentParser(description='argparse')
|
|
parser.add_argument('--port', type=int, default=8000, help="port")
|
|
args = parser.parse_args()
|
|
port.append(args.port)
|
|
print(f"Please visit http://127.0.0.1:{port[-1]}/ for the front-end display page. \nIn the event of a port conflict, please modify the port argument (e.g., python3 app.py --port 8012).")
|
|
app.run(host='0.0.0.0', debug=False, port=port[-1])
|