from flask import Flask, render_template, request, jsonify
from parser import parse_source_to_graph
from dataset_gen import create_dataset_entry
import os

app = Flask(__name__)

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

@app.route('/parse', methods=['POST'])
def parse():
    data = request.get_json()
    code = data.get('code', '')
    if not code.strip():
        return jsonify({"nodes": [], "connections": []})
        
    result = parse_source_to_graph(code)
    return jsonify(result)

@app.route('/add_to_dataset', methods=['POST'])
def add_to_dataset():
    data = request.get_json()
    code = data.get('code', '')
    if not code.strip():
        return jsonify({"status": "error", "message": "Empty code"})
        
    result = create_dataset_entry(code)
    return jsonify(result)

if __name__ == '__main__':
    # Running on 0.0.0.0 to be accessible if deployed or in containers
    app.run(host="0.0.0.0", port=7860, debug=True)