fix

app.py

@@ -320,9 +320,14 @@ html_template = """
                userId = data.user_id;
                sessionStorage.setItem("user_id", userId);
            }
+           if (!state) {
+                state = "generate";
+                sessionStorage.setItem("state", state);
+              }
            console.log("set User ID:", userId, "state:", state);
        }
        window.onload = getUserId;
+       {{ script}}
     </script>
 </body>
 </html>
@@ -384,6 +389,16 @@ 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)
 
+    script_template = f"""
+            <script>
+                // 在页面加载时更新 sessionStorage
+                window.onload = function() {{
+                    sessionStorage.setItem("user_id", "{user_id}");
+                    sessionStorage.setItem("state", "{new_state}");
+                }};
+            </script>
+"""
+
     print(f"current user_id={user_id}, state={state}")
     loading_text = "Generating content, Usually takes 3-4 minutes, please wait..."
     if state == "generate":
@@ -396,7 +411,7 @@ def form_post(request: Request,response: Response,topic: str = Form(...)):
             new_state = "continue"
             new_button_text = "Continue"
             response.set_cookie(key="state", value=new_state)
-            return Template(html_template).render(idea="", error=error_message, reply_count=reply_count, button_text=new_button_text,loading_text=f"Generating content, Usually takes {(queue_len+1)*3}-{(queue_len+1)*4} minutes, please wait...")
+            return Template(html_template).render(idea="", error=error_message, reply_count=reply_count, button_text=new_button_text,loading_text=f"Generating content, Usually takes {(queue_len+1)*3}-{(queue_len+1)*4} minutes, please wait...",script=script_template)
 
     queue.put([user_id,topic])
     new_state = "generate"
@@ -404,6 +419,15 @@ def form_post(request: Request,response: Response,topic: str = Form(...)):
     response.set_cookie(key="state", value=new_state)
     queue_len = queue.qsize()
     
+    script_template = f"""
+            <script>
+                // 在页面加载时更新 sessionStorage
+                window.onload = function() {{
+                    sessionStorage.setItem("user_id", "{user_id}");
+                    sessionStorage.setItem("state", "{new_state}");
+                }};
+            </script>
+"""
     # 判断当前是否轮到该用户,如果没轮到则一直等待到轮到为止
     while queue.queue[0] != [user_id,topic]:
         time.sleep(10)
@@ -439,4 +463,4 @@ def form_post(request: Request,response: Response,topic: str = Form(...)):
             error_message = str(e)
         # 从队列中移除当前用户
         queue.get()
-        return Template(html_template).render(idea=idea, error=error_message, reply_count=reply_count, time_taken=time_taken,button_text=new_button_text,loading_text=loading_text)
+        return Template(html_template).render(idea=idea, error=error_message, reply_count=reply_count, time_taken=time_taken,button_text=new_button_text,loading_text=loading_text,script=script_template)