test

app.py

@@ -10,19 +10,15 @@ from presets import *
 from overwrites import *
 from chat_func import *
 
-import requests
-from urllib import parse
-
-requestUrl = 'http://82.156.209.58:8520/api/user'
-headers = {
-  'Content-Type': 'application/x-www-form-urlencoded'
-}
-formData = {
-  
-}
-data = parse.urlencode(formData, True)
-response = requests.post(requestUrl, headers = headers, data = data)
-users=response.json()["data"]
+import sqlite3
+mydb=sqlite3.connect("user.db")
+## 读取sqlite3的数据
+cursor=mydb.cursor()
+
+#查询表名
+cursor.execute("SELECT username,password FROM user WHERE expire_time > date('now');")
+users=cursor.fetchall()
+
 
 logging.basicConfig(
     level=logging.DEBUG,
@@ -449,6 +445,12 @@ logging.info(
 # 默认开启本地服务器，默认可以直接从IP访问，默认不创建公开分享链接
 demo.title = "中建chatgpt"
 
+def authenticate(username, password):
+    # 验证用户名和密码是否匹配
+    for user in users:
+        if user[0] == username and user[1] == password:
+            return True
+    return False
 
 
 
@@ -457,12 +459,9 @@ if __name__ == "__main__":
     # if running in Docker
     if dockerflag:
         if authflag:
-            for user in users:
-                username=user["username"]
-                password=user["password"]
-                demo.queue().launch(
-                    server_name="0.0.0.0", server_port=7860, auth=(username, password),
-                    favicon_path="./assets/favicon.png"
+            demo.queue().launch(
+                server_name="0.0.0.0", server_port=7860, auth=authenticate,
+                favicon_path="./assets/favicon.png"
             )
         else:
             demo.queue().launch(server_name="0.0.0.0", server_port=7860, share=False, favicon_path="./assets/favicon.png")
@@ -470,9 +469,9 @@ if __name__ == "__main__":
     else:
         if authflag:
             for user in users:
-                username=user["username"]
-                password=user["password"]
-                demo.queue().launch(share=False, auth=(username, password), favicon_path="./assets/favicon.png", inbrowser=True)
+                username=user[0]
+                password=user[1]
+                demo.queue().launch(share=False, auth=authenticate, favicon_path="./assets/favicon.png", inbrowser=True)
         else:
             demo.queue(concurrency_count=1000).launch(share=False, favicon_path="./assets/favicon.ico", inbrowser=True)  # 改为 share=True 可以创建公开分享链接
         # demo.queue().launch(server_name="0.0.0.0", server_port=7860, share=False) # 可自定义端口

requirements.txt

@@ -10,3 +10,4 @@ Pygments
 llama_index
 langchain
 markdown
+requests

test.py

@@ -1,6 +1,9 @@
-import os
-os.path.exists("api_key.txt")
-with open("api_key.txt", "r") as f:
-            my_api_key = f.read().strip()
-my_api_key = os.environ.get("my_api_key")
-print(my_api_key)
+import sqlite3
+mydb=sqlite3.connect("user.db")
+## 读取sqlite3的数据
+cursor=mydb.cursor()
+
+#查询表名
+cursor.execute("SELECT username,password FROM user WHERE expire_time > date('now');")
+Tables=cursor.fetchall()
+print(Tables)