fix

app.py

@@ -312,13 +312,14 @@ html_template = """
     <script>
        async function getUserId() {
            let userId = sessionStorage.getItem("user_id");
+           console.log("original User ID:", userId);
            if (!userId) {
                const response = await fetch("/user_id");
                const data = await response.json();
                userId = data.user_id;
                sessionStorage.setItem("user_id", userId);
            }
-           console.log("User ID:", userId);
+           console.log("set User ID:", userId);
        }
        window.onload = getUserId;
     </script>
@@ -356,28 +357,13 @@ def fix_markdown(text):
     # Join the result list into a single string with newline characters
     return '<br>'.join(result)
 
-@app.middleware("http")
-async def add_user_id_and_state_cookie(request: Request, call_next):
-    user_id = request.cookies.get("user_id")
-    state = request.cookies.get("state")
-    print(f"Before call_next: user_id={user_id}, state={state}")
-    response = await call_next(request)
-    if not user_id:
-        user_id = str(uuid.uuid4())
-        state = "generate"
-        response.set_cookie(key="user_id", value=user_id)
-        response.set_cookie(key="state", value=state)
-    elif not state:
-        state = "generate"
-        response.set_cookie(key="state", value=state)
-    print(f"After call_next: user_id={user_id}, state={state}")
-    return response
 
 @app.get("/user_id")
 async def get_user_id(request: Request):
     user_id = request.cookies.get("user_id")
     if not user_id:
         user_id = str(uuid.uuid4())
+    print("successfully set user_id = ", user_id)
     return {"user_id": user_id}
 
 
@@ -397,7 +383,7 @@ def form_post(request: Request,response: Response,topic: str = Form(...)):
     response.set_cookie(key="user_id", value=user_id)
     response.set_cookie(key="state", value=state)
 
-    print(user_id,state)
+    print(f"current user_id={user_id}, state={state}")
     loading_text = "Generating content, Usually takes 3-4 minutes, please wait..."
     if state == "generate":
         if not queue.empty():