from flask import Flask, request, jsonify
from graphviz import Source
import os

app = Flask(__name__)


@app.route('/')
def hello_world():  # put application's code here
    return 'Hello World!'

@app.route('/convert-dot-to-png', methods=['POST'])
def convert_dot_to_png():
    # 从请求中获取文件名
    dot_filename = request.json.get('filename')

    if not dot_filename:
        return jsonify({'error': 'No filename provided'}), 400

    # 构建输出的PNG文件名
    base_name = os.path.splitext(dot_filename)[0]
    png_filename = f"{base_name}.png"

    try:
        # 读取.dot文件
        source = Source.from_file(dot_filename)
        # 渲染为.png图片
        source.render(filename=png_filename, format='png', cleanup=True)
        return jsonify({'message': f'{dot_filename} has been converted to {png_filename} successfully.'})
    except Exception as e:
        return jsonify({'error': str(e)}), 500

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