Upload 4 files

Dockerfile

@@ -0,0 +1,10 @@
+FROM python:3.9
+
+WORKDIR /code
+
+COPY ./requirements.txt /code/requirements.txt
+RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
+
+COPY . .
+
+CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]

app.py

@@ -0,0 +1,191 @@
+# ⚙️ 后端逻辑/核心服务端.py
+from fastapi import FastAPI, HTTPException
+from fastapi.middleware.cors import CORSMiddleware
+from pydantic import BaseModel
+from typing import Optional
+import time
+import uuid
+
+# 导入刚刚写好的数据库操作模块
+import 数据库连接 as db
+
+app = FastAPI(title="ComfyUI Ranking Community API")
+
+# 允许跨域请求，这是 ComfyUI 能够访问云端的关键
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],
+    allow_credentials=True,
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
+
+# --- 增加根目录探针，防止 Hugging Face 健康检查失败 ---
+@app.get("/")
+def read_root():
+    return {"status": "ok", "message": "ComfyUI Ranking API is running!"}
+
+# --- 数据模型 ---
+class UserRegister(BaseModel):
+    name: str
+    gender: str
+    avatarDataUrl: Optional[str] = None
+    age: Optional[int] = None
+    country: Optional[str] = None
+    region: Optional[str] = None
+    intro: Optional[str] = None
+
+class InteractionToggle(BaseModel):
+    item_id: str
+    user_id: str # 使用 name 作为唯一标识 (简化版)
+    action_type: str # "like", "favorite"
+    is_active: bool
+
+class CommentCreate(BaseModel):
+    item_id: str
+    author: str
+    content: str
+    reply_to_user: Optional[str] = None
+    parent_id: Optional[str] = None # 如果是回复，记录父评论ID
+
+# --- 1. 用户系统 ---
+@app.post("/api/users/register")
+async def register_user(user: UserRegister):
+    users_db = db.load_data("users.json", default_data={})
+    
+    if user.name in users_db:
+        raise HTTPException(status_code=400, detail="该用户名已被注册")
+    
+    # 构建用户档案，初始化各项统计为 0
+    new_user = user.dict()
+    new_user.update({
+        "receivedLikes": 0, "receivedFollows": 0,
+        "privacy": {"follows": False, "likes": False, "favorites": False, "downloads": False}
+    })
+    
+    users_db[user.name] = new_user
+    db.save_data("users.json", users_db)
+    
+    return {"status": "success", "message": "注册成功", "data": new_user}
+
+@app.get("/api/users/{username}")
+async def get_user_profile(username: str):
+    users_db = db.load_data("users.json", default_data={})
+    if username not in users_db:
+        raise HTTPException(status_code=404, detail="用户不存在")
+    return {"status": "success", "data": users_db[username]}
+
+# --- 2. 排行榜核心数据流 ---
+@app.get("/api/items")
+async def get_items(type: str = "tool", sort: str = "time", limit: int = 20):
+    items_db = db.load_data("items.json", default_data=[])
+    
+    # 过滤类型
+    filtered_items = [item for item in items_db if item.get("type") == type]
+    
+    # 排序逻辑
+    if sort == "likes":
+        filtered_items.sort(key=lambda x: x.get("likes", 0), reverse=True)
+    elif sort == "favorites":
+        filtered_items.sort(key=lambda x: x.get("favorites", 0), reverse=True)
+    elif sort == "downloads":
+        filtered_items.sort(key=lambda x: x.get("uses", 0), reverse=True)
+    else: # time (默认)
+        filtered_items.sort(key=lambda x: x.get("created_at", 0), reverse=True)
+        
+    return {"status": "success", "data": filtered_items[:limit]}
+
+# --- 3. 社交与互动 (点赞/收藏防抖与排重) ---
+@app.post("/api/interactions/toggle")
+async def toggle_interaction(interaction: InteractionToggle):
+    items_db = db.load_data("items.json", default_data=[])
+    
+    target_item = next((item for item in items_db if item["id"] == interaction.item_id), None)
+    if not target_item:
+        raise HTTPException(status_code=404, detail="目标工具/应用不存在")
+
+    # 确定要操作的列表字段，例如 "liked_by" 或 "favorited_by"
+    list_key = f"{interaction.action_type}d_by" # liked_by / favorited_by
+    count_key = "likes" if interaction.action_type == "like" else "favorites"
+    
+    if list_key not in target_item:
+        target_item[list_key] = []
+        target_item[count_key] = 0
+
+    user_list = target_item[list_key]
+    
+    # 执行 Toggle 逻辑
+    if interaction.is_active:
+        if interaction.user_id not in user_list:
+            user_list.append(interaction.user_id)
+            target_item[count_key] += 1
+    else:
+        if interaction.user_id in user_list:
+            user_list.remove(interaction.user_id)
+            target_item[count_key] = max(0, target_item[count_key] - 1)
+            
+    db.save_data("items.json", items_db)
+    
+    return {
+        "status": "success", 
+        "message": f"操作成功", 
+        "new_count": target_item[count_key]
+    }
+
+# --- 4. 评论系统 ---
+@app.post("/api/comments")
+async def post_comment(comment: CommentCreate):
+    comments_db = db.load_data("comments.json", default_data={}) # 结构: {"item_id": [comment_list]}
+    
+    item_comments = comments_db.get(comment.item_id, [])
+    
+    new_comment = {
+        "id": f"c_{int(time.time())}_{uuid.uuid4().hex[:6]}",
+        "author": comment.author,
+        "content": comment.content,
+        "replyToUser": comment.reply_to_user,
+        "isDeleted": False,
+        "replies": [],
+        "created_at": int(time.time())
+    }
+
+    # 如果传了 parent_id，说明是子回复，要把它塞进父评论的 replies 数组里
+    if comment.parent_id:
+        parent = next((c for c in item_comments if c["id"] == comment.parent_id), None)
+        if parent:
+            parent["replies"].append(new_comment)
+        else:
+            raise HTTPException(status_code=404, detail="找不到要回复的父级评论")
+    else:
+        # 否则作为顶级评论插入
+        item_comments.append(new_comment)
+        
+    comments_db[comment.item_id] = item_comments
+    db.save_data("comments.json", comments_db)
+    
+    return {"status": "success", "data": new_comment}
+
+@app.delete("/api/comments/{item_id}/{comment_id}")
+async def soft_delete_comment(item_id: str, comment_id: str, author: str):
+    comments_db = db.load_data("comments.json", default_data={})
+    item_comments = comments_db.get(item_id, [])
+    
+    # 递归查找要删除的评论
+    def find_and_delete(comments_list):
+        for c in comments_list:
+            if c["id"] == comment_id:
+                if c["author"] != author:
+                    raise HTTPException(status_code=403, detail="无权删除他人的评论")
+                c["isDeleted"] = True
+                c["content"] = "" # 清空敏感或不良内容
+                return True
+            # 检查子回复
+            if "replies" in c and find_and_delete(c["replies"]):
+                return True
+        return False
+
+    if find_and_delete(item_comments):
+        db.save_data("comments.json", comments_db)
+        return {"status": "success", "message": "评论已软删除"}
+    else:
+        raise HTTPException(status_code=404, detail="找不到该评论")

