From 985838f728f9596854eff598ddb8f6ad40dbf109 Mon Sep 17 00:00:00 2001
From: Laura Boenchen <laura.boenchen@limebit.de>
Date: Thu, 26 Jun 2025 10:40:26 +0200
Subject: [PATCH] Added prediction endpoint

---
 src/__pycache__/api.cpython-310.pyc | Bin 927 -> 0 bytes
 src/api.py                          |  25 ++++++++++++++++++++++++-
 2 files changed, 24 insertions(+), 1 deletion(-)
 delete mode 100644 src/__pycache__/api.cpython-310.pyc

diff --git a/src/__pycache__/api.cpython-310.pyc b/src/__pycache__/api.cpython-310.pyc
deleted file mode 100644
index afcbcfbb776d003e3d1df9553074c1595c478289..0000000000000000000000000000000000000000
GIT binary patch
literal 0
HcmV?d00001

literal 927
zcmZuvyKdVs6eTIirk_#TEbY_+T|6ZE0mBHIzNQpNS`;7(2&~B3aww4?<+SmduK9xw
z&e$L7ptVzfAyY3oBo0ynyt+KPm*?PI>d~l=VEumadC?&V{qcwG)q%qyZ1)p_Kmseo
z{Ff+P5k{(rMcAJs6;}yMLXK63btRGX60@F6WbYDVB)glwM9aZuC}NRZlWWXIvbP+I
zj_A&@pjSlnME?hZo}GXXZt<ZQ1w0M-_!i$0X~1^{K1Gw=zn~BuPe_}*QQV!i^h7$N
zYbPhTjgQ`+oVKyx4POuWs?qzEnI9DH{03(%di(kV)O0J^EVOWizh`@aFX#xiqYwlZ
zr1X5zF-sOwDb12^+A8r2S^xho+`P;_I%!>|_{#DxTGquv*4Za*&z#|f%wCmqU@VJF
z$a8JnegnE}$yMfTk#SQVn2R<kYazeycy=#?66@!+9%g|$56lk)V<-uNV@QHI<aqsf
zYG&`e-e;gdiRaVI%<jZ)Wj$zW2V3{=0T9?@Vh50~$5~@}S(o)Z_l`c`_q2gOg``0Z
z-jM5H+!YDNty6A{Dhu9}y3UqP*AvWWRl)!+jGRO^g1&ZikA!s_o%_-)rNQqYcNd(@
z@6|tnD}v~h*b(HDM=jBgC13s9u-?nS!ukQo3vHeCx>?8YT5!jR5iQO0npZN<+Xxgh
zY9$vryqY#PyjisIQkS(I`mV(4RU^Z2f=7Y#zf$S6(qg6L%umTZg+NhChL~bXDA|j~
LQ6G<EzsL9=#V^xW

diff --git a/src/api.py b/src/api.py
index a600480..8bc31db 100644
--- a/src/api.py
+++ b/src/api.py
@@ -1,8 +1,9 @@
 import os
 
-from flask import Flask, Response
+from flask import Flask, Response, request
 from flask_cors import CORS
 import pandas as pd
+import pickle
 
 app = Flask(__name__)
 
@@ -11,6 +12,11 @@
 
 training_data = pd.read_csv(os.path.join("data", "auto-mpg.csv"))
 
+# laden des Models
+file_to_open = open(os.path.join("data", "models", "regressor_mpg.pickle"), "rb")
+trained_model = pickle.load(file_to_open)
+file_to_open.close()
+
 
 @app.route("/", methods=["GET"])
 def index():
@@ -25,3 +31,20 @@ def hello_world():
 @app.route("/training_data", methods=["GET"])
 def get_training_data():
     return Response(training_data.to_json(), mimetype="application/json")
+
+
+@app.route("/predict", methods=["GET"])
+def predict():
+    # Lese die Variablen aus
+    zylinder = request.args.get("zylinder")
+    ps = request.args.get("ps")
+    gewicht = request.args.get("gewicht")
+    beschleunigung = request.args.get("beschleunigung")
+    baujahr = request.args.get("baujahr")
+
+    # Erstellen der Vorhersage
+    prediction = trained_model.predict(
+        [[int(zylinder), int(ps), int(gewicht), int(beschleunigung), int(baujahr)]]
+    )
+
+    return {"result": prediction[0]}