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(
|