from flask import Flask, request, jsonify, send_from_directory
from SpaceGen.model import SpaceGenModel
from flask_cors import CORS

app = Flask(__name__, static_folder='.', static_url_path='')
CORS(app)
MODEL_PATH = "SpaceGen/SpaceGen_Large.keras"



model = SpaceGenModel(model_path=MODEL_PATH)
@app.route('/api/data', methods=['POST'])
def space_text():
    if request.is_json:
        data = request.get_json()
        corrupted_text = data.get('corrupted_text')
        spaced_text = model.fix_space(corrupted_text)
        return jsonify({'spaced_text': spaced_text})

@app.get('/')
def index():
    return send_from_directory('.', 'main.html')

import os

if __name__ == '__main__':
    port = int(os.environ.get("PORT", 7860))
    app.run(host="0.0.0.0", port=port, debug=False)