Upload app.py

app.py

@@ -1,10 +1,11 @@
 # ⚙️ 后端逻辑/核心服务端.py (Hugging Face Spaces app.py)
-from fastapi import FastAPI, HTTPException
+from fastapi import FastAPI, HTTPException, File, UploadFile, Form
 from fastapi.middleware.cors import CORSMiddleware
 from pydantic import BaseModel
 from typing import Optional
 import time
 import uuid
+import hashlib
 
 # 导入数据库操作模块
 import 数据库连接 as db
@@ -25,6 +26,40 @@ app.add_middleware(
 def read_root():
     return {"status": "ok", "message": "ComfyUI Ranking API is running!"}
 
+# --- 【新增】：通用文件上传与哈希分类接口 ---
+@app.post("/api/upload")
+async def upload_file(file: UploadFile = File(...), file_type: str = Form(...)):
+    content = await file.read()
+    
+    # 计算文件内容的 MD5 哈希值，截取前10位保证唯一性
+    file_hash = hashlib.md5(content).hexdigest()[:10]
+    original_name = file.filename
+    # 将哈希值拼接到文件名前，防止同名文件覆盖 (后台物理名称)
+    new_filename = f"{file_hash}_{original_name}"
+    
+    # 根据文件类型，物理隔离到不同的目录
+    dir_mapping = {
+        "avatar": "avatars",
+        "cover": "covers",
+        "tool": "tools",
+        "app": "apps"
+    }
+    target_dir = dir_mapping.get(file_type, "others")
+    full_path_in_repo = f"{target_dir}/{new_filename}"
+    
+    # 将真实文件存入数据库(Dataset)
+    db.save_file(full_path_in_repo, content)
+    
+    # 组装供前端直接下载或显示的原始直连 URL
+    url = f"https://huggingface.co/datasets/{db.DATASET_REPO_ID}/resolve/main/{full_path_in_repo}"
+    
+    return {
+        "status": "success",
+        "url": url,                           # 前端 src 或 href 使用的真实链接
+        "display_name": original_name,        # 供前端 UI 干净展示的文件名
+        "hashed_name": new_filename           # 实际带哈希的文件名
+    }
+
 # --- 数据模型 ---
 class UserRegister(BaseModel):
     account: str   # 唯一账号 (新增为主键)
@@ -51,17 +86,28 @@ class InteractionToggle(BaseModel):
 
 class CommentCreate(BaseModel):
     item_id: str
-    author: str # 存储用户的 account 或 name，建议后续前端统一使用 account 作为关联
+    author: str # 存储用户的 account 或 name
     content: str
     reply_to_user: Optional[str] = None
     parent_id: Optional[str] = None 
 
+# 【核心修改】：新增发布项目的数据模型
+class ItemCreate(BaseModel):
+    type: str
+    title: str
+    shortDesc: str
+    fullDesc: str
+    link: str
+    coverUrl: Optional[str] = None
+    author: str 
+    price: float = 0.0  # 标价/积分字段
+
 # --- 1. 用户系统 ---
 @app.post("/api/users/register")
 async def register_user(user: UserRegister):
     users_db = db.load_data("users.json", default_data={})
     
-    # 核心修改：通过 account 查重，而不是 name
+    # 通过 account 查重
     if user.account in users_db:
         raise HTTPException(status_code=400, detail="该账号已被注册，请尝试其他账号名")
     
@@ -134,6 +180,39 @@ async def get_items(type: str = "tool", sort: str = "time", limit: int = 20):
         
     return {"status": "success", "data": filtered_items[:limit]}
 
+# 【核心修改】：处理前端发布的新插件/应用请求
+@app.post("/api/items")
+async def create_item(item: ItemCreate):
+    items_db = db.load_data("items.json", default_data=[])
+    
+    # 生成独一无二的项目 ID (如: tool_169..._a1b2c3)
+    new_id = f"{item.type}_{int(time.time())}_{uuid.uuid4().hex[:6]}"
+    
+    new_item = {
+        "id": new_id,
+        "type": item.type,
+        "title": item.title,
+        "author": item.author,
+        "shortDesc": item.shortDesc,
+        "fullDesc": item.fullDesc,
+        "link": item.link,
+        "coverUrl": item.coverUrl,
+        "price": item.price,
+        "likes": 0,
+        "favorites": 0,
+        "comments": 0,
+        "uses": 0,
+        "created_at": int(time.time()),
+        "liked_by": [],
+        "favorited_by": []
+    }
+    
+    # 核心：将新项目插入到列表的【最前面】(第0位)，保证 "最新发布" 排在第一
+    items_db.insert(0, new_item)
+    db.save_data("items.json", items_db)
+    
+    return {"status": "success", "data": new_item}
+
 # --- 3. 社交与互动 (点赞/收藏防抖与排重) ---
 @app.post("/api/interactions/toggle")
 async def toggle_interaction(interaction: InteractionToggle):