Upload app.py

app.py

@@ -7,12 +7,44 @@ import datetime
 # 流式端点
 API_URL = "https://ai.fakeopen.com/v1/chat/completions"  # 用户需要提供自己的 OPENAI_API_KEY
 
+# 时间转换
+def timeswitch(t,sourcetz = "utc",totype="str",targettz = "cn",strformat = "%Y-%m-%d %H:%M:%S"):
+	'''
+	t: int/str 需要转换的时间表示，秒级时间戳或格式化的字符串
+	sourcetz: t所关联的时区别名，在timezone字典里自定义别名，如“cn”表示东八区,"utc"为世界标准时间
+	targettz: 输出时间的时区别名
+	totype: "str" or "timestamp" 输出时间表示类型
+	strformat：输入或输出的字符串格式，如果t为字符串，则strformat需要和它对应。
+	'''
+    # 常用的两个时区定义在字典里，可以添加其他时区
+    timezone = {
+        "cn":datetime.timezone(datetime.timedelta(hours=8)),
+        "utc":datetime.timezone.utc
+    }
+    sourcetz = timezone.get(sourcetz) if sourcetz else None
+    targettz = timezone.get(targettz) if targettz else sourcetz
+
+    # transfer to datetime
+    if isinstance(t,str):
+        ctime = datetime.datetime.strptime(t,strformat).replace(tzinfo = sourcetz)
+    elif isinstance(t,int) or isinstance(t,float):
+        if t > 1e10: t=t/1000
+        ctime = datetime.datetime.fromtimestamp(t).replace(tzinfo = sourcetz)
+
+    # from datetime to target type
+    if totype == "str":
+        newtime = ctime.astimezone(targettz).strftime(strformat)
+    elif totype == 'timestamp':
+        newtime = ctime.astimezone(targettz).timestamp()
+        newtime = int(newtime)
+    return newtime
+
 # 推断函数
 def predict(openai_gptapi_key, model, system_msg, inputs, top_p, temperature, max_tokens, presence_penalty, frequency_penalty, chat_counter, chatbot=[], history=[]):  
 
     print(f"——————————————————————————————")
     # 获取当前时间
-    current_time = datetime.datetime.now().strftime("%Y年 - %m月 - %d日 %H时 : %M分 : %S秒")
+    current_time = timeswitch(datetime.datetime.now().timestamp(),sourcetz="utc",targettz="cn",totype='str',strformat="%Y年-%m月-%d日 %H时:%M分:%S秒")
     if inputs.strip() == '':
         inputs = "你好呀，使用英语与中文简单介绍下你自己吧！"
     if openai_gptapi_key.strip() == '':
@@ -101,13 +133,13 @@ def predict(openai_gptapi_key, model, system_msg, inputs, top_p, temperature, ma
           # 将每行解码为响应数据，因为响应数据是以字节形式返回的
           if len(chunk) > 12 and "content" in json.loads(chunk[6:])['choices'][0]['delta']:
               partial_words = partial_words + json.loads(chunk[6:])['choices'][0]["delta"]["content"]
-              print(f"[{current_time}] 聊天：模型回复 - {partial_words}")
               if token_counter == 0:
                 history.append(" " + partial_words)
               else:
                 history[-1] = partial_words
               chat = [(history[i], history[i + 1]) for i in range(0, len(history) - 1, 2) ]  # 转换为列表的元组
               token_counter+=1
+              print(f"[{current_time}] 聊天：模型回复 - {partial_words}")
               yield chat, history, chat_counter, response  # 类似于 {chatbot: chat, state: history}  
                    
 # 重置文本框