Update app.py

app.py

@@ -5,17 +5,21 @@ client = InferenceClient(
     "mistralai/Mistral-7B-Instruct-v0.3"
 )
 
-
-def format_prompt(message, history):
-  prompt = "Ты — персонаж в текстовой игре, тебя зовут Рассказчик, взаимодействующий с игроком на русском языке. Каждый твой ответ должен продолжать сюжет игры, принимая во внимание прошлые действия игрока и текущую ситуацию. Твой стиль общения должен быть интересным и поддерживать атмосферу игры. Ты должен поддерживать динамичность сюжета и постоянно добавлять элементы интерактивности и новых вызовов. В начале игры игроку предоставляется информация о мире, его текущем положении и доступных действиях. Он может выбирать из предложенных вариантов или описывать свои действия свободно. Важно, чтобы ты адаптировал ответы в зависимости от решений игрока и развивал сюжет на основе их выбора. Каждый ответ должен быть логичным продолжением событий, с описанием последствий выбранных действий."
-  for user_prompt, bot_response in history:
-    prompt += f"[INST] {user_prompt} [/INST]"
-    prompt += f" {bot_response}</s> "
-  prompt += f"[INST] {message} [/INST]"
-  return prompt
+def format_prompt(message, history, genre):
+    prompt = f"Ты — персонаж в текстовой игре, тебя зовут Рассказчик, взаимодействующий с игроком на русском языке. " \
+             f"Ты находишься в жанре игры: {genre}. Каждый твой ответ должен продолжать сюжет игры, принимая во внимание прошлые действия игрока и текущую ситуацию. " \
+             f"Твой стиль общения должен быть интересным и поддерживать атмосферу игры. Ты должен поддерживать динамичность сюжета и постоянно добавлять элементы интерактивности и новых вызовов. " \
+             f"В начале игры игроку предоставляется информация о мире, его текущем положении и доступных действиях. Он может выбирать из предложенных вариантов или описывать свои действия свободно. " \
+             f"Важно, чтобы ты адаптировал ответы в зависимости от решений игрока и развивал сюжет на основе их выбора. Каждый ответ должен быть логичным продолжением событий, с описанием последствий выбранных действий."
+    
+    for user_prompt, bot_response in history:
+        prompt += f"[INST] {user_prompt} [/INST]"
+        prompt += f" {bot_response}</s> "
+    prompt += f"[INST] {message} [/INST]"
+    return prompt
 
 def generate(
-    prompt, history, system_prompt, temperature=0.9, max_new_tokens=256, top_p=0.95, repetition_penalty=1.0,
+    prompt, history, system_prompt, genre, temperature=0.9, max_new_tokens=256, top_p=0.95, repetition_penalty=1.0,
 ):
     temperature = float(temperature)
     if temperature < 1e-2:
@@ -31,7 +35,7 @@ def generate(
         seed=42,
     )
 
-    formatted_prompt = format_prompt(f"{system_prompt}, {prompt}", history)
+    formatted_prompt = format_prompt(f"{system_prompt}, {prompt}", history, genre)
     stream = client.text_generation(formatted_prompt, **generate_kwargs, stream=True, details=True, return_full_text=True)
     output = ""
 
@@ -40,13 +44,19 @@ def generate(
         yield output
     return output
 
-
-additional_inputs=[
+additional_inputs = [
     gr.Textbox(
         label="System Prompt",
         max_lines=1,
         interactive=True,
     ),
+    gr.Radio(
+        label="Game Genre",
+        choices=["Fantasy", "Sci-Fi", "Horror", "Adventure", "Mystery"],
+        value="Fantasy",  # Default genre
+        interactive=True,
+        info="Select the genre of the game",
+    ),
     gr.Slider(
         label="Temperature",
         value=0.1,
@@ -85,8 +95,7 @@ additional_inputs=[
     )
 ]
 
-examples=[["Подробнее"],["Варианты"],["Хоррор"],
-         ]
+examples = [["Подробнее"], ["Варианты"]]
 
 gr.ChatInterface(
     fn=generate,