requirements.txt

@@ -0,0 +1,5 @@
+fastapi
+uvicorn
+pydantic
+huggingface_hub
+datasets

数据库连接.py

@@ -0,0 +1,57 @@
+# ⚙️ 后端逻辑/数据库连接.py
+import os
+import json
+from huggingface_hub import HfApi, hf_hub_download
+
+HF_TOKEN = os.environ.get("HF_TOKEN")
+# 已经替换为你真实的 Dataset 仓库
+DATASET_REPO_ID = "ZHIWEI666/ComfyUI-Ranking" 
+# HF Spaces 的 Docker 环境只有 /tmp 目录允许自由读写
+LOCAL_DB_DIR = "/tmp/local_db_data" 
+
+api = HfApi() if HF_TOKEN else None
+
+# 确保本地测试目录存在
+if not os.path.exists(LOCAL_DB_DIR):
+    os.makedirs(LOCAL_DB_DIR)
+
+def load_data(file_name: str, default_data=None):
+    """读取数据：优先从 HF Dataset 读取，如果没有 Token 则从本地读取"""
+    if default_data is None:
+        default_data = {} if file_name == "users.json" else []
+        
+    try:
+        if HF_TOKEN:
+            file_path = hf_hub_download(repo_id=DATASET_REPO_ID, filename=file_name, repo_type="dataset", token=HF_TOKEN)
+        else:
+            file_path = os.path.join(LOCAL_DB_DIR, file_name)
+            if not os.path.exists(file_path):
+                return default_data
+
+        with open(file_path, "r", encoding="utf-8") as f:
+            return json.load(f)
+    except Exception as e:
+        print(f"[{file_name}] 数据拉取失败或文件不存在，使用默认空数据。原因: {e}")
+        return default_data
+
+def save_data(file_name: str, data):
+    """保存数据：写入本地并同步到 HF Dataset"""
+    local_path = os.path.join(LOCAL_DB_DIR, file_name)
+    
+    # 1. 始终先写入本地临时目录
+    with open(local_path, "w", encoding="utf-8") as f:
+        json.dump(data, f, ensure_ascii=False, indent=2)
+    
+    # 2. 如果有 Token，同步推送到 Hugging Face
+    if HF_TOKEN:
+        try:
+            api.upload_file(
+                path_or_fileobj=local_path,
+                path_in_repo=file_name,
+                repo_id=DATASET_REPO_ID,
+                repo_type="dataset",
+                token=HF_TOKEN,
+                commit_message=f"Auto-update {file_name}"
+            )
+        except Exception as e:
+            print(f"同步到 HF Dataset 失败: {e}")