stablediffusionapi / post_methods.py
yusufgundogdu's picture
Create post_methods.py
8d9e5e3 verified
raw
history blame
867 Bytes
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
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
except Exception as e:
return jsonify({"error": str(e)}), 500