Update app.py

app.py

@@ -20,10 +20,14 @@ game_state = {
     "story_progress": "Начало игры..."  # Общая история
 }
 
-def format_prompt(message, history, genre, player_id):
-    """Форматирование промпта для генерации текста."""
+def format_prompt(message, history, genre):
+    # Получаем промпт для выбранного жанра
     genre_prompt = GENRE_PROMPTS.get(genre, "Жанр игры неизвестен.")
-    prompt = f"Игрок {player_id} в жанре {genre}. {genre_prompt} История: {game_state['story_progress']}. "
+    prompt = f"Ты — персонаж в текстовой игре, тебя зовут Рассказчик, взаимодействующий с игроком на русском языке. " \
+             f"{genre_prompt} Каждый твой ответ должен продолжать сюжет игры, принимая во внимание прошлые действия игрока и текущую ситуацию. " \
+             f"Твой стиль общения должен быть интересным и поддерживать атмосферу игры. Ты должен поддерживать динамичность сюжета и постоянно добавлять элементы интерактивности и новых вызовов. " \
+             f"Главный герой может погибнуть в зависимости от выбора. Счастливый конец возможен только в 10% случаев. " \
+             f"У игрока должно быть три - четыре варианта действий, прономеруй их. "
     for user_prompt, bot_response in history:
         prompt += f"[INST] {user_prompt} [/INST] {bot_response}</s> "
     prompt += f"[INST] {message} [/INST]"
@@ -35,7 +39,7 @@ def generate(player_id, message, genre, temperature=0.9, max_new_tokens=256, top
         game_state["players"][player_id] = {"history": []}  # Создать историю для нового игрока
 
     history = game_state["players"][player_id]["history"]
-    formatted_prompt = format_prompt(message, history, genre, player_id)
+    formatted_prompt = format_prompt(message, history, genre)
 
     generate_kwargs = {
         "temperature": temperature,