fix

app.py

@@ -312,17 +312,6 @@ html_template = """
         {% endif %}
     </div>
     <script>
-        function populateFormData() {
-            const userId = sessionStorage.getItem("user_id") || "";
-            const state = sessionStorage.getItem("state") || "";
-            
-            document.getElementById("user_id").value = userId;
-            document.getElementById("state").value = state;
-
-            console.log(`original userId: ${userId}, state: ${state}`);
-        }
-
-        window.onload = populateFormData;
        {{ script}}
     </script>
 </body>
@@ -362,7 +351,16 @@ def fix_markdown(text):
 
 @app.get("/", response_class=HTMLResponse)
 def form_get():
-    return Template(html_template).render(idea= "This is a example of the idea geneartion", error=None, reply_count=reply_count,button_text="Generate",loading_text="Generating content, Usually takes 3-4 minutes, please wait...")
+    script_template = """
+        window.onload = function() {
+            document.getElementById("user_id").value = "";
+            document.getElementById("state").value = "";
+            let userId = document.getElementById("user_id").value;
+            let state = document.getElementById("state").value;
+            console.log(`0 User ID: ${userId}, State: ${state}`);
+        }
+"""
+    return Template(html_template).render(idea= "This is a example of the idea geneartion", error=None, reply_count=reply_count,button_text="Generate",loading_text="Generating content, Usually takes 3-4 minutes, please wait...",script=script_template)
 
 @app.post("/", response_class=HTMLResponse)
 def form_post(request: Request,response: Response,topic: str = Form(...), user_id: str = Form(...), state: str = Form(...)):
@@ -377,23 +375,36 @@ def form_post(request: Request,response: Response,topic: str = Form(...), user_i
     script_template = f"""
         // 在页面加载时更新 sessionStorage
         window.onload = function() {{
-            sessionStorage.setItem("user_id", "{user_id}");
-            sessionStorage.setItem("state", "{state}");
+            document.getElementById("user_id").value = {user_id};
+            document.getElementById("state").value = {state};
+            let userId = document.getElementById("user_id").value;
+            let state = document.getElementById("state").value;
+            console.log(`1 User ID: ${{userId}}, State: ${{state}}`);
         }};
     """
 
-    print(f"current user_id={user_id}, state={state}")
+    print(f"current0 user_id={user_id}, state={state}")
     loading_text = "Generating content, Usually takes 3-4 minutes, please wait..."
     if state == "generate":
         if not queue.empty():
             queue_len = queue.qsize()
             if queue_len + reply_count >= MAX_REPLIES_PER_DAY:
                 error_message = "Today's maximum number of replies has been reached. Please try again tomorrow."
-                return Template(html_template).render(idea="", error=error_message, reply_count=reply_count, button_text="Generate",loading_text=loading_text)
+                return Template(html_template).render(idea="", error=error_message, reply_count=reply_count, button_text="Generate",loading_text=loading_text,script=script_template)
             error_message = "There are currently {} requests being processed. If you want to queue, please write your original topic and click the Continue button and you will enter the queue.".format(queue_len)
             new_state = "continue"
             new_button_text = "Continue"
-            response.set_cookie(key="state", value=new_state)
+            script_template = f"""
+        // 在页面加载时更新 sessionStorage
+        window.onload = function() {{
+            document.getElementById("user_id").value = {user_id};
+            document.getElementById("state").value = {new_state};
+            let userId = document.getElementById("user_id").value;
+            let state = document.getElementById("state").value;
+            console.log(`2 User ID: ${{userId}}, State: ${{state}}`);
+        }};
+    """
+            print(f"current1 user_id={user_id}, state={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...",script=script_template)
 
     queue.put([user_id,topic])
@@ -405,10 +416,13 @@ def form_post(request: Request,response: Response,topic: str = Form(...), user_i
     script_template = f"""
         // 在页面加载时更新 sessionStorage
         window.onload = function() {{
-            sessionStorage.setItem("user_id", "{user_id}");
-            sessionStorage.setItem("state", "{new_state}");
+            document.getElementById("user_id").value = {user_id};
+            document.getElementById("state").value = {new_state};
+            let userId = document.getElementById("user_id").value;
+            let state = document.getElementById("state").value;
+            console.log(`3 User ID: ${{userId}}, State: ${{state}}`);
         }};
-"""
+    """
     print(f"current2 user_id={user_id}, state={new_state}")
     # 判断当前是否轮到该用户,如果没轮到则一直等待到轮到为止
     while queue.queue[0] != [user_id,topic]: