feat: implement event handler wrappers and add test suite for Gradio components and app integration

app.py

@@ -40,12 +40,12 @@ async def run_minicpm(func, *args, **kwargs):
 MAX_TRACE_EVENTS = 50
 MAX_HISTORY_ITEMS = 20
 
-def append_trace(state: dict, event: dict) -> None:
+def append_trace(state, event: dict) -> None:
     state["trace"].append(event)
     if len(state["trace"]) > MAX_TRACE_EVENTS:
         state["trace"] = state["trace"][-MAX_TRACE_EVENTS:]
 
-def append_history(state: dict, displayed_task: str, raw_task: str) -> None:
+def append_history(state, displayed_task: str, raw_task: str) -> None:
     state["history"].append(displayed_task)
     state["raw_history"].append(raw_task)
     if len(state["history"]) > MAX_HISTORY_ITEMS:
@@ -252,7 +252,9 @@ def generate_task_html(task: str, animate=True) -> str:
     class_str = "fade-in" if animate else ""
     return f"<div id='current-task' class='{class_str}'>{task}</div>"
 
-async def process_step(state: dict, action: str, goal: str = None, style: str = None, last_task: str = None):
+async def process_step(state, action: str, goal: str = None, style: str = None, last_task: str = None):
+    if not state:
+        state = {}
     # Initialization
     if action == "start":
         state["goal"] = goal
@@ -315,7 +317,7 @@ async def process_step(state: dict, action: str, goal: str = None, style: str =
     return state, generate_task_html(styled_task), generate_breadcrumbs_html(state["history"][-3:]), gr.update(visible=True)
 
 with gr.Blocks(css=custom_css, title="Step-Zero") as app:
-    session_state = gr.State({})
+    session_state = gr.State()
     
     with gr.Column(visible=True) as screen_start:
         gr.HTML("<h1 class='text-4xl font-bold mb-8 text-center mt-20'>What is paralyzing you?</h1>")
@@ -356,12 +358,25 @@ with gr.Blocks(css=custom_css, title="Step-Zero") as app:
         
     # --- Event Handlers ---
     
+    async def handle_start(state, goal, style):
+        return await process_step(state, "start", goal=goal, style=style)
+        
+    async def handle_done(state):
+        if not state: state = {}
+        return await process_step(state, "done", last_task=state.get("current_task"))
+        
+    async def handle_skip(state):
+        return await process_step(state, "skip")
+        
+    async def handle_too_hard(state):
+        return await process_step(state, "too_hard")
+    
     # Start Session
     start_btn.click(
         fn=show_loading,
         outputs=[screen_start, screen_task, task_display, controls_row]
     ).then(
-        fn=lambda state, goal, style: asyncio.run(process_step(state, "start", goal=goal, style=style)),
+        fn=handle_start,
         inputs=[session_state, goal_input, style_radio],
         outputs=[session_state, task_display, breadcrumbs_display, controls_row]
     )
@@ -370,7 +385,7 @@ with gr.Blocks(css=custom_css, title="Step-Zero") as app:
         fn=show_loading,
         outputs=[screen_start, screen_task, task_display, controls_row]
     ).then(
-        fn=lambda state, goal, style: asyncio.run(process_step(state, "start", goal=goal, style=style)),
+        fn=handle_start,
         inputs=[session_state, goal_input, style_radio],
         outputs=[session_state, task_display, breadcrumbs_display, controls_row]
     )
@@ -380,7 +395,7 @@ with gr.Blocks(css=custom_css, title="Step-Zero") as app:
         fn=show_loading_step,
         outputs=[task_display, controls_row]
     ).then(
-        fn=lambda state: asyncio.run(process_step(state, "done", last_task=state["current_task"])),
+        fn=handle_done,
         inputs=[session_state],
         outputs=[session_state, task_display, breadcrumbs_display, controls_row]
     )
@@ -390,7 +405,7 @@ with gr.Blocks(css=custom_css, title="Step-Zero") as app:
         fn=show_loading_step,
         outputs=[task_display, controls_row]
     ).then(
-        fn=lambda state: asyncio.run(process_step(state, "skip")),
+        fn=handle_skip,
         inputs=[session_state],
         outputs=[session_state, task_display, breadcrumbs_display, controls_row]
     )
@@ -400,7 +415,7 @@ with gr.Blocks(css=custom_css, title="Step-Zero") as app:
         fn=show_loading_step,
         outputs=[task_display, controls_row]
     ).then(
-        fn=lambda state: asyncio.run(process_step(state, "too_hard")),
+        fn=handle_too_hard,
         inputs=[session_state],
         outputs=[session_state, task_display, breadcrumbs_display, controls_row]
     )

test_app.py

@@ -0,0 +1,6 @@
+try:
+    import app
+    app.app.get_api_info()
+except Exception as e:
+    import traceback
+    traceback.print_exc()

test_gradio.py

@@ -0,0 +1,4 @@
+import gradio as gr
+with gr.Blocks() as app:
+    session_state = gr.State({})
+app.launch(server_name="0.0.0.0", server_port=7860, share=False)

test_lambda.py

@@ -0,0 +1,10 @@
+import gradio as gr
+
+def test():
+    pass
+
+with gr.Blocks() as app:
+    b = gr.Button()
+    b.click(fn=lambda x: x, inputs=[b], outputs=[b])
+
+app.get_api_info()

test_schema.py

@@ -0,0 +1,7 @@
+import gradio as gr
+from gradio_client import utils
+schema = {"type": "object", "additionalProperties": False}
+try:
+    print(utils.json_schema_to_python_type(schema))
+except Exception as e:
+    print("ERROR:", type(e), str(e))