From 7c6eed2022d8763a7df6ad6e158da6a138790d27 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E6=A2=A6=E5=87=8C=E6=B1=90?= <mew@meowdream.cn>
Date: Thu, 5 Jun 2025 02:12:26 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E5=A4=8D=E4=B8=8A=E4=BC=A0=E5=9B=BE?=
 =?UTF-8?q?=E7=89=87=E6=A0=A1=E9=AA=8C=E5=A4=B1=E8=B4=A5=E7=9A=84=E9=97=AE?=
 =?UTF-8?q?=E9=A2=98?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 README.md                                 |   2 +-
 routes/__pycache__/images.cpython-311.pyc | Bin 15326 -> 15314 bytes
 routes/images.py                          |  42 +++++++++++-----------
 3 files changed, 22 insertions(+), 22 deletions(-)

diff --git a/README.md b/README.md
index 2e8a6f3..ee2711d 100644
--- a/README.md
+++ b/README.md
@@ -26,7 +26,7 @@ CatismImage 是一个基于Web的图片存储和管理系统，提供用户友
 4. [可选] 配置NSFWPY
 
    如果您希望启用NSFW过滤功能，您需要配置NSFWPY。
-   
+
    请参考[HG-ha/nsfwpy](https://github.com/HG-ha/nsfwpy)，通过您喜欢的方式完成配置。
 
 5. 配置环境变量：
diff --git a/routes/__pycache__/images.cpython-311.pyc b/routes/__pycache__/images.cpython-311.pyc
index b40318e405e8bfbfd95173240346e1ad439a7a42..ba82300705b2c03c56866c6ab241405bd296569f 100644
GIT binary patch
delta 476
zcmcateyN;qIWI340}v$lIAq+@-pJR>Hu)rn$L85=dQ6imdFnRzbH3(gwA@@Rs>#U8
z$Lhd%V{)ye)a1ir-P|9T8F(dsd}U+c5uMyGVO9T?gMmZv3cK=01|e22#=9K67x<Jf
za;RM4P`SXNa)X1XpR<c|f?_964^Icr2PS?_FUA|<^3!W4)!r4BxgxA{LBJe>Zive*
zV40AALsIdAvgrj0vk&YH5~?2=SS0wqfQSx{4_pkqqStv;FY%~e<WaxEqke%${f0z{
z!hGIYyfgXd@OOBAU}F%F?62*roe{Z$@gl$06@IN1k?Uhu#qO{?ka$to<BG1wMShPH
z#+Ug0F7W%^1zN~{LDT1gyzfPEzboQ?7XpGWhzDN~2!1e0T*B9j@dE=R(6tcquCUY$
z_Y0~{dzE)2?o`{O24sSewDUz_mn*_92a+!cdtTr{M?Zkz12fQ+%||8PGRGbO2KW&c
zJ~wWLBf^ftuIda&)fj=Cqw2D5?99hlK<r~|j372UGl<PC>!!_oOp6gn9#?P@@#1DU
hX(|dNPjPd3GclhMRr0cBK4s1bV%xHAR+QZ<2LS)PoH+mh

delta 520
zcmcaqey^NwIWI340}$M7cgV2P*~r(+w)rue9@FI6Y#x&<dFnRzbH3(gG~Qe-s>#Tx
zJb9_u%KEPy3_PM&xYa*02yyx`-sKUzAf$egN8<{Q#swaY8$1I2o?V_35<7i*d^&tS
zF!A&HG2W0?o^LzL_O7_X6>)<LBGwReLt1Hp%MAS+vZ@!<EicGeePCyh(fr83A|vz#
zL`>lLz{Maad0jyBl7Qw#0qrXS+7|@0Z^+cE%=ew;JJWxT{{+qtYz!i@(`_f&E|6T|
zcu`pIim=`W$?dY6WOr1aP`qg9bH&i-qOi}2;!DCo7lecE0xfjEpc`;OIq;%%&=u*R
z3nAebq{A<Wgg+1vna(+hbB5wXo+&&Nfbhp%VVMsMj6k;|lXu1C7O*>C&~)3Id_Zw$
z+MYBZ6NKd5FN%9y5%)Nud_mmzf&e=D0R$hIL56`?Kb~@nOpX+{+I(K(C3F2HV9+07
z;qy{rI3nyQ?5fUiRE-hHIjSz}#?E|<1;jqa#t34wn}FChP9mP_3?~IeK_pn|Np&tS
zIp&jwN?wx8r-T`S>{B8vK&7W7SwU<$W)NFR)|-j>G$SLBJZs^k5h2ZRK~XDEfnl?%
H>{>Yhj3BPt

diff --git a/routes/images.py b/routes/images.py
index f6dd398..19e09d5 100644
--- a/routes/images.py
+++ b/routes/images.py
@@ -60,30 +60,30 @@ def upload_image():
             file.seek(0)
         except Exception:
             return jsonify({'error': '无效的图片文件'}), 400
-            
-        # 如果启用了NSFW检查
-        if AppConfig.ENABLE_NSFW_FILTER:
-            print("NSFW检查已启用")
-            file.seek(0)
-            try:
-                response = requests.post(
-                    f"{AppConfig.NSFWPY_ENDPOINT}/classify",
-                    files={'image': file},
-                    headers={'accept': 'application/json'}
-                )
-                response.raise_for_status()
-                nsfw_data = response.json()
-                
-                # 检查NSFW评分
-                if nsfw_data.get('neutral', 0) + nsfw_data.get('drawing', 0) < 0.5:
-                    return jsonify({'error': '图片内容不符合安全标准'}), 403
-                
-                file.seek(0)
-            except Exception as e:
-                return jsonify({'error': f'NSFW检查失败: {str(e)}'}), 500
     except Exception:
         return jsonify({'error': '无效的图片文件'}), 400
         
+    # 如果启用了NSFW检查
+    if AppConfig.ENABLE_NSFW_FILTER:
+        print("NSFW检查已启用")
+        file.seek(0)
+        try:
+            response = requests.post(
+                f"{AppConfig.NSFWPY_ENDPOINT}/classify",
+                files={'image': file},
+                headers={'accept': 'application/json'}
+            )
+            response.raise_for_status()
+            nsfw_data = response.json()
+            
+            # 检查NSFW评分
+            if nsfw_data.get('neutral', 0) + nsfw_data.get('drawing', 0) < 0.5:
+                return jsonify({'error': '图片内容不符合安全标准'}), 403
+            
+            file.seek(0)
+        except Exception as e:
+            return jsonify({'error': f'NSFW检查失败: {str(e)}'}), 500
+
     # 获取文件大小并检查存储空间
     file.seek(0, 2)  # 移动到文件末尾
     file_size = file.tell()  # 获取文件大小(字节)