fix

app.py

@@ -108,6 +108,7 @@ html_template = """
             color: #007bff;
             animation: fadeIn 0.5s ease-in-out;
             text-align: center;
+            display: none;
         }
         .time-counter-container {
             display: flex;
@@ -298,7 +299,7 @@ html_template = """
                 <button onclick="fillTopic('Using LLM-based agent to generate idea')">Using LLM-based agent to generate idea</button>
             </div>
         </div>
-        <div id="loading" class="loading" display="none">{{ loading_text }}</div>
+        <div id="loading" class="loading">{{ loading_text }}</div>
         {% if error %}
         <div class="error">
             <div class="box">
@@ -309,19 +310,17 @@ html_template = """
         {% endif %}
     </div>
     <script>
-        async function getUserId() {
-            // 检查 sessionStorage 中是否有 user_id
-            let userId = sessionStorage.getItem("user_id");
-            if (!userId) {
-                // 请求新的 user_id
-                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);
-        }
-        window.onload = getUserId;
+       async function getUserId() {
+           let userId = sessionStorage.getItem("user_id");
+           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);
+       }
+       window.onload = getUserId;
     </script>
 </body>
 </html>
@@ -361,25 +360,23 @@ def fix_markdown(text):
 async def add_user_id_and_state_cookie(request: Request, call_next):
     user_id = request.cookies.get("user_id")
     state = request.cookies.get("state")
-    
+    response = await call_next(request)
+
     if not user_id:
         user_id = str(uuid.uuid4())
         state = "generate"
-        response = await call_next(request)
         response.set_cookie(key="user_id", value=user_id)
         response.set_cookie(key="state", value=state)
     elif not state:
         state = "generate"
-        response = await call_next(request)
         response.set_cookie(key="state", value=state)
-    else:
-        response = await call_next(request)
-    
     return response
 
 @app.get("/user_id")
-async def get_user_id():
-    user_id = str(uuid.uuid4())
+async def get_user_id(request: Request):
+    user_id = request.cookies.get("user_id")
+    if not user_id:
+        user_id = str(uuid.uuid4())
     return {"user_id": user_id}
 
 
@@ -393,9 +390,12 @@ def form_post(request: Request,response: Response,topic: str = Form(...)):
     start_time = time.time()
 
     # 获得每个网页的user_id
-    user_id = request.cookies.get("user_id")
+    user_id = request.cookies.get("user_id", str(uuid.uuid4()))
     state = request.cookies.get("state", "generate")
 
+    response.set_cookie(key="user_id", value=user_id)
+    response.set_cookie(key="state", value=state)
+
     print(user_id,state)
     loading_text = "Generating content, Usually takes 2-3 minutes, please wait..."
     if state == "generate":
@@ -404,7 +404,7 @@ def form_post(request: Request,response: Response,topic: str = Form(...)):
             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)
-            error_message = "There are currently {} requests being processed. If you want to queue, please click the Continue button and you will enter the queue.".format(queue_len)
+            error_message = "There are currently {} requests being processed. If you want to queue, please rewrite your 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)