Update post_methods.py

post_methods.py

@@ -1,26 +1,22 @@
-from flask import jsonify, request
-import numpy as np
-from shared_data import shared_data
-
-def predict():
-    try:
-        if not request.json:
-            return jsonify({"error": "No data received"}), 400
-        return jsonify({"probability_up": np.random.rand()})
-    except Exception as e:
-        return jsonify({"error": str(e)}), 500
+from shared_data import get_db
 
 def add_user():
     try:
         if not request.json or "name" not in request.json:
             return jsonify({"error": "Name is required"}), 400
             
-        new_user = {
-            "id": len(shared_data["users"]) + 1,
-            "name": request.json["name"],
-            "score": request.json.get("score", 0)
-        }
-        shared_data["users"].append(new_user)
-        return jsonify({"message": "User added", "user": new_user}), 201
+        db = get_db()
+        cursor = db.cursor()
+        cursor.execute('INSERT INTO users (name, score) VALUES (?, ?)',
+                     (request.json["name"], request.json.get("score", 0)))
+        db.commit()
+        return jsonify({
+            "message": "User added",
+            "user": {
+                "id": cursor.lastrowid,
+                "name": request.json["name"],
+                "score": request.json.get("score", 0)
+            }
+        }), 201
     except Exception as e:
         return jsonify({"error": str(e)}), 500