Build MCP server

app.py

@@ -60,7 +60,20 @@ class WordleEnv:
         return self._obs
 
 
-def step_fn(guess: str, wordle) -> str:
+def step_fn(guess: str, wordle) -> tuple[str, float, bool, dict]:
+    """
+    Perform a step in the Wordle environment.
+
+    Args:
+        guess (str): The guessed word (4-letter lowercase string).
+
+    Returns:
+        tuple[str, float, bool, dict]: A tuple containing:
+            - observation: The observation after the step .
+            - reward: The reward obtained from the step.
+            - done: Whether the game is done.
+            - info: Additional info.
+    """
     result = wordle.step(guess)
     return result.observation, result.reward, result.done, result.info, wordle
 
@@ -79,4 +92,4 @@ demo = gr.Interface(
 
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch(mcp_server=True)

requirements.txt

@@ -1 +1 @@
-gradio==5.46.0  # Set in README.md
+gradio[mcp]==5.46.0  # Set in README.md