Create flask_api.py

flask_api.py

@@ -0,0 +1,21 @@
+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)