Преглед изворни кода

fix progres bar and limit histry query

ming пре 3 година
родитељ
комит
92da35241a
1 измењених фајлова са 3 додато и 2 уклоњено
  1. 3 2
      main.py

+ 3 - 2
main.py

@@ -83,7 +83,7 @@ class ConnectionManager:
             await connection.send_text(message)
             await connection.send_text(message)
 
 
 
 
-manager = ConnectionManager()
+
 
 
 @app.get("/")
 @app.get("/")
 async def root():
 async def root():
@@ -181,6 +181,7 @@ async def make_anchor_video_v33(req:request):
 
 
 @app.websocket("/progress")
 @app.websocket("/progress")
 async def websocket_endpoint(websocket: WebSocket):
 async def websocket_endpoint(websocket: WebSocket):
+    manager = ConnectionManager()
     await manager.connect(websocket)
     await manager.connect(websocket)
     try:
     try:
         while True:
         while True:
@@ -194,7 +195,7 @@ async def history_input():
     db = dataset.connect('mysql://choozmo:pAssw0rd@db.ptt.cx:3306/AI_anchor?charset=utf8mb4')
     db = dataset.connect('mysql://choozmo:pAssw0rd@db.ptt.cx:3306/AI_anchor?charset=utf8mb4')
     statement = 'SELECT * FROM history_input ORDER BY timestamp DESC'
     statement = 'SELECT * FROM history_input ORDER BY timestamp DESC'
     logs = []
     logs = []
-    for row in db.query(statement):
+    for row in db.query(statement)[:50]:
         logs.append({'id':row['id'],'name':row['name'],'text_content':row['text_content'].split(','),'image_urls':row['image_urls'].split(',')})
         logs.append({'id':row['id'],'name':row['name'],'text_content':row['text_content'].split(','),'image_urls':row['image_urls'].split(',')})
     return logs
     return logs