fix

app.py

@@ -11,22 +11,17 @@ import threading
 import logging
 from queue import Queue
 import json
-import uuid
 import yaml
 import os
 
-# with open('/Users/jianghuyihei/code/MultiResearchAgent/config.yaml', 'r') as file:
-#     config = yaml.safe_load(file)
-# for key, value in config.items():
-#     os.environ[key] = str(value)
-
 lock = threading.Lock()
 app = FastAPI()
 
+current_user = None
 # 每日最大回复次数
 MAX_REPLIES_PER_DAY = 500
 # 当日回复次数计数器
-reply_count = 189
+reply_count = 0
 # 启动时设置计数器重置
 last_reset_time = datetime.now()
 
@@ -355,7 +350,7 @@ html_template = """
        {{ script}}
     </script>
     <script>
-    const socket = new WebSocket('wss://damo-nlp-sg-coi-agent.hf.space/ws');
+    const socket = new WebSocket("ws://localhost:7860/ws");
 
     socket.addEventListener('open', function (event) {
         const userId = document.getElementById("user_id").value;
@@ -420,6 +415,7 @@ queue = Queue()
 
 @app.websocket("/ws")
 async def websocket_endpoint(websocket: WebSocket):
+    global current_user
     await websocket.accept()
     print("WebSocket connection established.")
     try:
@@ -428,14 +424,15 @@ async def websocket_endpoint(websocket: WebSocket):
             message = json.loads(data)
             user_id = message.get("user_id")
             action = message.get("action")
-            print(action)
+            print(user_id, action)
             if action == "disconnect":
-                for item in list(queue.queue):
+                if user_id == current_user:
+                    continue
+                for item in queue.queue:
                     if item == user_id:
                         queue.queue.remove(item)
+                        print(f"User {user_id} disconnected.")
                         break
-                print(f"User {user_id} disconnected.")
-
     except WebSocketDisconnect:
         print("WebSocket connection closed.")
 
@@ -450,6 +447,7 @@ def form_get(request: Request):
 @app.post("/", response_class=HTMLResponse)
 def form_post(request: Request,topic: str = Form(...), user_id: str = Form(...), state: str = Form(...)):
     global reply_count
+    global current_user
     start_time = time.time()
     client_ip = request.client.host
     if user_id == "":
@@ -500,7 +498,7 @@ def form_post(request: Request,topic: str = Form(...), user_id: str = Form(...),
     print(queue.queue[0], [user_id,topic])
     while queue.queue[0] != user_id:
         # 检查用户是否还在队列中
-        if not any(user_id == item[0] for item in queue.queue):
+        if not any(user_id == item for item in queue.queue):
             return Template(html_template).render(
                 idea="", 
                 error="Request was cancelled.", 
@@ -514,6 +512,7 @@ def form_post(request: Request,topic: str = Form(...), user_id: str = Form(...),
 
     try:
         with lock:
+            current_user = user_id
             logging.info(f"Processing request for topic: {topic}")
             start_time = time.time()
             error_message = None