zishuo/gpt4free
1
0
Fork 0
mirror of https://github.com/xtekky/gpt4free.git synced 2025-10-05 08:16:58 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add ip ban

Browse Source
This commit is contained in:
hlohaus
2025-06-24 22:25:42 +02:00
parent 3ce8edce27
commit 744dfeb957
4 changed files with 28 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
g4f/gui/server/backend_api.py
View File

@@ -135,9 +135,14 @@ class Backend_Api(Api):
else:
json_data["provider"] = models.HuggingFace
if app.demo:
ip = request.headers.get("X-Forwarded-For", "")
ip_bans = Path(get_cookies_dir()) / ".ip_bans"
if ip_bans.exists():
ip_bans = ip_bans.read_text().splitlines()
if (ip and ip in ip_bans):
return "You are banned from using this service.", 403
user = request.headers.get("Cf-Ipcountry", "")
ip = request.headers.get("X-Forwarded-For", "").split(":")[-1]
json_data["user"] = request.headers.get("x_user", f"{user}:{ip}")
json_data["user"] = request.headers.get("x_user", f"{user}:{ip.split(':')[-1]}")
json_data["referer"] = request.headers.get("referer", "")
json_data["user-agent"] = request.headers.get("user-agent", "")
if not json_data.get("referer") or "python" in json_data.get("user-agent", "").lower():