完善服务端 客户端完成

.gitignore

@@ -0,0 +1,2 @@
+
+config.json

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()
+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")

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__':