Update app.py

app.py

@@ -23,22 +23,42 @@ LORA_STYLES = [
 MAX_SEED = 2**31 - 1
 
 
-def upload_to_imgbb(image_path):
-    """上传图片到免费图床，返回公开 URL"""
-    IMGBB_API_KEY = os.environ.get("IMGBB_API_KEY")
-    if not IMGBB_API_KEY:
-        return None
-    with open(image_path, "rb") as f:
-        import base64
-        b64 = base64.b64encode(f.read()).decode()
-    resp = requests.post(
-        "https://api.imgbb.com/1/upload",
-        data={"key": IMGBB_API_KEY, "image": b64}
+# def upload_to_imgbb(image_path):
+#     """上传图片到免费图床，返回公开 URL"""
+#     IMGBB_API_KEY = os.environ.get("IMGBB_API_KEY")
+#     if not IMGBB_API_KEY:
+#         return None
+#     with open(image_path, "rb") as f:
+#         import base64
+#         b64 = base64.b64encode(f.read()).decode()
+#     resp = requests.post(
+#         "https://api.imgbb.com/1/upload",
+#         data={"key": IMGBB_API_KEY, "image": b64}
+#     )
+#     if resp.status_code == 200:
+#         print(f'图床resp.json()={resp.json()}')
+#         return resp.json()["data"]["url"]
+#     return None
+
+def upload_to_hf(image_path):
+    """上传到 HF dataset，返回公开直链"""
+    from huggingface_hub import HfApi
+    import uuid
+    
+    api = HfApi(token=HF_TOKEN)
+    filename = f"{uuid.uuid4().hex[:8]}_{os.path.basename(image_path)}"
+    
+    # 上传到你自己的 HF dataset repo（需要先创建一个 public dataset repo）
+    url = api.upload_file(
+        path_or_fileobj=image_path,
+        path_in_repo=f"uploads/{filename}",
+        repo_id="kenfoo",  # 改成你自己的
+        repo_type="dataset",
     )
-    if resp.status_code == 200:
-        print(f'图床resp.json()={resp.json()}')
-        return resp.json()["data"]["url"]
-    return None
+    print(f'图床 ={url}')
+    # 转成直链格式
+    # url 格式: https://huggingface.co/datasets/xxx/yyy/resolve/main/uploads/zzz.jpg
+    return url
 
 
 def infer(
@@ -59,9 +79,12 @@ def infer(
         seed = random.randint(0, MAX_SEED)
 
     # 方案：上传到公开图床，拿到 URL 再传给远端
-    public_url = upload_to_imgbb(image)
+    #public_url = upload_to_imgbb(image)
+
+    public_url = upload_to_hf(image)
+
     if not public_url:
-        print("图床上传失败，请检查 IMGBB_API_KEY")
+        print("图床上传失败 ")
         return None, seed
 
     print(f"图片公开 URL: {public_url}")
@@ -69,7 +92,7 @@ def infer(
     # 直接传 URL 字符串，让远端自己下载
     # 直接传字符串列表
     images_input = [public_url]
-    
+
     print(f"prompt: {prompt}, lora: {lora_adapter}, seed: {seed}")
 
     try: