Refactor output handling in app.py to use state variables for better clarity and maintainability, while removing unnecessary CSS specification in the Blocks context.

app.py

@@ -219,7 +219,7 @@ css = """
 }
 """
 
-with gr.Blocks(css=css) as demo:
+with gr.Blocks() as demo:
     with gr.Column(elem_id="col-container"):
         gr.HTML("""
         <div id="logo-title">
@@ -380,18 +380,18 @@ with gr.Blocks(css=css) as demo:
             height,
             width,
         ],
-        outputs=[stage2_result, gr.State(), gr.State(), gr.State(), gr.State(), gr.State(), gr.State()],
+        outputs=[stage2_result, state_image, state_seed, state_guidance, state_steps, state_height, state_width],
     )
 
     stage2_event.then(
         fn=infer_combined,
         inputs=[
-            stage2_event.outputs[1],  # image
-            stage2_event.outputs[2],  # seed
-            stage2_event.outputs[3],  # true_guidance_scale
-            stage2_event.outputs[4],  # num_inference_steps
-            stage2_event.outputs[5],  # height
-            stage2_event.outputs[6],  # width
+            state_image,
+            state_seed,
+            state_guidance,
+            state_steps,
+            state_height,
+            state_width,
             stage1_weight,
             stage2_weight,
         ],
@@ -399,4 +399,4 @@ with gr.Blocks(css=css) as demo:
     )
 
 if __name__ == "__main__":
-    demo.queue().launch(mcp_server=True)
+    demo.queue().launch(mcp_server=True, css=css)