Update app.py

app.py

@@ -137,11 +137,6 @@ with gr.Blocks(title="Style Mixing for Clothing Design") as iface:
     
     user_id_state = gr.State()
 
-    @iface.load
-    def on_load(request: gr.Request):
-        user_id = request.query_params.get('user_id', '')
-        return user_id
-
     gr.Markdown("## Style Mixing for Clothing Design\nUpload two projected clothing images and mix their styles.")
 
     with gr.Row():
@@ -161,8 +156,25 @@ with gr.Blocks(title="Style Mixing for Clothing Design") as iface:
         result_image, buffer = style_mixing_interface(image1, image2, mix_value)
         return result_image, buffer
 
-    mix_slider.change(mix_and_store, inputs=[image1_input, image2_input, mix_slider], outputs=[output_image, image_buffer])
-    save_button.click(send_to_backend, inputs=[image_buffer, user_id_state], outputs=[save_status])
+    mix_slider.change(
+        mix_and_store, 
+        inputs=[image1_input, image2_input, mix_slider], 
+        outputs=[output_image, image_buffer]
+    )
+
+    save_button.click(
+        send_to_backend, 
+        inputs=[image_buffer, user_id_state], 
+        outputs=[save_status]
+    )
+
+    @iface.load
+    def on_load(request: gr.Request):
+        user_id = request.query_params.get("user_id", "")
+        return gr.update(value=user_id)  # This updates the user_id_state
+
+    # Bind the return of on_load to user_id_state
+    iface.load(on_load, None, [user_id_state])
 
 iface.launch()