app.py

from flask import Flask, render_template, request, jsonify
from mst import kruskal_mst
from flask_sqlalchemy import SQLAlchemy
from flask_bootstrap import Bootstrap4
import click
from graph_store import db, add_graph, load_graphs, load_graph


app = Flask(__name__)

app.config['SQLALCHEMY_DATABASE_URI'] ='sqlite:///graph.db'
db.init_app(app)
bootstrap = Bootstrap4(app)



@app.route('/')
def index():
    return render_template('index.html')


@app.route('/map_mst')
def map_mst():
    return render_template('map_mst.html')


@app.route('/normal_mst')
def normal_mst():
    return render_template('normal_mst.html')


@app.route('/calculate', methods=['POST'])
def calculate_mst():
    data = request.get_json()
    edges = data.get('edges')
    print(edges)
    v = data.get('V')
    mst_edges, cost = kruskal_mst(v, edges)

    return jsonify(mst_edges)



@app.route('/save_to_database', methods=['POST'])
def save_to_database():
    data = request.get_json()
    edges = data.get('edges')
    nodes = data.get('nodes')
    graph_name = data.get('graphName')
    graph_description = data.get('graphDescription')

    return add_graph(nodes, edges, graph_name, graph_description)




@app.route('/load_graph_list', methods=['GET'])
def load_graph_list():
    return load_graphs()


@app.route('/load_selected_graph/<int:graph_id>', methods=['GET'])
def load_selected_graph(graph_id):
    return load_graph(graph_id)


@app.cli.command()  
@click.option('--drop', is_flag=True, help='Create after drop.')  
def initdb(drop):
    """Initialize the database."""
    if drop:  
        db.drop_all()
    db.create_all()
    click.echo('Initialized database.')  


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