from flask import Flask, jsonify, send_from_directory, abort
import os
app = Flask(__name__)
DATASETS_DIR = os.path.abspath(os.getenv("DATASETS_DIR", "./datasets"))
@app.route('/health', methods=['GET'])
def health():
    return jsonify({"status": "OK", "service": "flask", "timestamp": __import__('datetime').datetime.utcnow().isoformat()})
@app.route('/datasets', methods=['GET'])
def list_datasets():
    if not os.path.isdir(DATASETS_DIR):
        return jsonify({"error": "Datasets directory not found"}), 404
    files = os.listdir(DATASETS_DIR)
    return jsonify({"datasets": files})
@app.route('/datasets/<filename>', methods=['GET'])
def get_dataset(filename):
    try:
        return send_from_directory(DATASETS_DIR, filename, as_attachment=True)
    except FileNotFoundError:
        abort(404)
if __name__ == '__main__':
    app.run(host="0.0.0.0", port=5000)