From 7a999502c58dd306abbbcd2a9f663faedd8ceb36 Mon Sep 17 00:00:00 2001
From: Kakune55 <xiao_doubi9637@outlook.com>
Date: Sat, 4 Nov 2023 10:20:17 +0800
Subject: [PATCH] =?UTF-8?q?=E5=AE=8C=E5=96=84=E6=9C=8D=E5=8A=A1=E7=AB=AF?=
 =?UTF-8?q?=20=E5=AE=A2=E6=88=B7=E7=AB=AF=E5=AE=8C=E6=88=90?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .gitignore |  2 ++
 client.py  | 34 ++++++++++++++++++++++++++++------
 server.py  | 20 +++++++++++---------
 3 files changed, 41 insertions(+), 15 deletions(-)
 create mode 100644 .gitignore

diff --git a/.gitignore b/.gitignore
new file mode 100644
index 0000000..af392e0
--- /dev/null
+++ b/.gitignore
@@ -0,0 +1,2 @@
+
+config.json
diff --git a/client.py b/client.py
index 7448168..c09f275 100644
--- a/client.py
+++ b/client.py
@@ -1,15 +1,37 @@
 import requests , json
 
-
-url = "http://localhost:5000/api/user"
-
 data = {
     "prompt":"你好",
     "userkey":"5b32f7z1"
 }
 
-req = json.loads(requests.post(url=url,json=data).text)
 
-print(req["output"])
 
-input()
\ No newline at end of file
+config = {
+    "url":"http://localhost:5000/api/user",
+    "userkey":"123456"
+}
+try:
+    f = open("config.json", "r")
+except IOError:
+    f = open("config.json", "w")
+    f.write(json.dumps(config))
+    f.close()
+    print("config.json配置文件创建完成 请修改配置文件为用户数据")
+    input()
+    exit()
+
+with open('config.json', 'r') as file:
+    config = json.loads(file.read())
+
+data["userkey"] = config["userkey"]
+
+while True:
+    data["prompt"] = str(input(">> You:\n"))
+    req = json.loads(requests.post(url=config['url'],json=data).text)
+    if req["code"] == 200:
+        print(f"\n>> AI:\n{req['output']}\n\n<剩余tokens:{req['surplus']} >")
+    elif req["code"] == 403:
+        print(f"\n>> System:\n{req['output']}\n\n<剩余tokens:{req['surplus']} >")
+    elif req["code"] == 400:
+        print(f"\n>> System:\n{req['output']}\n")
\ No newline at end of file
diff --git a/server.py b/server.py
index a0c2624..a12d251 100644
--- a/server.py
+++ b/server.py
@@ -21,13 +21,8 @@ def request_TY(userkey,prompt,history = ""):
     # 发送POST请求
     response = json.loads(requests.post(url, json=data ,headers=header).text)
     print(response)
-    
-    out = {
-        "code":200,
-        "output":f"{response['output']['text']}"
-    }
     reduce_value(userkey, response["usage"]["total_tokens"])
-    return out
+    return response['output']['text']
 
 def userSurplus(userkey):
     #打开数据库连接
@@ -90,12 +85,19 @@ app = flask.Flask(__name__)
 def post_data():
     userrequest = flask.request.json
     surplus = userSurplus(userrequest['userkey'])
+
+    if userrequest["prompt"] == "":
+        return {"code":400,"output":"Input is empty"}
+    
+    if userrequest["prompt"] == "":
+        return {"code":400,"output":"UserKey is empty"}
+
     if surplus > 0:
-        return request_TY(userrequest['userkey'],userrequest['prompt'])
+        return {"code":200,"output":f"{request_TY(userrequest['userkey'],userrequest['prompt'])}","surplus":userSurplus(userrequest['userkey'])}
     elif surplus == -99999:
-        return {"code":403,"output":"No User"}
+        return {"code":400,"output":"No User"}
     elif surplus <= 0:
-        return {"code":403,"output":"No prompt Tokens"}
+        return {"code":403,"output":"No prompt Tokens","surplus":surplus}
 
 
 if __name__ == '__main__':