feat: 下载文件

main.py

@@ -1,4 +1,5 @@
 from fastapi import FastAPI, File, UploadFile, HTTPException
+from fastapi.responses import StreamingResponse
 import requests
 import os
 import json
@@ -25,6 +26,11 @@ if not os.path.exists(UPLOAD_DIRECTORY):
     os.makedirs(UPLOAD_DIRECTORY)
 
 
+def download_file(url: str):
+    with requests.get(url, stream=True) as response:
+        for chunk in response.iter_content(chunk_size=1024):  # 每次读取 1KB
+            if chunk:
+                yield chunk
 # HF_API_TOKEN = "your_huggingface_api_token"  # 替换为您的 Hugging Face API Token
 # HfFolder.save_token(HF_API_TOKEN)
 
@@ -33,7 +39,7 @@ def read_root():
     return {"Hello": "World!"}
 
 @app.post("/upload")
-async def upload_file(file: UploadFile = File(...)):
+async def upload_file(file: UploadFile = File()):
     logger.info(f'收到请求 {file.filename}')
     # 定义文件存储路径和名称
     file_location = os.path.join(UPLOAD_DIRECTORY, file.filename)
@@ -43,4 +49,9 @@ async def upload_file(file: UploadFile = File(...)):
         logger.info(f'写入本地文件 {file.filename}')
         buffer.write(await file.read())
 
-    return {"info": f"file '{file.filename}' saved at '{file_location}'"}
+    return {"info": f"file '{file.filename}' saved at '{file_location}'"}
+
+
+@app.get("/download")
+def download():
+    return StreamingResponse(download_file("https://huggingface.co/wsj1995/stable-diffusion-models/resolve/main/3Guofeng3_v34.safetensors?download=true"), media_type="application/octet-stream")