use BrowserState

app.py

@@ -341,7 +341,8 @@ example_prompts = [
 
 # Use a state variable to store session ID
 with gr.Blocks(theme=gr.themes.Soft()) as demo:
-    session_id = gr.State(None)  # Add this to store session ID
+    # session_id = gr.State(None)  # Add this to store session ID
+    session_id = gr.BrowserState()(None)  # Add this to store session ID
 
     gr.Markdown("# 🌊 WaveSpeedAI HiDream Arena")