Spaces:
Running
Running
Commit
·
86b351a
0
Parent(s):
A new start
Browse files- .env.example +1 -0
- .gitattributes +35 -0
- .gitignore +178 -0
- README.md +57 -0
- requirements.txt +16 -0
- src/agent/game_generator.py +0 -0
- src/agent/llm.py +43 -0
- src/agent/llm_agent.py +38 -0
- src/agent/llm_graph.py +14 -0
- src/agent/tools.py +40 -0
- src/audio/audio_generator.py +103 -0
- src/config.py +28 -0
- src/css.py +126 -0
- src/game_constructor.py +183 -0
- src/game_setting.py +12 -0
- src/game_state.py +15 -0
- src/images/image_generator.py +140 -0
- src/main.py +320 -0
.env.example
ADDED
@@ -0,0 +1 @@
|
|
|
|
|
1 |
+
GEMINI_API_KEY=KEY
|
.gitattributes
ADDED
@@ -0,0 +1,35 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
*.7z filter=lfs diff=lfs merge=lfs -text
|
2 |
+
*.arrow filter=lfs diff=lfs merge=lfs -text
|
3 |
+
*.bin filter=lfs diff=lfs merge=lfs -text
|
4 |
+
*.bz2 filter=lfs diff=lfs merge=lfs -text
|
5 |
+
*.ckpt filter=lfs diff=lfs merge=lfs -text
|
6 |
+
*.ftz filter=lfs diff=lfs merge=lfs -text
|
7 |
+
*.gz filter=lfs diff=lfs merge=lfs -text
|
8 |
+
*.h5 filter=lfs diff=lfs merge=lfs -text
|
9 |
+
*.joblib filter=lfs diff=lfs merge=lfs -text
|
10 |
+
*.lfs.* filter=lfs diff=lfs merge=lfs -text
|
11 |
+
*.mlmodel filter=lfs diff=lfs merge=lfs -text
|
12 |
+
*.model filter=lfs diff=lfs merge=lfs -text
|
13 |
+
*.msgpack filter=lfs diff=lfs merge=lfs -text
|
14 |
+
*.npy filter=lfs diff=lfs merge=lfs -text
|
15 |
+
*.npz filter=lfs diff=lfs merge=lfs -text
|
16 |
+
*.onnx filter=lfs diff=lfs merge=lfs -text
|
17 |
+
*.ot filter=lfs diff=lfs merge=lfs -text
|
18 |
+
*.parquet filter=lfs diff=lfs merge=lfs -text
|
19 |
+
*.pb filter=lfs diff=lfs merge=lfs -text
|
20 |
+
*.pickle filter=lfs diff=lfs merge=lfs -text
|
21 |
+
*.pkl filter=lfs diff=lfs merge=lfs -text
|
22 |
+
*.pt filter=lfs diff=lfs merge=lfs -text
|
23 |
+
*.pth filter=lfs diff=lfs merge=lfs -text
|
24 |
+
*.rar filter=lfs diff=lfs merge=lfs -text
|
25 |
+
*.safetensors filter=lfs diff=lfs merge=lfs -text
|
26 |
+
saved_model/**/* filter=lfs diff=lfs merge=lfs -text
|
27 |
+
*.tar.* filter=lfs diff=lfs merge=lfs -text
|
28 |
+
*.tar filter=lfs diff=lfs merge=lfs -text
|
29 |
+
*.tflite filter=lfs diff=lfs merge=lfs -text
|
30 |
+
*.tgz filter=lfs diff=lfs merge=lfs -text
|
31 |
+
*.wasm filter=lfs diff=lfs merge=lfs -text
|
32 |
+
*.xz filter=lfs diff=lfs merge=lfs -text
|
33 |
+
*.zip filter=lfs diff=lfs merge=lfs -text
|
34 |
+
*.zst filter=lfs diff=lfs merge=lfs -text
|
35 |
+
*tfevents* filter=lfs diff=lfs merge=lfs -text
|
.gitignore
ADDED
@@ -0,0 +1,178 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
# Byte-compiled / optimized / DLL files
|
2 |
+
__pycache__/
|
3 |
+
*.py[cod]
|
4 |
+
*$py.class
|
5 |
+
|
6 |
+
# C extensions
|
7 |
+
*.so
|
8 |
+
|
9 |
+
# Distribution / packaging
|
10 |
+
.Python
|
11 |
+
build/
|
12 |
+
develop-eggs/
|
13 |
+
dist/
|
14 |
+
downloads/
|
15 |
+
eggs/
|
16 |
+
.eggs/
|
17 |
+
lib/
|
18 |
+
lib64/
|
19 |
+
parts/
|
20 |
+
sdist/
|
21 |
+
var/
|
22 |
+
wheels/
|
23 |
+
share/python-wheels/
|
24 |
+
*.egg-info/
|
25 |
+
.installed.cfg
|
26 |
+
*.egg
|
27 |
+
MANIFEST
|
28 |
+
|
29 |
+
# PyInstaller
|
30 |
+
# Usually these files are written by a python script from a template
|
31 |
+
# before PyInstaller builds the exe, so as to inject date/other infos into it.
|
32 |
+
*.manifest
|
33 |
+
*.spec
|
34 |
+
|
35 |
+
# Installer logs
|
36 |
+
pip-log.txt
|
37 |
+
pip-delete-this-directory.txt
|
38 |
+
|
39 |
+
# Unit test / coverage reports
|
40 |
+
htmlcov/
|
41 |
+
.tox/
|
42 |
+
.nox/
|
43 |
+
.coverage
|
44 |
+
.coverage.*
|
45 |
+
.cache
|
46 |
+
nosetests.xml
|
47 |
+
coverage.xml
|
48 |
+
*.cover
|
49 |
+
*.py,cover
|
50 |
+
.hypothesis/
|
51 |
+
.pytest_cache/
|
52 |
+
cover/
|
53 |
+
|
54 |
+
# Translations
|
55 |
+
*.mo
|
56 |
+
*.pot
|
57 |
+
|
58 |
+
# Django stuff:
|
59 |
+
*.log
|
60 |
+
local_settings.py
|
61 |
+
db.sqlite3
|
62 |
+
db.sqlite3-journal
|
63 |
+
|
64 |
+
# Flask stuff:
|
65 |
+
instance/
|
66 |
+
.webassets-cache
|
67 |
+
|
68 |
+
# Scrapy stuff:
|
69 |
+
.scrapy
|
70 |
+
|
71 |
+
# Sphinx documentation
|
72 |
+
docs/_build/
|
73 |
+
|
74 |
+
# PyBuilder
|
75 |
+
.pybuilder/
|
76 |
+
target/
|
77 |
+
|
78 |
+
# Jupyter Notebook
|
79 |
+
.ipynb_checkpoints
|
80 |
+
|
81 |
+
# IPython
|
82 |
+
profile_default/
|
83 |
+
ipython_config.py
|
84 |
+
|
85 |
+
# pyenv
|
86 |
+
# For a library or package, you might want to ignore these files since the code is
|
87 |
+
# intended to run in multiple environments; otherwise, check them in:
|
88 |
+
# .python-version
|
89 |
+
|
90 |
+
# pipenv
|
91 |
+
# According to pypa/pipenv#598, it is recommended to include Pipfile.lock in version control.
|
92 |
+
# However, in case of collaboration, if having platform-specific dependencies or dependencies
|
93 |
+
# having no cross-platform support, pipenv may install dependencies that don't work, or not
|
94 |
+
# install all needed dependencies.
|
95 |
+
#Pipfile.lock
|
96 |
+
|
97 |
+
# UV
|
98 |
+
# Similar to Pipfile.lock, it is generally recommended to include uv.lock in version control.
|
99 |
+
# This is especially recommended for binary packages to ensure reproducibility, and is more
|
100 |
+
# commonly ignored for libraries.
|
101 |
+
#uv.lock
|
102 |
+
|
103 |
+
# poetry
|
104 |
+
# Similar to Pipfile.lock, it is generally recommended to include poetry.lock in version control.
|
105 |
+
# This is especially recommended for binary packages to ensure reproducibility, and is more
|
106 |
+
# commonly ignored for libraries.
|
107 |
+
# https://python-poetry.org/docs/basic-usage/#commit-your-poetrylock-file-to-version-control
|
108 |
+
#poetry.lock
|
109 |
+
|
110 |
+
# pdm
|
111 |
+
# Similar to Pipfile.lock, it is generally recommended to include pdm.lock in version control.
|
112 |
+
#pdm.lock
|
113 |
+
# pdm stores project-wide configurations in .pdm.toml, but it is recommended to not include it
|
114 |
+
# in version control.
|
115 |
+
# https://pdm.fming.dev/latest/usage/project/#working-with-version-control
|
116 |
+
.pdm.toml
|
117 |
+
.pdm-python
|
118 |
+
.pdm-build/
|
119 |
+
|
120 |
+
# PEP 582; used by e.g. github.com/David-OConnor/pyflow and github.com/pdm-project/pdm
|
121 |
+
__pypackages__/
|
122 |
+
|
123 |
+
# Celery stuff
|
124 |
+
celerybeat-schedule
|
125 |
+
celerybeat.pid
|
126 |
+
|
127 |
+
# SageMath parsed files
|
128 |
+
*.sage.py
|
129 |
+
|
130 |
+
# Environments
|
131 |
+
.env
|
132 |
+
.venv
|
133 |
+
env/
|
134 |
+
venv/
|
135 |
+
ENV/
|
136 |
+
env.bak/
|
137 |
+
venv.bak/
|
138 |
+
|
139 |
+
# Spyder project settings
|
140 |
+
.spyderproject
|
141 |
+
.spyproject
|
142 |
+
|
143 |
+
# Rope project settings
|
144 |
+
.ropeproject
|
145 |
+
|
146 |
+
# mkdocs documentation
|
147 |
+
/site
|
148 |
+
|
149 |
+
# mypy
|
150 |
+
.mypy_cache/
|
151 |
+
.dmypy.json
|
152 |
+
dmypy.json
|
153 |
+
|
154 |
+
# Pyre type checker
|
155 |
+
.pyre/
|
156 |
+
|
157 |
+
# pytype static type analyzer
|
158 |
+
.pytype/
|
159 |
+
|
160 |
+
# Cython debug symbols
|
161 |
+
cython_debug/
|
162 |
+
|
163 |
+
# PyCharm
|
164 |
+
# JetBrains specific template is maintained in a separate JetBrains.gitignore that can
|
165 |
+
# be found at https://github.com/github/gitignore/blob/main/Global/JetBrains.gitignore
|
166 |
+
# and can be added to the global gitignore or merged into this file. For a more nuclear
|
167 |
+
# option (not recommended) you can uncomment the following to ignore the entire idea folder.
|
168 |
+
#.idea/
|
169 |
+
|
170 |
+
# Ruff stuff:
|
171 |
+
.ruff_cache/
|
172 |
+
|
173 |
+
# PyPI configuration file
|
174 |
+
.pypirc
|
175 |
+
versions/
|
176 |
+
|
177 |
+
.gradio
|
178 |
+
generated/
|
README.md
ADDED
@@ -0,0 +1,57 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
---
|
2 |
+
title: Test
|
3 |
+
emoji: 🎮
|
4 |
+
colorFrom: yellow
|
5 |
+
colorTo: green
|
6 |
+
sdk: gradio
|
7 |
+
sdk_version: 5.32.0
|
8 |
+
app_file: src/main.py
|
9 |
+
pinned: false
|
10 |
+
license: mit
|
11 |
+
---
|
12 |
+
|
13 |
+
# 🎮 LLMGameHub | gradio-mcp-hackaton
|
14 |
+
|
15 |
+
**LLMGameHub** is a next-generation playground where anyone can create, share, and play AI-powered games based on their own stories and texts.
|
16 |
+
|
17 |
+
|
18 |
+
https://github.com/user-attachments/assets/1ec276e2-ba95-4e85-b5b9-e8831703a52c
|
19 |
+
|
20 |
+
|
21 |
+
## What is LLMGameHub?
|
22 |
+
|
23 |
+
LLMGameHub is an interactive platform that brings together the power of large language models and generative AI to let users co-create narrative games, visual novels, and interactive experiences in just a few clicks.
|
24 |
+
Whether you're a writer, gamer, or simply curious, you can turn your ideas into playable games — no coding required!
|
25 |
+
|
26 |
+
## Key Features
|
27 |
+
|
28 |
+
- **User-driven Storytelling:**
|
29 |
+
Players build their own visual novel game by selecting key elements: setting, main character, and genre (e.g., horror, detective, romance, adventure).
|
30 |
+
- **Auto-generated Content:**
|
31 |
+
The platform dynamically generates engaging narratives, character descriptions, dialogue, and visual assets (images and short videos) tailored to the user's preferences.
|
32 |
+
- **Multi-Agent LLM System:**
|
33 |
+
Behind every game is a team of collaborating AI agents — each with their own roles — to create rich, believable, and creative worlds.
|
34 |
+
- **Short, Replayable Sessions:**
|
35 |
+
Each game is designed for quick, fun playthroughs (~5 minutes), making it perfect for demos and repeated experimentation.
|
36 |
+
- **Fast and Immersive:**
|
37 |
+
Instant feedback for user choices (responses in under 5 seconds) and high-quality assets generated in the background.
|
38 |
+
Atmospheric audio and multiple possible endings are included for full immersion.
|
39 |
+
- **Creative Freedom:**
|
40 |
+
Anyone can build a game from their own story idea or inspiration. Share your creation or play games made by others.
|
41 |
+
|
42 |
+
## Why LLMGameHub?
|
43 |
+
|
44 |
+
We believe AI should be a creative partner. With LLMGameHub, you don't just play games — you shape them, experiment with genres, and explore what's possible when human imagination meets the latest in generative AI.
|
45 |
+
Bring your story to life — or let AI surprise you!
|
46 |
+
|
47 |
+
## How does it work?
|
48 |
+
|
49 |
+
1. **Start:** Choose a game genre, world setting, and main character.
|
50 |
+
2. **Co-Create:** Answer a few simple questions or pick from suggestions.
|
51 |
+
3. **Watch the Magic:** The AI agents generate a unique story, visuals, audio, and game flow based on your choices.
|
52 |
+
4. **Play:** Dive into your own visual novel, make decisions, and explore different endings.
|
53 |
+
5. **Share:** Save and share your favorite games with friends (feature coming soon).
|
54 |
+
|
55 |
+
---
|
56 |
+
|
57 |
+
**LLMGameHub** — Create. Play. Imagine.
|
requirements.txt
ADDED
@@ -0,0 +1,16 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
gradio==5.32.0
|
2 |
+
google-genai==1.18.0
|
3 |
+
Pillow==10.1.0
|
4 |
+
requests==2.31.0
|
5 |
+
python-dotenv==1.0.0
|
6 |
+
asyncio
|
7 |
+
aiohttp==3.9.1
|
8 |
+
pygame==2.5.2
|
9 |
+
numpy
|
10 |
+
langchain==0.3.17
|
11 |
+
langchain-core==0.3.58
|
12 |
+
langchain-community==0.3.16
|
13 |
+
langchain-google-genai==2.1.4
|
14 |
+
pydantic-core==2.23.4
|
15 |
+
pydantic-settings==2.7.1
|
16 |
+
pydantic==2.9.2
|
src/agent/game_generator.py
ADDED
File without changes
|
src/agent/llm.py
ADDED
@@ -0,0 +1,43 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
from langchain_google_genai import ChatGoogleGenerativeAI
|
2 |
+
import logging
|
3 |
+
from config import settings
|
4 |
+
|
5 |
+
logger = logging.getLogger(__name__)
|
6 |
+
|
7 |
+
_google_api_keys_list = []
|
8 |
+
_current_google_key_idx = 0
|
9 |
+
|
10 |
+
|
11 |
+
def create_llm(temperature: float = settings.temperature, top_p: float = settings.top_p):
|
12 |
+
global _google_api_keys_list, _current_google_key_idx
|
13 |
+
|
14 |
+
if not _google_api_keys_list:
|
15 |
+
api_keys_str = settings.gemini_api_key.get_secret_value()
|
16 |
+
if api_keys_str:
|
17 |
+
_google_api_keys_list = [key.strip() for key in api_keys_str.split(',') if key.strip()]
|
18 |
+
|
19 |
+
if not _google_api_keys_list:
|
20 |
+
logger.error("Google API keys are not configured or are empty in settings.")
|
21 |
+
raise ValueError("Google API keys are not configured or are invalid for round-robin.")
|
22 |
+
|
23 |
+
if not _google_api_keys_list: # Safeguard, though previous block should handle it.
|
24 |
+
logger.error("No Google API keys available for round-robin.")
|
25 |
+
raise ValueError("No Google API keys available for round-robin.")
|
26 |
+
|
27 |
+
key_index_to_use = _current_google_key_idx
|
28 |
+
selected_api_key = _google_api_keys_list[key_index_to_use]
|
29 |
+
|
30 |
+
_current_google_key_idx = (key_index_to_use + 1) % len(_google_api_keys_list)
|
31 |
+
|
32 |
+
logger.info(f"Using Google API key at index {key_index_to_use} (ending with ...{selected_api_key[-4:] if len(selected_api_key) > 4 else selected_api_key}) for round-robin.")
|
33 |
+
|
34 |
+
return ChatGoogleGenerativeAI(
|
35 |
+
model="gemini-2.5-flash-preview-05-20",
|
36 |
+
google_api_key=selected_api_key,
|
37 |
+
temperature=temperature,
|
38 |
+
top_p=top_p,
|
39 |
+
thinking_budget=1024
|
40 |
+
)
|
41 |
+
|
42 |
+
def create_precise_llm():
|
43 |
+
return create_llm(temperature=0, top_p=1)
|
src/agent/llm_agent.py
ADDED
@@ -0,0 +1,38 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
from agent.llm import create_llm
|
2 |
+
from pydantic import BaseModel, Field
|
3 |
+
from typing import Optional, List
|
4 |
+
import logging
|
5 |
+
|
6 |
+
logger = logging.getLogger(__name__)
|
7 |
+
|
8 |
+
class ChangeScene(BaseModel):
|
9 |
+
change_scene: bool = Field(description="Whether the scene should be changed")
|
10 |
+
scene_description: Optional[str] = None
|
11 |
+
|
12 |
+
class ChangeMusic(BaseModel):
|
13 |
+
change_music: bool = Field(description="Whether the music should be changed")
|
14 |
+
music_description: Optional[str] = None
|
15 |
+
|
16 |
+
class PlayerOption(BaseModel):
|
17 |
+
option_description: str = Field(description="The description of the option, Examples: [Change location] Go to the forest; [Say] Hello!")
|
18 |
+
|
19 |
+
class LLMOutput(BaseModel):
|
20 |
+
change_scene: ChangeScene
|
21 |
+
change_music: ChangeMusic
|
22 |
+
game_message: str = Field(description="The message to the player, Example: You entered the forest, and you see unknown scary creatures. What do you do?")
|
23 |
+
player_options: List[PlayerOption] = Field(description="The list of up to 3 options for the player to choose from.")
|
24 |
+
|
25 |
+
llm = create_llm().with_structured_output(LLMOutput)
|
26 |
+
|
27 |
+
async def process_user_input(input: str) -> LLMOutput:
|
28 |
+
"""
|
29 |
+
Process user input and update the state.
|
30 |
+
"""
|
31 |
+
logger.info(f"User's choice: {input}")
|
32 |
+
|
33 |
+
response: LLMOutput = await llm.ainvoke(input)
|
34 |
+
|
35 |
+
logger.info(f"LLM response: {response}")
|
36 |
+
|
37 |
+
return response
|
38 |
+
|
src/agent/llm_graph.py
ADDED
@@ -0,0 +1,14 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
from agent.tools import available_tools
|
2 |
+
from agent.llm import create_llm
|
3 |
+
from langgraph.graph import MessagesState
|
4 |
+
class CustomState(MessagesState):
|
5 |
+
"""Расширенное состояние графа."""
|
6 |
+
|
7 |
+
|
8 |
+
|
9 |
+
llm = create_llm().bind_tools(available_tools)
|
10 |
+
|
11 |
+
|
12 |
+
|
13 |
+
|
14 |
+
|
src/agent/tools.py
ADDED
@@ -0,0 +1,40 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
from langchain_core.tools import tool
|
2 |
+
from typing import Annotated, Any, Dict, List
|
3 |
+
from images.image_generator import generate_image
|
4 |
+
from langgraph.prebuilt import InjectedState
|
5 |
+
import logging
|
6 |
+
|
7 |
+
logger = logging.getLogger(__name__)
|
8 |
+
|
9 |
+
def _err(msg: str) -> str:
|
10 |
+
logger.error(msg)
|
11 |
+
return f"{{ 'error': '{msg}' }}"
|
12 |
+
|
13 |
+
def _success(msg: str) -> str:
|
14 |
+
logger.info(msg)
|
15 |
+
return f"{{ 'success': '{msg}' }}"
|
16 |
+
|
17 |
+
@tool
|
18 |
+
async def generate_scene_image(
|
19 |
+
prompt: Annotated[
|
20 |
+
str,
|
21 |
+
"The prompt to generate an image from"
|
22 |
+
],
|
23 |
+
state: InjectedState,
|
24 |
+
) -> Annotated[
|
25 |
+
str,
|
26 |
+
"The path to the generated image"
|
27 |
+
]:
|
28 |
+
"""
|
29 |
+
Generate an image from a prompt and set current scene image.
|
30 |
+
"""
|
31 |
+
try:
|
32 |
+
image_path, img_description = generate_image(prompt)
|
33 |
+
state["current_scene"]["image"] = image_path
|
34 |
+
state["current_scene"]["image_description"] = img_description
|
35 |
+
return _success(f"Image generated and set as current scene image: {img_description}")
|
36 |
+
except Exception as e:
|
37 |
+
return _err(str(e))
|
38 |
+
|
39 |
+
|
40 |
+
available_tools = [generate_scene_image]
|
src/audio/audio_generator.py
ADDED
@@ -0,0 +1,103 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
import asyncio
|
2 |
+
from google import genai
|
3 |
+
from google.genai import types
|
4 |
+
from config import settings
|
5 |
+
import os
|
6 |
+
import tempfile
|
7 |
+
import wave
|
8 |
+
import numpy as np
|
9 |
+
import queue
|
10 |
+
import logging
|
11 |
+
import gradio as gr
|
12 |
+
|
13 |
+
logger = logging.getLogger(__name__)
|
14 |
+
|
15 |
+
client = genai.Client(api_key=settings.gemini_api_key.get_secret_value(), http_options={'api_version': 'v1alpha'})
|
16 |
+
audio_queue = queue.Queue(maxsize=1)
|
17 |
+
|
18 |
+
async def generate_music(request: gr.Request, receive_audio):
|
19 |
+
async with (
|
20 |
+
client.aio.live.music.connect(model='models/lyria-realtime-exp') as session,
|
21 |
+
asyncio.TaskGroup() as tg,
|
22 |
+
):
|
23 |
+
# Set up task to receive server messages.
|
24 |
+
tg.create_task(receive_audio(session))
|
25 |
+
|
26 |
+
# Send initial prompts and config
|
27 |
+
await session.set_weighted_prompts(
|
28 |
+
prompts=[
|
29 |
+
types.WeightedPrompt(text='The mysterious music of the forest', weight=1.0),
|
30 |
+
]
|
31 |
+
)
|
32 |
+
await session.set_music_generation_config(
|
33 |
+
config=types.LiveMusicGenerationConfig(bpm=90, temperature=1.0)
|
34 |
+
)
|
35 |
+
await session.play()
|
36 |
+
logger.info(f"Started music generation for session {request.session_hash}")
|
37 |
+
sessions[request.session_hash] = session
|
38 |
+
|
39 |
+
async def change_music_tone(request: gr.Request, new_tone):
|
40 |
+
logger.info(f"Changing music tone to {new_tone}")
|
41 |
+
session = sessions.get(request.session_hash)
|
42 |
+
if not session:
|
43 |
+
logger.error(f"No session found for request {request.session_hash}")
|
44 |
+
return
|
45 |
+
await session.reset_context()
|
46 |
+
await session.set_weighted_prompts(
|
47 |
+
prompts=[types.WeightedPrompt(text=new_tone, weight=1.0)]
|
48 |
+
)
|
49 |
+
|
50 |
+
|
51 |
+
SAMPLE_RATE = 48000
|
52 |
+
|
53 |
+
async def receive_audio(session):
|
54 |
+
"""Process incoming audio from the music generation."""
|
55 |
+
while True:
|
56 |
+
try:
|
57 |
+
async for message in session.receive():
|
58 |
+
if message.server_content and message.server_content.audio_chunks:
|
59 |
+
audio_data = message.server_content.audio_chunks[0].data
|
60 |
+
audio_queue.put(audio_data)
|
61 |
+
await asyncio.sleep(10**-12)
|
62 |
+
except Exception as e:
|
63 |
+
logger.error(f"Error in receive_audio: {e}")
|
64 |
+
await asyncio.sleep(1)
|
65 |
+
|
66 |
+
sessions = {}
|
67 |
+
|
68 |
+
async def start_music_generation(request: gr.Request):
|
69 |
+
"""Start the music generation in a separate thread."""
|
70 |
+
await generate_music(request, receive_audio)
|
71 |
+
|
72 |
+
async def cleanup_music_session(request: gr.Request):
|
73 |
+
if request.session_hash in sessions:
|
74 |
+
logger.info(f"Cleaning up music session for session {request.session_hash}")
|
75 |
+
await sessions[request.session_hash].stop()
|
76 |
+
del sessions[request.session_hash]
|
77 |
+
|
78 |
+
current_audio_file = None
|
79 |
+
|
80 |
+
def update_audio():
|
81 |
+
"""Continuously stream audio from the queue."""
|
82 |
+
global current_audio_file
|
83 |
+
while True:
|
84 |
+
audio_data = audio_queue.get()
|
85 |
+
if isinstance(audio_data, bytes):
|
86 |
+
audio_array = np.frombuffer(audio_data, dtype=np.int16)
|
87 |
+
else:
|
88 |
+
audio_array = np.array(audio_data, dtype=np.int16)
|
89 |
+
|
90 |
+
temp_fd, temp_path = tempfile.mkstemp(suffix='.wav')
|
91 |
+
os.close(temp_fd)
|
92 |
+
# Write to WAV file
|
93 |
+
with wave.open(temp_path, 'wb') as wav_file:
|
94 |
+
wav_file.setnchannels(2) # Stereo
|
95 |
+
wav_file.setsampwidth(2) # 16-bit
|
96 |
+
wav_file.setframerate(SAMPLE_RATE)
|
97 |
+
wav_file.writeframes(audio_array.tobytes())
|
98 |
+
|
99 |
+
if current_audio_file:
|
100 |
+
os.remove(current_audio_file)
|
101 |
+
|
102 |
+
current_audio_file = temp_path
|
103 |
+
yield temp_path
|
src/config.py
ADDED
@@ -0,0 +1,28 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
from dotenv import load_dotenv
|
2 |
+
|
3 |
+
load_dotenv()
|
4 |
+
from pydantic_settings import BaseSettings
|
5 |
+
import logging
|
6 |
+
from pydantic import SecretStr
|
7 |
+
|
8 |
+
logging.basicConfig(
|
9 |
+
level=logging.INFO,
|
10 |
+
format="%(levelname)s:\t%(asctime)s [%(name)s] %(message)s",
|
11 |
+
datefmt="%Y-%m-%d %H:%M:%S %z",
|
12 |
+
)
|
13 |
+
|
14 |
+
class BaseAppSettings(BaseSettings):
|
15 |
+
"""Base settings class with common configuration."""
|
16 |
+
|
17 |
+
class Config:
|
18 |
+
env_file = ".env"
|
19 |
+
env_file_encoding = "utf-8"
|
20 |
+
extra = "ignore"
|
21 |
+
|
22 |
+
class AppSettings(BaseAppSettings):
|
23 |
+
gemini_api_key: SecretStr
|
24 |
+
top_p: float = 0.95
|
25 |
+
temperature: float = 0.5
|
26 |
+
|
27 |
+
|
28 |
+
settings = AppSettings()
|
src/css.py
ADDED
@@ -0,0 +1,126 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
# Custom CSS for fullscreen image with overlay
|
2 |
+
custom_css = """
|
3 |
+
/* Make the image container fullscreen */
|
4 |
+
.image-container {
|
5 |
+
position: fixed !important;
|
6 |
+
top: 0 !important;
|
7 |
+
left: 0 !important;
|
8 |
+
width: 100vw !important;
|
9 |
+
height: 100vh !important;
|
10 |
+
z-index: 1 !important;
|
11 |
+
}
|
12 |
+
|
13 |
+
.image-container img {
|
14 |
+
width: 100vw !important;
|
15 |
+
height: 100vh !important;
|
16 |
+
object-fit: cover !important;
|
17 |
+
}
|
18 |
+
|
19 |
+
/* Style the overlay content */
|
20 |
+
.overlay-content {
|
21 |
+
position: fixed !important;
|
22 |
+
bottom: 0 !important;
|
23 |
+
left: 0 !important;
|
24 |
+
right: 0 !important;
|
25 |
+
background: linear-gradient(transparent, rgba(0,0,0,0.8)) !important;
|
26 |
+
padding: 40px 20px 20px !important;
|
27 |
+
z-index: 10 !important;
|
28 |
+
color: white !important;
|
29 |
+
}
|
30 |
+
|
31 |
+
/* Style the narrative text */
|
32 |
+
.narrative-text {
|
33 |
+
background: rgba(0,0,0,0.7) !important;
|
34 |
+
border: none !important;
|
35 |
+
color: white !important;
|
36 |
+
font-size: 18px !important;
|
37 |
+
line-height: 1.5 !important;
|
38 |
+
padding: 20px !important;
|
39 |
+
border-radius: 10px !important;
|
40 |
+
margin-bottom: 20px !important;
|
41 |
+
}
|
42 |
+
|
43 |
+
img {
|
44 |
+
pointer-events: none;
|
45 |
+
}
|
46 |
+
|
47 |
+
.narrative-text textarea {
|
48 |
+
background: transparent !important;
|
49 |
+
border: none !important;
|
50 |
+
color: white !important;
|
51 |
+
-webkit-text-fill-color: white !important;
|
52 |
+
font-size: 18px !important;
|
53 |
+
resize: none !important;
|
54 |
+
}
|
55 |
+
|
56 |
+
/* Style the choice buttons */
|
57 |
+
.choice-buttons {
|
58 |
+
background: rgba(0,0,0,0.7) !important;
|
59 |
+
border-radius: 10px !important;
|
60 |
+
padding: 15px !important;
|
61 |
+
}
|
62 |
+
|
63 |
+
.choice-buttons label {
|
64 |
+
color: white !important;
|
65 |
+
font-size: 16px !important;
|
66 |
+
margin-bottom: 10px !important;
|
67 |
+
}
|
68 |
+
|
69 |
+
/* Fix radio button backgrounds */
|
70 |
+
.choice-buttons input[type="radio"] {
|
71 |
+
background: transparent !important;
|
72 |
+
border: 2px solid white !important;
|
73 |
+
}
|
74 |
+
|
75 |
+
.choice-buttons input[type="radio"]:checked {
|
76 |
+
background: white !important;
|
77 |
+
}
|
78 |
+
|
79 |
+
.choice-buttons .form-radio {
|
80 |
+
background: transparent !important;
|
81 |
+
}
|
82 |
+
|
83 |
+
/* Style radio button containers */
|
84 |
+
.choice-buttons > div {
|
85 |
+
background: transparent !important;
|
86 |
+
}
|
87 |
+
|
88 |
+
.choice-buttons fieldset {
|
89 |
+
background: transparent !important;
|
90 |
+
border: none !important;
|
91 |
+
}
|
92 |
+
|
93 |
+
/* Remove any remaining white backgrounds */
|
94 |
+
.choice-buttons * {
|
95 |
+
background-color: transparent !important;
|
96 |
+
}
|
97 |
+
|
98 |
+
.choice-buttons input {
|
99 |
+
background-color: transparent !important;
|
100 |
+
border: 1px solid rgba(255,255,255,0.5) !important;
|
101 |
+
color: white !important;
|
102 |
+
}
|
103 |
+
|
104 |
+
.choice-buttons label span {
|
105 |
+
color: white !important;
|
106 |
+
}
|
107 |
+
|
108 |
+
/* Hide gradio header and footer */
|
109 |
+
.gradio-header, .gradio-footer {
|
110 |
+
display: none !important;
|
111 |
+
}
|
112 |
+
|
113 |
+
/* Hide image control buttons using correct DOM selector */
|
114 |
+
.image-container .icon-button-wrapper {
|
115 |
+
display: none !important;
|
116 |
+
}
|
117 |
+
|
118 |
+
.image-container .icon-buttons {
|
119 |
+
display: none !important;
|
120 |
+
}
|
121 |
+
|
122 |
+
/* Make form element transparent */
|
123 |
+
.overlay-content .form {
|
124 |
+
background: transparent !important;
|
125 |
+
}
|
126 |
+
"""
|
src/game_constructor.py
ADDED
@@ -0,0 +1,183 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
import gradio as gr
|
2 |
+
import json
|
3 |
+
import uuid
|
4 |
+
from game_setting import Character, GameSetting
|
5 |
+
from game_state import story, state, get_current_scene
|
6 |
+
from agent.llm_agent import process_user_input
|
7 |
+
from images.image_generator import generate_image
|
8 |
+
|
9 |
+
# Predefined suggestions for demo
|
10 |
+
SETTING_SUGGESTIONS = [
|
11 |
+
"A mystical forest shrouded in eternal twilight, where ancient trees whisper secrets and magical creatures roam freely",
|
12 |
+
"A sprawling cyberpunk metropolis in 2099, where neon lights illuminate towering skyscrapers and technology controls every aspect of life",
|
13 |
+
"A Victorian-era mansion on a remote cliff, filled with hidden passages, antique furniture, and an atmosphere of dark mysteries",
|
14 |
+
"A post-apocalyptic wasteland where survivors struggle to rebuild civilization among the ruins of the old world",
|
15 |
+
"A magical academy floating in the clouds, where young wizards learn to master their powers and uncover ancient spells",
|
16 |
+
]
|
17 |
+
|
18 |
+
CHARACTER_SUGGESTIONS = [
|
19 |
+
{
|
20 |
+
"name": "Elena Nightwhisper",
|
21 |
+
"age": "25",
|
22 |
+
"background": "A skilled detective with supernatural intuition, haunted by visions of crimes before they happen",
|
23 |
+
"personality": "Determined, intuitive, struggles with self-doubt but fiercely protective of the innocent",
|
24 |
+
},
|
25 |
+
{
|
26 |
+
"name": "Marcus Steelborn",
|
27 |
+
"age": "32",
|
28 |
+
"background": "A former soldier turned cybernetic engineer in a dystopian future, seeking to expose corporate corruption",
|
29 |
+
"personality": "Brave, tech-savvy, has trust issues but deeply loyal to those who earn his respect",
|
30 |
+
},
|
31 |
+
{
|
32 |
+
"name": "Aria Moonstone",
|
33 |
+
"age": "19",
|
34 |
+
"background": "A young witch discovering her powers while attending a prestigious magical academy",
|
35 |
+
"personality": "Curious, ambitious, sometimes reckless but has a good heart and strong sense of justice",
|
36 |
+
},
|
37 |
+
{
|
38 |
+
"name": "Dr. Victoria Blackthorne",
|
39 |
+
"age": "45",
|
40 |
+
"background": "A renowned archaeologist who specializes in occult artifacts and ancient mysteries",
|
41 |
+
"personality": "Intelligent, sophisticated, perfectionist with a hidden romantic side",
|
42 |
+
},
|
43 |
+
]
|
44 |
+
|
45 |
+
GENRE_OPTIONS = [
|
46 |
+
"Horror - Supernatural terror and psychological thrills",
|
47 |
+
"Detective/Mystery - Crime solving and investigation",
|
48 |
+
"Romance - Love stories and relationship drama",
|
49 |
+
"Fantasy - Magic and mythical creatures",
|
50 |
+
"Sci-Fi - Futuristic technology and space exploration",
|
51 |
+
"Adventure - Action-packed journeys and quests",
|
52 |
+
"Psychological Thriller - Mind games and suspense",
|
53 |
+
"Historical Fiction - Stories set in past eras",
|
54 |
+
]
|
55 |
+
|
56 |
+
|
57 |
+
def load_setting_suggestion(suggestion: str):
|
58 |
+
"""Load a predefined setting suggestion"""
|
59 |
+
return suggestion
|
60 |
+
|
61 |
+
|
62 |
+
def load_character_suggestion(character_name: str):
|
63 |
+
"""Load a predefined character suggestion"""
|
64 |
+
if character_name == "None":
|
65 |
+
return "", "", "", ""
|
66 |
+
|
67 |
+
for char in CHARACTER_SUGGESTIONS:
|
68 |
+
if char["name"] in character_name:
|
69 |
+
return char["name"], char["age"], char["background"], char["personality"]
|
70 |
+
return "", "", "", ""
|
71 |
+
|
72 |
+
|
73 |
+
def save_game_config(
|
74 |
+
setting_desc: str,
|
75 |
+
char_name: str,
|
76 |
+
char_age: str,
|
77 |
+
char_background: str,
|
78 |
+
char_personality: str,
|
79 |
+
genre: str,
|
80 |
+
):
|
81 |
+
"""Save the game configuration to a JSON file"""
|
82 |
+
if not all(
|
83 |
+
[setting_desc, char_name, char_age, char_background, char_personality, genre]
|
84 |
+
):
|
85 |
+
return "❌ Please fill in all fields before saving."
|
86 |
+
|
87 |
+
config = {
|
88 |
+
"id": str(uuid.uuid4()),
|
89 |
+
"setting": {"description": setting_desc},
|
90 |
+
"character": {
|
91 |
+
"name": char_name,
|
92 |
+
"age": char_age,
|
93 |
+
"background": char_background,
|
94 |
+
"personality": char_personality,
|
95 |
+
},
|
96 |
+
"genre": genre,
|
97 |
+
"created_at": str(uuid.uuid4()), # In real app, would use actual timestamp
|
98 |
+
}
|
99 |
+
|
100 |
+
try:
|
101 |
+
filename = f"game_config_{config['id'][:8]}.json"
|
102 |
+
with open(f"generated/{filename}", "w") as f:
|
103 |
+
json.dump(config, f, indent=2)
|
104 |
+
return f"✅ Game configuration saved as {filename}"
|
105 |
+
except Exception as e:
|
106 |
+
return f"❌ Error saving configuration: {str(e)}"
|
107 |
+
|
108 |
+
async def start_game_with_settings(
|
109 |
+
setting_desc: str,
|
110 |
+
char_name: str,
|
111 |
+
char_age: str,
|
112 |
+
char_background: str,
|
113 |
+
char_personality: str,
|
114 |
+
genre: str,
|
115 |
+
):
|
116 |
+
"""Initialize the game with custom settings and switch to game interface"""
|
117 |
+
if not all(
|
118 |
+
[setting_desc, char_name, char_age, char_background, char_personality, genre]
|
119 |
+
):
|
120 |
+
return (
|
121 |
+
gr.update(visible=True), # constructor_interface
|
122 |
+
gr.update(visible=False), # game_interface
|
123 |
+
gr.update(
|
124 |
+
value="❌ Please fill in all fields before starting the game.",
|
125 |
+
visible=True,
|
126 |
+
), # error_message
|
127 |
+
gr.update(),
|
128 |
+
gr.update(),
|
129 |
+
gr.update(), # game components unchanged
|
130 |
+
)
|
131 |
+
|
132 |
+
character = Character(
|
133 |
+
name=char_name,
|
134 |
+
age=char_age,
|
135 |
+
background=char_background,
|
136 |
+
personality=char_personality,
|
137 |
+
)
|
138 |
+
|
139 |
+
game_setting = GameSetting(character=character, setting=setting_desc, genre=genre)
|
140 |
+
|
141 |
+
# Initialize the game story with the custom settings
|
142 |
+
initial_story = f"""Welcome to your story, {game_setting.character.name}!
|
143 |
+
|
144 |
+
Setting: {game_setting.setting}
|
145 |
+
|
146 |
+
You are {game_setting.character.name}, a {game_setting.character.age}-year-old character. {game_setting.character.background}
|
147 |
+
|
148 |
+
Your personality: {game_setting.character.personality}
|
149 |
+
|
150 |
+
Genre: {game_setting.genre}
|
151 |
+
|
152 |
+
You find yourself at the beginning of your adventure. The world around you feels alive with possibilities. What do you choose to do first?
|
153 |
+
|
154 |
+
NOTE FOR THE ASSISTANT: YOU HAVE TO GENERATE THE IMAGE FOR THE START SCENE.
|
155 |
+
"""
|
156 |
+
|
157 |
+
response = await process_user_input(initial_story)
|
158 |
+
|
159 |
+
img = "forest.jpg"
|
160 |
+
if response.change_scene.change_scene:
|
161 |
+
img_path, _ = await generate_image(response.change_scene.scene_description)
|
162 |
+
if img_path:
|
163 |
+
img = img_path
|
164 |
+
|
165 |
+
story["start"] = {
|
166 |
+
"text": response.game_message,
|
167 |
+
"image": img,
|
168 |
+
"choices": [option.option_description for option in response.player_options],
|
169 |
+
"music_tone": response.change_music.music_description,
|
170 |
+
}
|
171 |
+
state["scene"] = "start"
|
172 |
+
|
173 |
+
# Get the current scene data
|
174 |
+
scene_text, scene_image, scene_choices = get_current_scene()
|
175 |
+
|
176 |
+
return (
|
177 |
+
gr.update(visible=False), # constructor_interface
|
178 |
+
gr.update(visible=True), # game_interface
|
179 |
+
gr.update(visible=False), # error_message
|
180 |
+
gr.update(value=scene_text), # game_text
|
181 |
+
gr.update(value=scene_image), # game_image
|
182 |
+
gr.update(choices=scene_choices, value=None), # game_choices
|
183 |
+
)
|
src/game_setting.py
ADDED
@@ -0,0 +1,12 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
from pydantic import BaseModel
|
2 |
+
|
3 |
+
class Character(BaseModel):
|
4 |
+
name: str
|
5 |
+
age: str
|
6 |
+
background: str
|
7 |
+
personality: str
|
8 |
+
|
9 |
+
class GameSetting(BaseModel):
|
10 |
+
character: Character
|
11 |
+
setting: str
|
12 |
+
genre: str
|
src/game_state.py
ADDED
@@ -0,0 +1,15 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
|
2 |
+
story = {
|
3 |
+
"start": {
|
4 |
+
"text": "You wake up in a mysterious forest. What do you do?",
|
5 |
+
"image": "forest.jpg",
|
6 |
+
"choices": ["Explore", "Wait"],
|
7 |
+
"music_tone": "neutral",
|
8 |
+
},
|
9 |
+
}
|
10 |
+
|
11 |
+
state = {"scene": "start"}
|
12 |
+
|
13 |
+
def get_current_scene():
|
14 |
+
scene = story[state["scene"]]
|
15 |
+
return scene["text"], scene["image"], scene["choices"]
|
src/images/image_generator.py
ADDED
@@ -0,0 +1,140 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
from google import genai
|
2 |
+
from google.genai import types
|
3 |
+
import os
|
4 |
+
from PIL import Image
|
5 |
+
from io import BytesIO
|
6 |
+
from datetime import datetime
|
7 |
+
from config import settings
|
8 |
+
import logging
|
9 |
+
|
10 |
+
logger = logging.getLogger(__name__)
|
11 |
+
|
12 |
+
client = genai.Client(api_key=settings.gemini_api_key.get_secret_value())
|
13 |
+
|
14 |
+
async def generate_image(prompt: str) -> tuple[str, str] | None:
|
15 |
+
"""
|
16 |
+
Generate an image using Google's Gemini model and save it to generated/images directory.
|
17 |
+
|
18 |
+
Args:
|
19 |
+
prompt (str): The text prompt to generate the image from
|
20 |
+
|
21 |
+
Returns:
|
22 |
+
str: Path to the generated image file, or None if generation failed
|
23 |
+
"""
|
24 |
+
# Ensure the generated/images directory exists
|
25 |
+
output_dir = "generated/images"
|
26 |
+
os.makedirs(output_dir, exist_ok=True)
|
27 |
+
|
28 |
+
try:
|
29 |
+
response = client.models.generate_content(
|
30 |
+
model="gemini-2.0-flash-preview-image-generation",
|
31 |
+
contents=prompt,
|
32 |
+
config=types.GenerateContentConfig(
|
33 |
+
response_modalities=['TEXT', 'IMAGE']
|
34 |
+
)
|
35 |
+
)
|
36 |
+
|
37 |
+
# Process the response parts
|
38 |
+
image_saved = False
|
39 |
+
for part in response.candidates[0].content.parts:
|
40 |
+
if part.text is not None:
|
41 |
+
logger.info(f"Generated text: {part.text}")
|
42 |
+
elif part.inline_data is not None:
|
43 |
+
# Create a filename with timestamp
|
44 |
+
timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
|
45 |
+
filename = f"gemini_{timestamp}.png"
|
46 |
+
filepath = os.path.join(output_dir, filename)
|
47 |
+
|
48 |
+
# Save the image
|
49 |
+
image = Image.open(BytesIO(part.inline_data.data))
|
50 |
+
image.save(filepath, "PNG")
|
51 |
+
logger.info(f"Image saved to: {filepath}")
|
52 |
+
image_saved = True
|
53 |
+
|
54 |
+
return filepath, part.text
|
55 |
+
|
56 |
+
if not image_saved:
|
57 |
+
logger.error("No image was generated in the response.")
|
58 |
+
return None, None
|
59 |
+
|
60 |
+
except Exception as e:
|
61 |
+
logger.error(f"Error generating image: {e}")
|
62 |
+
return None, None
|
63 |
+
|
64 |
+
|
65 |
+
def modify_image(image_path: str, modification_prompt: str) -> str | None:
|
66 |
+
"""
|
67 |
+
Modify an existing image using Google's Gemini model based on a text prompt.
|
68 |
+
|
69 |
+
Args:
|
70 |
+
image_path (str): Path to the existing image file
|
71 |
+
modification_prompt (str): The text prompt describing how to modify the image
|
72 |
+
|
73 |
+
Returns:
|
74 |
+
str: Path to the modified image file, or None if modification failed
|
75 |
+
"""
|
76 |
+
# Ensure the generated/images directory exists
|
77 |
+
output_dir = "generated/images"
|
78 |
+
os.makedirs(output_dir, exist_ok=True)
|
79 |
+
|
80 |
+
# Check if the input image exists
|
81 |
+
if not os.path.exists(image_path):
|
82 |
+
logger.error(f"Error: Image file not found at {image_path}")
|
83 |
+
return None
|
84 |
+
|
85 |
+
key = settings.gemini_api_key.get_secret_value()
|
86 |
+
|
87 |
+
client = genai.Client(api_key=key)
|
88 |
+
|
89 |
+
try:
|
90 |
+
# Load the input image
|
91 |
+
input_image = Image.open(image_path)
|
92 |
+
|
93 |
+
# Make the API call with both text and image
|
94 |
+
response = client.models.generate_content(
|
95 |
+
model="gemini-2.0-flash-preview-image-generation",
|
96 |
+
contents=[modification_prompt, input_image],
|
97 |
+
config=types.GenerateContentConfig(
|
98 |
+
response_modalities=['TEXT', 'IMAGE']
|
99 |
+
)
|
100 |
+
)
|
101 |
+
|
102 |
+
# Process the response parts
|
103 |
+
image_saved = False
|
104 |
+
for part in response.candidates[0].content.parts:
|
105 |
+
if part.text is not None:
|
106 |
+
logger.info(f"Generated text: {part.text}")
|
107 |
+
elif part.inline_data is not None:
|
108 |
+
# Create a filename with timestamp
|
109 |
+
timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
|
110 |
+
filename = f"gemini_modified_{timestamp}.png"
|
111 |
+
filepath = os.path.join(output_dir, filename)
|
112 |
+
|
113 |
+
# Save the modified image
|
114 |
+
modified_image = Image.open(BytesIO(part.inline_data.data))
|
115 |
+
modified_image.save(filepath, "PNG")
|
116 |
+
logger.info(f"Modified image saved to: {filepath}")
|
117 |
+
image_saved = True
|
118 |
+
|
119 |
+
return filepath, part.text
|
120 |
+
|
121 |
+
if not image_saved:
|
122 |
+
logger.error("No modified image was generated in the response.")
|
123 |
+
return None, None
|
124 |
+
|
125 |
+
except Exception as e:
|
126 |
+
logger.error(f"Error modifying image: {e}")
|
127 |
+
return None, None
|
128 |
+
|
129 |
+
|
130 |
+
if __name__ == "__main__":
|
131 |
+
# Example usage
|
132 |
+
sample_prompt = "A Luke Skywalker half height sprite with white background for visual novel game"
|
133 |
+
generated_image_path = generate_image(sample_prompt)
|
134 |
+
|
135 |
+
# if generated_image_path:
|
136 |
+
# # Example modification
|
137 |
+
# modification_prompt = "Now the house is destroyed, and the jawas are running away"
|
138 |
+
# modified_image_path = modify_image(generated_image_path, modification_prompt)
|
139 |
+
# if modified_image_path:
|
140 |
+
# print(f"Successfully modified image: {modified_image_path}")
|
src/main.py
ADDED
@@ -0,0 +1,320 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
import gradio as gr
|
2 |
+
from css import custom_css
|
3 |
+
from audio.audio_generator import (
|
4 |
+
update_audio,
|
5 |
+
start_music_generation,
|
6 |
+
change_music_tone,
|
7 |
+
cleanup_music_session,
|
8 |
+
)
|
9 |
+
import logging
|
10 |
+
from agent.llm_agent import process_user_input
|
11 |
+
from images.image_generator import generate_image
|
12 |
+
import uuid
|
13 |
+
from game_state import story, state
|
14 |
+
from game_constructor import (
|
15 |
+
SETTING_SUGGESTIONS,
|
16 |
+
CHARACTER_SUGGESTIONS,
|
17 |
+
GENRE_OPTIONS,
|
18 |
+
load_setting_suggestion,
|
19 |
+
load_character_suggestion,
|
20 |
+
start_game_with_settings,
|
21 |
+
)
|
22 |
+
import asyncio
|
23 |
+
|
24 |
+
logger = logging.getLogger(__name__)
|
25 |
+
|
26 |
+
|
27 |
+
def return_to_constructor():
|
28 |
+
"""Return to the game constructor interface"""
|
29 |
+
return (
|
30 |
+
gr.update(visible=True), # constructor_interface
|
31 |
+
gr.update(visible=False), # game_interface
|
32 |
+
gr.update(visible=False), # error_message
|
33 |
+
)
|
34 |
+
|
35 |
+
|
36 |
+
async def update_scene(request: gr.Request, choice):
|
37 |
+
logger.info(f"Updating scene with choice: {choice}")
|
38 |
+
if isinstance(choice, str):
|
39 |
+
old_scene = state["scene"]
|
40 |
+
new_scene = str(uuid.uuid4())
|
41 |
+
story[new_scene] = {
|
42 |
+
**story[old_scene],
|
43 |
+
}
|
44 |
+
state["scene"] = new_scene
|
45 |
+
|
46 |
+
user_story = f"""Current scene description:
|
47 |
+
{story[old_scene]["text"]}
|
48 |
+
User's choice: {choice}
|
49 |
+
"""
|
50 |
+
|
51 |
+
response = await process_user_input(user_story)
|
52 |
+
|
53 |
+
story[new_scene]["text"] = response.game_message
|
54 |
+
|
55 |
+
story[new_scene]["choices"] = [
|
56 |
+
option.option_description for option in response.player_options
|
57 |
+
]
|
58 |
+
|
59 |
+
if response.change_scene.change_scene:
|
60 |
+
img_path, _ = await generate_image(response.change_scene.scene_description)
|
61 |
+
if img_path:
|
62 |
+
story[new_scene]["image"] = img_path
|
63 |
+
|
64 |
+
if response.change_music.change_music:
|
65 |
+
await change_music_tone(request, response.change_music.music_description)
|
66 |
+
|
67 |
+
scene = story[state["scene"]]
|
68 |
+
return (
|
69 |
+
scene["text"],
|
70 |
+
scene["image"],
|
71 |
+
gr.Radio(
|
72 |
+
choices=scene["choices"],
|
73 |
+
label="What do you choose?",
|
74 |
+
value=None,
|
75 |
+
elem_classes=["choice-buttons"],
|
76 |
+
),
|
77 |
+
)
|
78 |
+
|
79 |
+
|
80 |
+
def update_preview(setting, name, age, background, personality, genre):
|
81 |
+
"""Update the configuration preview"""
|
82 |
+
if not any([setting, name, age, background, personality]):
|
83 |
+
return "Fill in the fields to see a preview..."
|
84 |
+
|
85 |
+
preview = f"""🌍 SETTING: {setting[:100]}{"..." if len(setting) > 100 else ""}
|
86 |
+
|
87 |
+
👤 CHARACTER: {name} (Age: {age})
|
88 |
+
📖 Background: {background}
|
89 |
+
💭 Personality: {personality}
|
90 |
+
|
91 |
+
🎭 GENRE: {genre}"""
|
92 |
+
return preview
|
93 |
+
|
94 |
+
|
95 |
+
async def start_game_with_music(
|
96 |
+
request: gr.Request,
|
97 |
+
setting_desc: str,
|
98 |
+
char_name: str,
|
99 |
+
char_age: str,
|
100 |
+
char_background: str,
|
101 |
+
char_personality: str,
|
102 |
+
genre: str,
|
103 |
+
):
|
104 |
+
"""Start the game with custom settings and initialize music"""
|
105 |
+
# First, get the game interface updates
|
106 |
+
result = await start_game_with_settings(
|
107 |
+
setting_desc, char_name, char_age, char_background, char_personality, genre
|
108 |
+
)
|
109 |
+
# If game is starting successfully (interface is switching), start music
|
110 |
+
if len(result) >= 2 and result[1].get(
|
111 |
+
"visible", False
|
112 |
+
): # game_interface becoming visible
|
113 |
+
asyncio.create_task(start_music_generation(request))
|
114 |
+
|
115 |
+
if state["scene"] in story and "music_tone" in story[state["scene"]]:
|
116 |
+
await change_music_tone(request, story[state["scene"]]["music_tone"])
|
117 |
+
|
118 |
+
return result
|
119 |
+
|
120 |
+
with gr.Blocks(
|
121 |
+
theme="soft", title="Game Constructor & Visual Novel", css=custom_css
|
122 |
+
) as demo:
|
123 |
+
# Constructor Interface (visible by default)
|
124 |
+
with gr.Column(visible=True, elem_id="constructor-interface") as constructor_interface:
|
125 |
+
gr.Markdown("# 🎮 Interactive Game Constructor")
|
126 |
+
gr.Markdown(
|
127 |
+
"Create your own interactive story game by defining the setting, character, and genre!"
|
128 |
+
)
|
129 |
+
|
130 |
+
# Error message area
|
131 |
+
error_message = gr.Textbox(
|
132 |
+
label="⚠️ Error",
|
133 |
+
visible=False,
|
134 |
+
interactive=False,
|
135 |
+
elem_classes=["error-message"],
|
136 |
+
)
|
137 |
+
|
138 |
+
with gr.Row():
|
139 |
+
with gr.Column(scale=2):
|
140 |
+
# Setting Description Section
|
141 |
+
with gr.Group():
|
142 |
+
gr.Markdown("## 🌍 Setting Description")
|
143 |
+
setting_suggestions = gr.Dropdown(
|
144 |
+
choices=["Select a suggestion..."] + SETTING_SUGGESTIONS,
|
145 |
+
label="Quick Suggestions",
|
146 |
+
value="Select a suggestion...",
|
147 |
+
interactive=True,
|
148 |
+
)
|
149 |
+
setting_description = gr.Textbox(
|
150 |
+
label="Describe your game setting",
|
151 |
+
placeholder="Enter a detailed description of where your story takes place...",
|
152 |
+
lines=4,
|
153 |
+
max_lines=6,
|
154 |
+
)
|
155 |
+
|
156 |
+
# Character Description Section
|
157 |
+
with gr.Group():
|
158 |
+
gr.Markdown("## 👤 Character Description")
|
159 |
+
character_suggestions = gr.Dropdown(
|
160 |
+
choices=["None"]
|
161 |
+
+ [
|
162 |
+
f"{char['name']} - {char['background'][:50]}..."
|
163 |
+
for char in CHARACTER_SUGGESTIONS
|
164 |
+
],
|
165 |
+
label="Character Templates",
|
166 |
+
value="None",
|
167 |
+
interactive=True,
|
168 |
+
)
|
169 |
+
|
170 |
+
with gr.Row():
|
171 |
+
char_name = gr.Textbox(
|
172 |
+
label="Character Name",
|
173 |
+
placeholder="Enter character name...",
|
174 |
+
)
|
175 |
+
char_age = gr.Textbox(label="Age", placeholder="25")
|
176 |
+
|
177 |
+
char_background = gr.Textbox(
|
178 |
+
label="Background/Profession",
|
179 |
+
placeholder="Describe your character's background, profession, or role...",
|
180 |
+
lines=2,
|
181 |
+
)
|
182 |
+
char_personality = gr.Textbox(
|
183 |
+
label="Personality & Traits",
|
184 |
+
placeholder="Describe personality, quirks, motivations, fears...",
|
185 |
+
lines=2,
|
186 |
+
)
|
187 |
+
|
188 |
+
# Genre Selection Section
|
189 |
+
with gr.Group():
|
190 |
+
gr.Markdown("## 🎭 Genre & Style")
|
191 |
+
genre_selection = gr.Dropdown(
|
192 |
+
choices=GENRE_OPTIONS,
|
193 |
+
label="Choose your story genre",
|
194 |
+
value=GENRE_OPTIONS[0],
|
195 |
+
interactive=True,
|
196 |
+
)
|
197 |
+
|
198 |
+
with gr.Column(scale=1):
|
199 |
+
# Preview Section
|
200 |
+
with gr.Group():
|
201 |
+
gr.Markdown("## 📋 Configuration Preview")
|
202 |
+
preview_box = gr.Textbox(
|
203 |
+
label="Game Summary",
|
204 |
+
lines=8,
|
205 |
+
interactive=False,
|
206 |
+
placeholder="Fill in the fields to see a preview...",
|
207 |
+
)
|
208 |
+
|
209 |
+
with gr.Group():
|
210 |
+
gr.Markdown("## 🎮 Ready to Play?")
|
211 |
+
start_btn = gr.Button("▶️ Start Game", variant="primary", size="lg")
|
212 |
+
|
213 |
+
with gr.Column(visible=False, elem_id="game-interface") as game_interface:
|
214 |
+
gr.Markdown("# 🎮 Your Interactive Story")
|
215 |
+
|
216 |
+
with gr.Row():
|
217 |
+
back_btn = gr.Button("⬅️ Back to Constructor", variant="secondary")
|
218 |
+
gr.Markdown("### Playing your custom game!")
|
219 |
+
|
220 |
+
# Audio component for background music
|
221 |
+
audio_out = gr.Audio(
|
222 |
+
autoplay=True, streaming=True, interactive=False, visible=False
|
223 |
+
)
|
224 |
+
|
225 |
+
# Background image (fullscreen)
|
226 |
+
with gr.Column(elem_classes=["image-container"]):
|
227 |
+
game_image = gr.Image(type="filepath", interactive=False, show_label=False)
|
228 |
+
|
229 |
+
# Overlay content (text and buttons)
|
230 |
+
with gr.Column(elem_classes=["overlay-content"]):
|
231 |
+
game_text = gr.Textbox(
|
232 |
+
label="",
|
233 |
+
interactive=False,
|
234 |
+
show_label=False,
|
235 |
+
elem_classes=["narrative-text"],
|
236 |
+
lines=3,
|
237 |
+
)
|
238 |
+
game_choices = gr.Radio(
|
239 |
+
choices=[],
|
240 |
+
label="What do you choose?",
|
241 |
+
value=None,
|
242 |
+
elem_classes=["choice-buttons"],
|
243 |
+
)
|
244 |
+
|
245 |
+
# Event handlers for constructor interface
|
246 |
+
setting_suggestions.change(
|
247 |
+
fn=load_setting_suggestion,
|
248 |
+
inputs=[setting_suggestions],
|
249 |
+
outputs=[setting_description],
|
250 |
+
)
|
251 |
+
|
252 |
+
character_suggestions.change(
|
253 |
+
fn=load_character_suggestion,
|
254 |
+
inputs=[character_suggestions],
|
255 |
+
outputs=[char_name, char_age, char_background, char_personality],
|
256 |
+
)
|
257 |
+
|
258 |
+
# Update preview when any field changes
|
259 |
+
for component in [
|
260 |
+
setting_description,
|
261 |
+
char_name,
|
262 |
+
char_age,
|
263 |
+
char_background,
|
264 |
+
char_personality,
|
265 |
+
genre_selection,
|
266 |
+
]:
|
267 |
+
component.change(
|
268 |
+
fn=update_preview,
|
269 |
+
inputs=[
|
270 |
+
setting_description,
|
271 |
+
char_name,
|
272 |
+
char_age,
|
273 |
+
char_background,
|
274 |
+
char_personality,
|
275 |
+
genre_selection,
|
276 |
+
],
|
277 |
+
outputs=[preview_box],
|
278 |
+
)
|
279 |
+
|
280 |
+
# Interface switching handlers
|
281 |
+
start_btn.click(
|
282 |
+
fn=start_game_with_music,
|
283 |
+
inputs=[
|
284 |
+
setting_description,
|
285 |
+
char_name,
|
286 |
+
char_age,
|
287 |
+
char_background,
|
288 |
+
char_personality,
|
289 |
+
genre_selection,
|
290 |
+
],
|
291 |
+
outputs=[
|
292 |
+
constructor_interface,
|
293 |
+
game_interface,
|
294 |
+
error_message,
|
295 |
+
game_text,
|
296 |
+
game_image,
|
297 |
+
game_choices,
|
298 |
+
],
|
299 |
+
)
|
300 |
+
|
301 |
+
back_btn.click(
|
302 |
+
fn=return_to_constructor,
|
303 |
+
inputs=[],
|
304 |
+
outputs=[constructor_interface, game_interface, error_message],
|
305 |
+
)
|
306 |
+
|
307 |
+
game_choices.change(
|
308 |
+
fn=update_scene,
|
309 |
+
inputs=[game_choices],
|
310 |
+
outputs=[game_text, game_image, game_choices],
|
311 |
+
)
|
312 |
+
|
313 |
+
demo.unload(cleanup_music_session)
|
314 |
+
demo.load(
|
315 |
+
fn=update_audio,
|
316 |
+
inputs=None,
|
317 |
+
outputs=[audio_out],
|
318 |
+
)
|
319 |
+
|
320 |
+
demo.launch()
|