加入检测IP区域的功能

ChuanhuChatbot.py

@@ -130,7 +130,7 @@ with gr.Blocks(
 
     with gr.Row():
         gr.HTML(title)
-        status_display = gr.Markdown("status: ready", elem_id="status_display")
+        status_display = gr.Markdown(get_geoip(), elem_id="status_display")
 
     with gr.Row(scale=1).style(equal_height=True):
         with gr.Column(scale=5):
@@ -244,7 +244,7 @@ with gr.Blocks(
                             value=1.0,
                             step=0.05,
                             interactive=True,
-                            label="Top-p (nucleus sampling)",
+                            label="Top-p",
                         )
                         temperature = gr.Slider(
                             minimum=-0,

chat_func.py

@@ -11,6 +11,8 @@ import urllib3
 from tqdm import tqdm
 import colorama
 from duckduckgo_search import ddg
+import asyncio
+import aiohttp
 
 from presets import *
 from llama_func import *

utils.py

@@ -7,6 +7,7 @@ import os
 import datetime
 import hashlib
 import csv
+import requests
 
 import gradio as gr
 from pypinyin import lazy_pinyin
@@ -282,3 +283,14 @@ def sha1sum(filename):
 def replace_today(prompt):
     today = datetime.datetime.today().strftime("%Y-%m-%d")
     return prompt.replace("{current_date}", today)
+
+def get_geoip():
+    response = requests.get('https://ipapi.co/json/', timeout=5)
+    data = response.json()
+    country = data['country_name']
+    if country == "China":
+        text = "**您的IP区域：中国。请立即检查代理设置，在不受支持的地区使用API可能导致账号被封禁。**"
+    else:
+        text = f"您的IP区域：{country}。"
+    logging.info(text)
+    return text