Update app.py

app.py

@@ -3,56 +3,57 @@ import gradio as gr
 
 client = InferenceClient("mistralai/Mistral-7B-Instruct-v0.3")
 
-# Шаблоны промптов для каждого жанра
-GENRE_PROMPTS = {
-    "Investigation": "Ты находишься в жанре игры Investigation...",
-    # Остальные жанры остаются неизменными
+# Хранение состояния игры
+game_state = {
+    "players": {},
+    "story_progress": "Начало игры..."
 }
 
-def format_prompt(message, history, genre, players):
+def format_prompt(message, history, genre, player_id):
+    # Получаем промпт для выбранного жанра
     genre_prompt = GENRE_PROMPTS.get(genre, "Жанр игры неизвестен.")
-    prompt = f"Ты — персонаж в текстовой игре, тебя зовут Рассказчик, взаимодействующий с игроками на русском языке. " \
-             f"{genre_prompt} Учитывай, что в игре участвуют несколько игроков: {', '.join(players)}. " \
-             f"Каждый твой ответ должен учитывать действия всех игроков, их взаимодействие и текущую ситуацию. " \
-             f"Добавляй элементы кооперации: совместное решение задач, голосование за действия и влияние выборов одного игрока на другого."
+    prompt = f"Игрок {player_id} в жанре {genre}. {genre_prompt} История: {game_state['story_progress']}."
     for user_prompt, bot_response in history:
-        prompt += f"[INST] {user_prompt} [/INST]"
-        prompt += f" {bot_response}</s> "
+        prompt += f"[INST] {user_prompt} [/INST] {bot_response}</s> "
     prompt += f"[INST] {message} [/INST]"
     return prompt
 
-def generate(
-    prompt, history, system_prompt, genre, players, temperature=0.9, max_new_tokens=256, top_p=0.95, repetition_penalty=1.0,
-):
-    formatted_prompt = format_prompt(f"{system_prompt}, {prompt}", history, genre, players)
-    stream = client.text_generation(formatted_prompt, stream=True, details=True, return_full_text=True)
-    output = ""
-    for response in stream:
-        output += response.token.text
-        yield output
-    return output
-
-additional_inputs = [
-    gr.Textbox(label="System Prompt", max_lines=1, interactive=True, visible=False),
-    gr.Radio(label="Game Genre", choices=list(GENRE_PROMPTS.keys()), value="Horror", interactive=True),
-    gr.Textbox(label="Players (comma-separated)", interactive=True, placeholder="Player1, Player2", visible=True),
-    gr.Slider(label="Temperature", value=0.1, minimum=0.0, maximum=1.0, step=0.05, interactive=True, visible=False),
-    gr.Slider(label="Max new tokens", value=1024, minimum=128, maximum=8192, step=64, interactive=True, visible=False),
-    gr.Slider(label="Top-p (nucleus sampling)", value=0.90, minimum=0.0, maximum=1, step=0.05, interactive=True, visible=False),
-    gr.Slider(label="Repetition penalty", value=1.2, minimum=1.0, maximum=2.0, step=0.05, interactive=True, visible=False),
-]
-
-examples = [["Подробнее"], ["Обстановка"]]
-
-def wrapped_generate(prompt, history, system_prompt, genre, temperature, max_new_tokens, top_p, repetition_penalty, players):
-    player_list = [p.strip() for p in players.split(",") if p.strip()]
-    return generate(prompt, history, system_prompt, genre, player_list, temperature, max_new_tokens, top_p, repetition_penalty)
-
-gr.ChatInterface(
-    fn=wrapped_generate,
-    chatbot=gr.Chatbot(show_label=False),
-    additional_inputs=additional_inputs,
-    title="theGame with Co-op",
-    examples=examples,
-    concurrency_limit=20,
-).launch(show_api=False)
+def generate(player_id, message, genre, temperature=0.9, max_new_tokens=256, top_p=0.95, repetition_penalty=1.0):
+    if player_id not in game_state["players"]:
+        game_state["players"][player_id] = {"history": []}
+    
+    history = game_state["players"][player_id]["history"]
+    formatted_prompt = format_prompt(message, history, genre, player_id)
+    
+    generate_kwargs = {
+        "temperature": temperature,
+        "max_new_tokens": max_new_tokens,
+        "top_p": top_p,
+        "repetition_penalty": repetition_penalty,
+        "do_sample": True,
+        "seed": 42,
+    }
+    
+    response = client.text_generation(formatted_prompt, **generate_kwargs)
+    game_state["players"][player_id]["history"].append((message, response))
+    game_state["story_progress"] += f"\n{response}"
+    
+    return response
+
+def gradio_interface():
+    with gr.Blocks() as demo:
+        player_id = gr.Textbox(label="Player ID", placeholder="Введите ваш уникальный идентификатор")
+        genre = gr.Radio(label="Game Genre", choices=list(GENRE_PROMPTS.keys()), value="Horror")
+        message = gr.Textbox(label="Ваше сообщение")
+        output = gr.Textbox(label="Ответ игры")
+        
+        def play_game(player_id, message, genre):
+            return generate(player_id, message, genre)
+        
+        submit = gr.Button("Отправить")
+        submit.click(play_game, inputs=[player_id, message, genre], outputs=output)
+    
+    demo.launch()
+
+if __name__ == "__main__":
+    gradio_interface()