app.py

from flask import Flask, render_template, request, Response
from utils.extractor import fetch_data, extract_columns, filter_data, enrich_data, transform_to_json
import json

app = Flask(__name__)


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


@app.route("/api", methods=['GET', 'POST'])
def api():
    if request.method == 'GET':
        return 'GET'
    elif request.method == 'POST':

        start_date = request.form['start_date']
        end_date = request.form['end_date']

        tags_include = request.form['tags_include']
        try:
            tags_exclude = request.form['tags_exclude']
        except:
            tags_exclude = False
        try:
            fisbroker_check = request.form['fisbroker']
        except:
            fisbroker_check = False
        try:
            gsi_check = request.form['gsi']
        except:
            gsi_check = False

        try:
            datasets_list = fetch_data(start_date, end_date)
        except:
            return Response(status=400, mimetype='application/json')

        dataset_df = extract_columns(datasets_list)
        filtered_df = filter_data(dataset_df, tags_include, tags_exclude, fisbroker_check, gsi_check)
        enriched_data = enrich_data(filtered_df)
        final_json = transform_to_json(enriched_data)
        payload = json.dumps(final_json)

        return Response(payload, status=201, mimetype='application/json')

    


if __name__ == "__main__":
    from waitress import serve
    serve(app, host="0.0.0.0", port=3333)