Upload app.py

app.py

@@ -1,4 +1,4 @@
-# ⚙️ 后端逻辑/核心服务端.py
+# ⚙️ 后端逻辑/核心服务端.py (Hugging Face Spaces app.py)
 from fastapi import FastAPI, HTTPException, File, UploadFile, Form
 from fastapi.middleware.cors import CORSMiddleware
 from pydantic import BaseModel
@@ -7,6 +7,7 @@ import time
 import uuid
 import hashlib
 
+# 导入数据库操作模块
 import 数据库连接 as db
 
 app = FastAPI(title="ComfyUI Ranking Community API")
@@ -37,7 +38,10 @@ async def upload_file(file: UploadFile = File(...), file_type: str = Form(...)):
     db.save_file(full_path_in_repo, content)
     url = f"https://huggingface.co/datasets/{db.DATASET_REPO_ID}/resolve/main/{full_path_in_repo}"
     
-    return {"status": "success", "url": url, "display_name": original_name, "hashed_name": new_filename}
+    return {
+        "status": "success", "url": url, 
+        "display_name": original_name, "hashed_name": new_filename
+    }
 
 # --- 数据模型 ---
 class UserRegister(BaseModel):
@@ -135,9 +139,17 @@ def add_notification(target_account: str, notif_data: dict):
 @app.post("/api/users/register")
 async def register_user(user: UserRegister):
     users_db = db.load_data("users.json", default_data={})
-    if user.account in users_db: raise HTTPException(status_code=400, detail="该账号已被注册")
+    if user.account in users_db: raise HTTPException(status_code=400, detail="该账号已被注册，请尝试其他账号名")
+    for existing_user in users_db.values():
+        if existing_user.get("email") == user.email: raise HTTPException(status_code=400, detail="该邮箱已被绑定")
+        if existing_user.get("phone") == user.phone: raise HTTPException(status_code=400, detail="该手机号已被绑定")
+    
     new_user = user.dict()
-    new_user.update({"created_at": int(time.time()), "followers": [], "following": [], "privacy": {"follows": False, "likes": False, "favorites": False, "downloads": False}})
+    new_user.update({
+        "receivedLikes": 0, "receivedFollows": 0, "created_at": int(time.time()), "followers": [], "following": [],
+        "privacy": {"follows": False, "likes": False, "favorites": False, "downloads": False}
+    })
+    
     users_db[user.account] = new_user
     db.save_data("users.json", users_db)
     return {"status": "success", "message": "注册成功", "data": {k: v for k, v in new_user.items() if k != "password"}}
@@ -155,6 +167,7 @@ async def get_user_profile(account: str):
     users_db = db.load_data("users.json", default_data={})
     if account not in users_db: raise HTTPException(status_code=404, detail="用户不存在")
     user_data = users_db[account]
+    
     items_db = db.load_data("items.json", default_data=[])
     user_items = [item for item in items_db if item.get("author") == account]
     user_data["receivedLikes"] = sum(item.get("likes", 0) for item in user_items)
@@ -193,7 +206,6 @@ async def toggle_follow(follow: FollowToggle):
     if follow.is_active:
         if follow.user_id not in target_followers: 
             target_followers.append(follow.user_id)
-            # 派发关注通知
             add_notification(follow.target_account, {"type": "follow", "from_user": follow.user_id})
         if follow.target_account not in current_following: current_following.append(follow.target_account)
     else:
@@ -202,14 +214,14 @@ async def toggle_follow(follow: FollowToggle):
     db.save_data("users.json", users_db)
     return {"status": "success"}
 
-# --- 2. 排行榜与发布 ---
+# --- 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=[])
     comments_db = db.load_data("comments.json", default_data={})
+    
     filtered_items = [item for item in items_db if item.get("type") == type]
     for item in filtered_items:
-        # 动态挂载评论区数据
         item["commentsData"] = comments_db.get(item["id"], [])
         item["comments"] = len(item["commentsData"])
         
@@ -239,7 +251,7 @@ async def get_creators(sort: str = "downloads", limit: int = 20):
             "shortDesc": u.get("intro", "这个人很懒，什么都没写..."), "fullDesc": u.get("intro", "这个人很懒，什么都没写..."),
             "likes": likes, "favorites": favorites, "downloads": uses, "toolsCount": tools_count, "appsCount": apps_count,
             "followers": len(u.get("followers", [])), "created_at": u.get("created_at", 0),
-            "commentsData": comments_db.get(account, []), # 挂载留言板数据
+            "commentsData": comments_db.get(account, []), 
             "trendData": {"tools": [0,0,0,0,0, tools_count], "apps": [0,0,0,0,0, apps_count]}
         })
         
@@ -279,18 +291,19 @@ async def toggle_interaction(interaction: InteractionToggle):
         if interaction.user_id not in user_list: 
             user_list.append(interaction.user_id)
             target_item[count_key] += 1
-            # 派发点赞/收藏通知给作者
             add_notification(target_item.get("author", ""), {
                 "type": interaction.action_type, "from_user": interaction.user_id,
                 "target_item_id": target_item["id"], "target_item_title": target_item["title"]
             })
     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)
+        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", "new_count": target_item[count_key]}
 
-# --- 4. 评论系统与消息 ---
+# --- 4. 评论系统与通知 ---
 @app.post("/api/comments")
 async def post_comment(comment: CommentCreate):
     comments_db = db.load_data("comments.json", default_data={}) 
@@ -311,13 +324,11 @@ async def post_comment(comment: CommentCreate):
         parent = next((c for c in item_comments if c["id"] == comment.parent_id), None)
         if parent: 
             parent["replies"].append(new_comment)
-            # 通知被回复者
             if comment.reply_to_user:
                 add_notification(comment.reply_to_user, {"type": "reply", "from_user": comment.author, "target_item_id": comment.item_id, "target_item_title": "收到新的回复", "content": comment.content})
-        else: raise HTTPException(status_code=404, detail="找不到要回复的评论")
+        else: raise HTTPException(status_code=404, detail="找不到要回复的父级评论")
     else: 
         item_comments.append(new_comment)
-        # 通知作品作者 或 留言板主人
         items_db = db.load_data("items.json", default_data=[])
         target_item = next((item for item in items_db if item["id"] == comment.item_id), None)
         if target_item: