Update app.py
Browse files
app.py
CHANGED
|
@@ -17,6 +17,41 @@ if 'OPENAI_API_KEY' not in os.environ:
|
|
| 17 |
pxt.drop_dir('ai_rpg', force=True)
|
| 18 |
pxt.create_dir('ai_rpg')
|
| 19 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 20 |
@pxt.udf
|
| 21 |
def generate_messages(genre: str, player_name: str, initial_scenario: str, player_input: str, turn_number: int, stats: str) -> list[dict]:
|
| 22 |
return [
|
|
@@ -100,41 +135,6 @@ def get_options(response: str) -> list[str]:
|
|
| 100 |
|
| 101 |
return ["๊ณ์ํ๊ธฐ...", "๋ค๋ฅธ ํ๋ ์ทจํ๊ธฐ", "๋ญ๊ฐ ์๋ก์ด ์๋ํ๊ธฐ", "์ฃผ๋ณ ํ์ํ๊ธฐ"]
|
| 102 |
|
| 103 |
-
@pxt.udf
|
| 104 |
-
def initialize_stats(genre: str) -> str:
|
| 105 |
-
"""Initialize player stats based on the selected genre"""
|
| 106 |
-
base_stats = {
|
| 107 |
-
"๐งโโ๏ธ Fantasy": "์ฒด๋ ฅ: 100, ๋ง๋: 80, ํ: 7, ์ง๋ฅ: 8, ๋ฏผ์ฒฉ: 6, ์์ง๊ธ: 50๊ณจ๋",
|
| 108 |
-
"๐ Sci-Fi": "์ฒด๋ ฅ: 100, ์๋์ง: 90, ๊ธฐ์ ๋ ฅ: 8, ์ง๋ฅ: 9, ๋ฏผ์ฒฉ: 6, ํฌ๋ ๋ง: 500",
|
| 109 |
-
"๐ป Horror": "์ฒด๋ ฅ: 80, ์ ์ ๋ ฅ: 100, ํ: 6, ์ง๋ฅ: 7, ๋ฏผ์ฒฉ: 8, ์์งํ: ์์ ๋ฑ, ๊ธฐ๋ณธ ์ฝํ",
|
| 110 |
-
"๐ Mystery": "์ฒด๋ ฅ: 90, ์ง์ค๋ ฅ: 100, ๊ด์ฐฐ๋ ฅ: 9, ์ง๋ฅ: 8, ์นด๋ฆฌ์ค๋ง: 7, ๋จ์: 0",
|
| 111 |
-
"๐ Post-Apocalyptic": "์ฒด๋ ฅ: 95, ๋ฐฉ์ฌ๋ฅ ์ ํญ: 75, ํ: 8, ์์กด๋ ฅ: 9, ๋ฌผ์: ์ ํ๋จ",
|
| 112 |
-
"๐ค Cyberpunk": "์ฒด๋ ฅ: 90, ์ฌ์ด๋ฒ์จ์ด: 85%, ํดํน: 8, ๊ฑฐ๋ฆฌ ์ ์ฉ๋: 6, ์ฃ์ง: 7, ๋์: 1000",
|
| 113 |
-
"โ๏ธ Steampunk": "์ฒด๋ ฅ: 95, ์ฆ๊ธฐ๋ ฅ: 85, ๊ธฐ๊ณ๊ณตํ: 8, ์์ ์ฑ: 7, ์ฌ๊ต์ฑ: 6, ์ค๋ง: 200"
|
| 114 |
-
}
|
| 115 |
-
|
| 116 |
-
if genre in base_stats:
|
| 117 |
-
return base_stats[genre]
|
| 118 |
-
else:
|
| 119 |
-
# Default stats if genre not found
|
| 120 |
-
return "์ฒด๋ ฅ: 100, ์๋์ง: 100, ํ: 7, ์ง๋ฅ: 7, ๋ฏผ์ฒฉ: 7, ์์ง๊ธ: 100"
|
| 121 |
-
|
| 122 |
-
@pxt.udf
|
| 123 |
-
def generate_random_event(turn_number: int) -> str:
|
| 124 |
-
"""Generate a random event based on turn number"""
|
| 125 |
-
if turn_number % 3 == 0 and turn_number > 0: # Every 3rd turn
|
| 126 |
-
events = [
|
| 127 |
-
"๊ฐ์๊ธฐ ๋ถ๊ทผ์์ ์ด์ํ ์๋ฆฌ๊ฐ ๋ค๋ฆฝ๋๋ค",
|
| 128 |
-
"๋ฏ์ ์ฌํ์๊ฐ ๋น์ ์ ๋ฐ๋ผ๋ณด๊ณ ์์ต๋๋ค",
|
| 129 |
-
"์ง๋ฉด์ด ๋ฏธ์ธํ๊ฒ ์ง๋ํ๊ธฐ ์์ํฉ๋๋ค",
|
| 130 |
-
"์ฃผ๋จธ๋์์ ๋ฌด์ธ๊ฐ๊ฐ ๋น๋ฉ๋๋ค",
|
| 131 |
-
"๋ฉ๋ฆฌ์ ๋ฌด์ธ๊ฐ๊ฐ ๋น์ ์ ํฅํด ๋ค๊ฐ์ค๊ณ ์์ต๋๋ค",
|
| 132 |
-
"๊ฐ์๊ธฐ ๋ ์จ๊ฐ ๋ณํ๊ธฐ ์์ํฉ๋๋ค",
|
| 133 |
-
"์ฃผ๋ณ์ ์จ๊ฒจ์ง ํต๋ก๋ฅผ ๋ฐ๊ฒฌํฉ๋๋ค"
|
| 134 |
-
]
|
| 135 |
-
return random.choice(events)
|
| 136 |
-
return ""
|
| 137 |
-
|
| 138 |
# Create a single table for all game data
|
| 139 |
interactions = pxt.create_table(
|
| 140 |
'ai_rpg.interactions',
|
|
@@ -184,8 +184,9 @@ class RPGGame:
|
|
| 184 |
self.current_session_id = session_id
|
| 185 |
self.turn_number = 0
|
| 186 |
|
| 187 |
-
#
|
| 188 |
-
|
|
|
|
| 189 |
|
| 190 |
interactions.insert([{
|
| 191 |
'session_id': session_id,
|
|
@@ -195,7 +196,7 @@ class RPGGame:
|
|
| 195 |
'turn_number': 0,
|
| 196 |
'player_input': "Game starts",
|
| 197 |
'timestamp': datetime.now(),
|
| 198 |
-
'player_stats':
|
| 199 |
'random_event': ""
|
| 200 |
}])
|
| 201 |
|
|
@@ -227,10 +228,23 @@ class RPGGame:
|
|
| 227 |
).collect()
|
| 228 |
|
| 229 |
self.current_stats = prev_turn['player_stats'][0]
|
| 230 |
-
random_event = generate_random_event(self.turn_number)
|
| 231 |
|
| 232 |
-
|
| 233 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 234 |
|
| 235 |
interactions.insert([{
|
| 236 |
'session_id': self.current_session_id,
|
|
@@ -241,7 +255,7 @@ class RPGGame:
|
|
| 241 |
'player_input': action,
|
| 242 |
'timestamp': datetime.now(),
|
| 243 |
'player_stats': self.current_stats,
|
| 244 |
-
'random_event':
|
| 245 |
}])
|
| 246 |
|
| 247 |
result = interactions.select(
|
|
|
|
| 17 |
pxt.drop_dir('ai_rpg', force=True)
|
| 18 |
pxt.create_dir('ai_rpg')
|
| 19 |
|
| 20 |
+
# ์ผ๋ฐ ํจ์๋ก ๋ณ๊ฒฝ (UDF๊ฐ ์๋)
|
| 21 |
+
def initialize_stats(genre: str) -> str:
|
| 22 |
+
"""Initialize player stats based on the selected genre"""
|
| 23 |
+
base_stats = {
|
| 24 |
+
"๐งโโ๏ธ Fantasy": "์ฒด๋ ฅ: 100, ๋ง๋: 80, ํ: 7, ์ง๋ฅ: 8, ๋ฏผ์ฒฉ: 6, ์์ง๊ธ: 50๊ณจ๋",
|
| 25 |
+
"๐ Sci-Fi": "์ฒด๋ ฅ: 100, ์๋์ง: 90, ๊ธฐ์ ๋ ฅ: 8, ์ง๋ฅ: 9, ๋ฏผ์ฒฉ: 6, ํฌ๋ ๋ง: 500",
|
| 26 |
+
"๐ป Horror": "์ฒด๋ ฅ: 80, ์ ์ ๋ ฅ: 100, ํ: 6, ์ง๋ฅ: 7, ๋ฏผ์ฒฉ: 8, ์์งํ: ์์ ๋ฑ, ๊ธฐ๋ณธ ์ฝํ",
|
| 27 |
+
"๐ Mystery": "์ฒด๋ ฅ: 90, ์ง์ค๋ ฅ: 100, ๊ด์ฐฐ๋ ฅ: 9, ์ง๋ฅ: 8, ์นด๋ฆฌ์ค๋ง: 7, ๋จ์: 0",
|
| 28 |
+
"๐ Post-Apocalyptic": "์ฒด๋ ฅ: 95, ๋ฐฉ์ฌ๋ฅ ์ ํญ: 75, ํ: 8, ์์กด๋ ฅ: 9, ๋ฌผ์: ์ ํ๋จ",
|
| 29 |
+
"๐ค Cyberpunk": "์ฒด๋ ฅ: 90, ์ฌ์ด๋ฒ์จ์ด: 85%, ํดํน: 8, ๊ฑฐ๋ฆฌ ์ ์ฉ๋: 6, ์ฃ์ง: 7, ๋์: 1000",
|
| 30 |
+
"โ๏ธ Steampunk": "์ฒด๋ ฅ: 95, ์ฆ๊ธฐ๋ ฅ: 85, ๊ธฐ๊ณ๊ณตํ: 8, ์์ ์ฑ: 7, ์ฌ๊ต์ฑ: 6, ์ค๋ง: 200"
|
| 31 |
+
}
|
| 32 |
+
|
| 33 |
+
if genre in base_stats:
|
| 34 |
+
return base_stats[genre]
|
| 35 |
+
else:
|
| 36 |
+
# Default stats if genre not found
|
| 37 |
+
return "์ฒด๋ ฅ: 100, ์๋์ง: 100, ํ: 7, ์ง๋ฅ: 7, ๋ฏผ์ฒฉ: 7, ์์ง๊ธ: 100"
|
| 38 |
+
|
| 39 |
+
@pxt.udf
|
| 40 |
+
def generate_random_event(turn_number: int) -> str:
|
| 41 |
+
"""Generate a random event based on turn number"""
|
| 42 |
+
if turn_number % 3 == 0 and turn_number > 0: # Every 3rd turn
|
| 43 |
+
events = [
|
| 44 |
+
"๊ฐ์๊ธฐ ๋ถ๊ทผ์์ ์ด์ํ ์๋ฆฌ๊ฐ ๋ค๋ฆฝ๋๋ค",
|
| 45 |
+
"๋ฏ์ ์ฌํ์๊ฐ ๋น์ ์ ๋ฐ๋ผ๋ณด๊ณ ์์ต๋๋ค",
|
| 46 |
+
"์ง๋ฉด์ด ๋ฏธ์ธํ๊ฒ ์ง๋ํ๊ธฐ ์์ํฉ๋๋ค",
|
| 47 |
+
"์ฃผ๋จธ๋์์ ๋ฌด์ธ๊ฐ๊ฐ ๋น๋ฉ๋๋ค",
|
| 48 |
+
"๋ฉ๋ฆฌ์ ๋ฌด์ธ๊ฐ๊ฐ ๋น์ ์ ํฅํด ๋ค๊ฐ์ค๊ณ ์์ต๋๋ค",
|
| 49 |
+
"๊ฐ์๊ธฐ ๋ ์จ๊ฐ ๋ณํ๊ธฐ ์์ํฉ๋๋ค",
|
| 50 |
+
"์ฃผ๋ณ์ ์จ๊ฒจ์ง ํต๋ก๋ฅผ ๋ฐ๊ฒฌํฉ๋๋ค"
|
| 51 |
+
]
|
| 52 |
+
return random.choice(events)
|
| 53 |
+
return ""
|
| 54 |
+
|
| 55 |
@pxt.udf
|
| 56 |
def generate_messages(genre: str, player_name: str, initial_scenario: str, player_input: str, turn_number: int, stats: str) -> list[dict]:
|
| 57 |
return [
|
|
|
|
| 135 |
|
| 136 |
return ["๊ณ์ํ๊ธฐ...", "๋ค๋ฅธ ํ๋ ์ทจํ๊ธฐ", "๋ญ๊ฐ ์๋ก์ด ์๋ํ๊ธฐ", "์ฃผ๋ณ ํ์ํ๊ธฐ"]
|
| 137 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 138 |
# Create a single table for all game data
|
| 139 |
interactions = pxt.create_table(
|
| 140 |
'ai_rpg.interactions',
|
|
|
|
| 184 |
self.current_session_id = session_id
|
| 185 |
self.turn_number = 0
|
| 186 |
|
| 187 |
+
# ํจ์๋ฅผ ์ง์ ํธ์ถํ์ฌ ๊ฒฐ๊ณผ๋ฅผ ์ ์ฅ
|
| 188 |
+
initial_stats = initialize_stats(genre)
|
| 189 |
+
self.current_stats = initial_stats
|
| 190 |
|
| 191 |
interactions.insert([{
|
| 192 |
'session_id': session_id,
|
|
|
|
| 196 |
'turn_number': 0,
|
| 197 |
'player_input': "Game starts",
|
| 198 |
'timestamp': datetime.now(),
|
| 199 |
+
'player_stats': initial_stats, # ๋ฌธ์์ด ๊ฒฐ๊ณผ ์ ์ฅ
|
| 200 |
'random_event': ""
|
| 201 |
}])
|
| 202 |
|
|
|
|
| 228 |
).collect()
|
| 229 |
|
| 230 |
self.current_stats = prev_turn['player_stats'][0]
|
|
|
|
| 231 |
|
| 232 |
+
# ์ผ๋ฐ ํจ์๋ก ๋ณํํ์ผ๋ฏ๋ก ๊ทธ๋ฅ ํธ์ถํด์ ์ฌ์ฉ
|
| 233 |
+
random_event_val = ""
|
| 234 |
+
if self.turn_number % 3 == 0 and self.turn_number > 0:
|
| 235 |
+
events = [
|
| 236 |
+
"๊ฐ์๊ธฐ ๋ถ๊ทผ์์ ์ด์ํ ์๋ฆฌ๊ฐ ๋ค๋ฆฝ๋๋ค",
|
| 237 |
+
"๋ฏ์ ์ฌํ์๊ฐ ๋น์ ์ ๋ฐ๋ผ๋ณด๊ณ ์์ต๋๋ค",
|
| 238 |
+
"์ง๋ฉด์ด ๋ฏธ์ธํ๊ฒ ์ง๋ํ๊ธฐ ์์ํฉ๋๋ค",
|
| 239 |
+
"์ฃผ๋จธ๋์์ ๋ฌด์ธ๊ฐ๊ฐ ๋น๋ฉ๋๋ค",
|
| 240 |
+
"๋ฉ๋ฆฌ์ ๋ฌด์ธ๊ฐ๊ฐ ๋น์ ์ ํฅํด ๋ค๊ฐ์ค๊ณ ์์ต๋๋ค",
|
| 241 |
+
"๊ฐ์๊ธฐ ๋ ์จ๊ฐ ๋ณํ๊ธฐ ์์ํฉ๋๋ค",
|
| 242 |
+
"์ฃผ๋ณ์ ์จ๊ฒจ์ง ํต๋ก๋ฅผ ๋ฐ๊ฒฌํฉ๋๋ค"
|
| 243 |
+
]
|
| 244 |
+
random_event_val = random.choice(events)
|
| 245 |
+
|
| 246 |
+
if random_event_val:
|
| 247 |
+
action = f"{action} ({random_event_val})"
|
| 248 |
|
| 249 |
interactions.insert([{
|
| 250 |
'session_id': self.current_session_id,
|
|
|
|
| 255 |
'player_input': action,
|
| 256 |
'timestamp': datetime.now(),
|
| 257 |
'player_stats': self.current_stats,
|
| 258 |
+
'random_event': random_event_val
|
| 259 |
}])
|
| 260 |
|
| 261 |
result = interactions.select(
|