feat: implement Redis-backed user state management and update app structure

README.md

@@ -1,12 +1,12 @@
 ---
-title: Test
-emoji: 🎮
+title: LLMGameHub
+emoji: "🎮"
 colorFrom: yellow
 colorTo: green
 sdk: gradio
 sdk_version: 5.32.0
 python_version: "3.11"
-app_file: src/main.py
+app_file: src/app.py
 pinned: false
 license: mit
 ---

packages.txt

@@ -0,0 +1 @@
+redis-server

requirements.txt

@@ -7,6 +7,7 @@ asyncio
 aiohttp==3.9.1 
 pygame==2.5.2 
 numpy
+langgraph
 langchain==0.3.17
 langchain-core==0.3.58
 langchain-community==0.3.16
@@ -14,4 +15,6 @@ langchain-google-genai==2.1.4
 pydantic-core==2.23.4
 pydantic-settings==2.7.1
 pydantic==2.9.2
-langgraph
+redis[hiredis]>=5
+aioredis>=2
+msgpack

src/agent/llm_graph.py

@@ -14,7 +14,7 @@ from agent.tools import (
     generate_story_frame,
     update_state_with_choice,
 )
-from agent.state import get_user_state
+from agent.redis_state import get_user_state
 from audio.audio_generator import change_music_tone
 logger = logging.getLogger(__name__)
 

src/agent/redis_state.py

@@ -0,0 +1,51 @@
+"""Async Redis-backed user state storage."""
+
+from __future__ import annotations
+
+import json
+import msgpack
+import redis.asyncio as redis
+
+from agent.models import UserState
+
+
+class UserRepository:
+    """Repository for storing UserState objects in Redis."""
+
+    def __init__(self, redis_url: str = "redis://localhost") -> None:
+        self.redis = redis.from_url(redis_url)
+
+    async def get(self, user_id: str) -> UserState:
+        """Return user state for the given id, creating it if absent."""
+        key = f"llmgamehub:{user_id}"
+        data = await self.redis.hget(key, "data")
+        if data is None:
+            return UserState()
+        state_dict = msgpack.unpackb(data, raw=False)
+        return UserState.parse_obj(state_dict)
+
+    async def set(self, user_id: str, state: UserState) -> None:
+        """Persist updated user state."""
+        key = f"llmgamehub:{user_id}"
+        packed = msgpack.packb(json.loads(state.json()))
+        await self.redis.hset(key, mapping={"data": packed})
+
+    async def reset(self, user_id: str) -> None:
+        """Remove stored state for a user."""
+        key = f"llmgamehub:{user_id}"
+        await self.redis.delete(key)
+
+
+_repo = UserRepository()
+
+
+async def get_user_state(user_hash: str) -> UserState:
+    return await _repo.get(user_hash)
+
+
+async def set_user_state(user_hash: str, state: UserState) -> None:
+    await _repo.set(user_hash, state)
+
+
+async def reset_user_state(user_hash: str) -> None:
+    await _repo.reset(user_hash)

src/agent/runner.py

@@ -10,7 +10,7 @@ from agent.tools import generate_scene_image
 
 from agent.llm_graph import GraphState, llm_game_graph
 from agent.models import UserState
-from agent.state import get_user_state
+from agent.redis_state import get_user_state
 
 logger = logging.getLogger(__name__)
 

src/agent/state.py

@@ -1,24 +0,0 @@
-"""Simple in-memory user state storage."""
-
-from typing import Dict
-
-from agent.models import UserState
-
-_USER_STATE: Dict[str, UserState] = {}
-
-
-def get_user_state(user_hash: str) -> UserState:
-    """Return user state for the given id, creating it if necessary."""
-    if user_hash not in _USER_STATE:
-        _USER_STATE[user_hash] = UserState()
-    return _USER_STATE[user_hash]
-
-
-def set_user_state(user_hash: str, state: UserState) -> None:
-    """Persist updated user state."""
-    _USER_STATE[user_hash] = state
-
-
-def reset_user_state(user_hash: str) -> None:
-    """Reset stored state for a user."""
-    _USER_STATE[user_hash] = UserState()

src/agent/tools.py

@@ -17,7 +17,7 @@ from agent.models import (
     UserChoice,
 )
 from agent.prompts import ENDING_CHECK_PROMPT, SCENE_PROMPT, STORY_FRAME_PROMPT
-from agent.state import get_user_state, set_user_state
+from agent.redis_state import get_user_state, set_user_state
 from images.image_generator import modify_image, generate_image
 from agent.image_agent import ChangeScene
 

src/app.py

@@ -0,0 +1,44 @@
+import subprocess
+import time
+import atexit
+import shutil
+from redis import Redis, ConnectionError
+
+
+REDIS_BIN = shutil.which("redis-server")
+
+if not REDIS_BIN:
+    raise RuntimeError("redis-server not found. Ensure redis is installed via packages.txt")
+
+
+redis_cmd = [
+    REDIS_BIN,
+    "--save",
+    "",
+    "--appendonly",
+    "no",
+    "--dir",
+    "/tmp",
+    "--pidfile",
+    "/tmp/redis.pid",
+]
+redis_process = subprocess.Popen(redis_cmd)
+
+
+redis_client = Redis()
+for _ in range(20):
+    try:
+        redis_client.ping()
+        break
+    except ConnectionError:
+        time.sleep(0.5)
+else:
+    raise RuntimeError("Failed to start redis-server")
+
+
+atexit.register(redis_process.terminate)
+
+
+time.sleep(0.5)
+
+import main

src/main.py

@@ -21,14 +21,15 @@ import asyncio
 from game_setting import get_user_story
 from config import settings
 
+
 logger = logging.getLogger(__name__)
 
 
 async def return_to_constructor(user_hash: str):
     """Return to the constructor and reset user state and audio."""
-    from agent.state import reset_user_state
+    from agent.redis_state import reset_user_state
 
-    reset_user_state(user_hash)
+    await reset_user_state(user_hash)
     await cleanup_music_session(user_hash)
     # Generate a new hash to avoid stale state
     new_hash = str(uuid.uuid4())