--- a
+++ b/prediction_api/src/controller.py
@@ -0,0 +1,25 @@
+from flask import Blueprint, request, jsonify
+from src.prediction_system import get_prediction, healthcheck
+import json
+
+model_controller = Blueprint('controller', __name__)
+
+@model_controller.route('/predict', methods=['POST'])
+def prediction_endpoint():
+    if not request.is_json:
+        return jsonify({'error': 'Invalid JSON data'}), 400
+
+    data = json.loads(request.data)
+
+    if 'array'in data:
+        response = get_prediction(data)
+        return jsonify(response)
+    else:
+        return jsonify({'error': 'Invalid or missing "array" field'}), 400
+
+
+@model_controller.route('/healthcheck', methods=['GET'])
+def healthcheck_endpoint():
+    return jsonify({'model': healthcheck()}), 400
+    
+