Update app.py

app.py

@@ -11,6 +11,7 @@ from websocket import WebSocketConnectionClosedException
 import gradio as gr
 import requests
 import logging
+import re
 
 from fastchat.conversation import SeparatorStyle
 from fastchat.constants import (
@@ -50,6 +51,15 @@ no_change_btn = gr.Button.update()
 enable_btn = gr.Button.update(interactive=True)
 disable_btn = gr.Button.update(interactive=False)
 
+
+def get_internet_ip():
+    r = requests.get("http://txt.go.sohu.com/ip/soip")
+    ip = re.findall(r'\d+.\d+.\d+.\d+', r.text)
+    if ip is not None and len(ip) > 0:
+        return ip[0]
+    return None
+
+
 enable_moderation = True if os.environ.get('enable_moderation', default='False')=="True" else False
 concurrency_count = int(os.environ.get('concurrency_count', default='10'))
 model_list_mode = os.environ.get('model_list_mode', default='reload')
@@ -69,10 +79,6 @@ dataset_sample = {
         "train": ["abcdef"],
         "valid": ["zxcvbn"]
     },
-    "cat": {
-        "train": ["aaaaaa"],
-        "valid": ["bbbbbb"]
-    }
 }
 
 
@@ -99,6 +105,8 @@ ip_expiration_dict = defaultdict(lambda: 0)
 def is_legal_char(c):
     if c.isalnum():
         return True
+    if '\u4e00' <= c <= '\u9fff':
+        return True
     if c in "！？｡＂＃＄％＆＇（）＊＋，－／：；＜＝＞＠［＼］＾＿｀｛｜｝～｟｠｢｣､、〃》「」『』【】〔〕〖〗〘〙〚〛〜〝〞〟〰〾〿–—‘’‛“”„‟…‧﹏.":
         return True
     if c in '!"#$%&\'()*+,-./:;<=>?@[\\]^_`{|}~':
@@ -384,6 +392,9 @@ def bot_response(state, temperature, top_p, max_new_tokens, request: gr.Request)
     try:
         for data in stream_iter:
             if data["error_code"] == 0:
+                finish_reason = data.get("finish_reason", None)
+                if finish_reason is not None and finish_reason == "length":
+                    gr.Warning("Answer interrupted because the setting of [Max output tokens], try set a larger value.")
                 output = data["text"].strip()
                 if "vicuna" in model_name:
                     output = post_process_code(output)
@@ -760,7 +771,10 @@ def build_demo(models):
 
     return demo
 
-
+try:
+    print("Internet IP:", get_internet_ip())
+except Exception as e:
+    print(f"Get Internet IP error: {e}")
 models = get_model_list(midware_url)
 
 # Launch the demo