Update get_methods.py

get_methods.py

@@ -1,13 +1,11 @@
-from flask import jsonify
-import random
-from shared_data import shared_data
+from shared_data import get_db
 
 def get_users():
-    return jsonify({"users": shared_data["users"]})
-
-def random_model():
-    return jsonify({"selected_model": random.choice(shared_data["models"])})
+    db = get_db()
+    users = db.execute('SELECT * FROM users').fetchall()
+    return jsonify([dict(id=row[0], name=row[1], score=row[2]) for row in users])
 
 def get_user(user_id):
-    user = next((u for u in shared_data["users"] if u["id"] == user_id), None)
-    return jsonify(user) if user else (jsonify({"error": "User not found"}), 404)
+    db = get_db()
+    user = db.execute('SELECT * FROM users WHERE id = ?', (user_id,)).fetchone()
+    return jsonify(dict(id=user[0], name=user[1], score=user[2])) if user else (jsonify({"error": "User not found"}), 404)