fix

app.py

@@ -349,18 +349,21 @@ def fix_markdown(text):
     return '<br>'.join(result)
 
 
+script_template = """
+    function setstate() {
+        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}}`);
+    }
+    window.onload = setstate();
+"""
+
 @app.get("/", response_class=HTMLResponse)
 def form_get():
-    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)
+    script = script_template.format(user_id="''", 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)
 
 @app.post("/", response_class=HTMLResponse)
 def form_post(request: Request,response: Response,topic: str = Form(...), user_id: str = Form(...), state: str = Form(...)):
@@ -372,16 +375,7 @@ def form_post(request: Request,response: Response,topic: str = Form(...), user_i
     if state == "":
         state = "generate"
 
-    script_template = f"""
-        // 在页面加载时更新 sessionStorage
-        window.onload = function() {{
-            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}}`);
-        }};
-    """
+    script = script_template.format(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..."
@@ -390,39 +384,20 @@ def form_post(request: Request,response: Response,topic: str = Form(...), user_i
             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,script=script_template)
+                return Template(html_template).render(idea="", error=error_message, reply_count=reply_count, button_text="Generate",loading_text=loading_text,script=script)
             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"
-            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}}`);
-        }};
-    """
+            script = script_template.format(user_id=user_id, state=new_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)
+            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)
 
     queue.put([user_id,topic])
     new_state = "generate"
     new_button_text = "Generate"
-    response.set_cookie(key="state", value=new_state)
     queue_len = queue.qsize()
     
-    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(`3 User ID: ${{userId}}, State: ${{state}}`);
-        }};
-    """
+    script = script_template.format(user_id=user_id, state=new_state)
     print(f"current2 user_id={user_id}, state={new_state}")
     # 判断当前是否轮到该用户,如果没轮到则一直等待到轮到为止
     while queue.queue[0] != [user_id,topic]:
@@ -459,4 +434,4 @@ def form_post(request: Request,response: Response,topic: str = Form(...), user_i
             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,script=script_template)
+        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)