diff --git a/app/app.py b/app/app.py index 4e4d1ad..ad0b15d 100644 --- a/app/app.py +++ b/app/app.py @@ -113,7 +113,7 @@ def require_subscription(f): import hashlib def make_cache_key(model, prompt): - raw = f"{model}:{prompt}" + raw = f"{email}:{model}:{prompt}" return hashlib.sha256(raw.encode()).hexdigest() def get_cache(model, prompt): @@ -1176,7 +1176,8 @@ def generate(): user = db.query(User).filter_by(email=email).first() db.close() - task = enqueue_by_priority(user, cid, model, prompt) + #task = enqueue_by_priority(user, cid, model, prompt) + task = enqueue_by_priority(user.email, cid, model, prompt) return jsonify({"task_id": task.id})