feat: 修改目录结构

.gitignore

@@ -0,0 +1 @@
+__pycache__

api.py

@@ -0,0 +1,62 @@
+from fastapi import FastAPI, File, UploadFile, HTTPException, Request, Depends, APIRouter
+from fastapi.responses import StreamingResponse
+import requests
+import os
+import json
+import base64
+from io import BytesIO
+from datetime import datetime
+import hashlib
+import time
+
+
+async def verify_internal_token(request: Request):
+    auth_key = request.query_params.get("auth_key")
+    if not auth_key:
+        raise HTTPException(status_code=403, detail="Forbidden")
+    timestamp = int(time.time())
+    timestampStr,rand,uid,hashStr = auth_key.split('-', 3)
+    path = request.url.path
+    if timestamp > int(timestampStr):
+        raise HTTPException(status_code=403, detail="Forbidden")
+    md5hash = hashlib.md5(f"{path}-{timestampStr}-{rand}-{uid}-{os.environ.get('AUTH_KEY')}".encode())
+    md5 = md5hash.hexdigest()
+    if md5 != hashStr:
+        raise HTTPException(status_code=403, detail="Forbidden")
+    
+
+router = APIRouter()
+
+UPLOAD_DIRECTORY = "/tmp/uploaded_files"
+
+# 如果目录不存在，则创建
+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
+
+
+@router.get("/")
+def read_root():
+    return {"Hello": "World!"}
+
+@router.post("/upload")
+async def upload_file(file: UploadFile = File(...)):
+    # 定义文件存储路径和名称
+    file_location = os.path.join(UPLOAD_DIRECTORY, file.filename)
+
+    # 将上传的文件写入本地文件系统
+    with open(file_location, "wb") as buffer:
+        buffer.write(await file.read())
+
+    return {"info": f"file '{file.filename}' saved at '{file_location}'"}
+
+
+@router.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")

main.py

@@ -1,65 +1,8 @@
-from fastapi import FastAPI, File, UploadFile, HTTPException, Request, Depends
-from fastapi.responses import StreamingResponse
-import requests
-import os
-import json
-import base64
-from io import BytesIO
-from datetime import datetime
-import hashlib
-import time
+from fastapi import FastAPI
+import api
 
 
-async def verify_internal_token(request: Request):
-    auth_key = request.query_params.get("auth_key")
-    if not auth_key:
-        raise HTTPException(status_code=403, detail="Forbidden")
-    timestamp = int(time.time())
-    timestampStr,rand,uid,hashStr = auth_key.split('-', 3)
-    path = request.url.path
-    if timestamp > int(timestampStr):
-        raise HTTPException(status_code=403, detail="Forbidden")
-    md5hash = hashlib.md5(f"{path}-{timestampStr}-{rand}-{uid}-{os.environ.get('AUTH_KEY')}".encode())
-    md5 = md5hash.hexdigest()
-    if md5 != hashStr:
-        raise HTTPException(status_code=403, detail="Forbidden")
-    
+app = FastAPI(dependencies=[Depends(api.verify_internal_token)])
 
+app.include_router(api.router)
 
-app = FastAPI(dependencies=[Depends(verify_internal_token)])
-
-
-UPLOAD_DIRECTORY = "/tmp/uploaded_files"
-
-# 如果目录不存在，则创建
-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)
-
-@app.get("/")
-def read_root():
-    return {"Hello": "World!"}
-
-@app.post("/upload")
-async def upload_file(file: UploadFile = File(...)):
-    # 定义文件存储路径和名称
-    file_location = os.path.join(UPLOAD_DIRECTORY, file.filename)
-
-    # 将上传的文件写入本地文件系统
-    with open(file_location, "wb") as buffer:
-        buffer.write(await file.read())
-
-    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")