Update app.py

app.py

@@ -1,10 +1,17 @@
-from huggingface_hub import InferenceClient
+from huggingface_hub import InferenceClient 
 import gradio as gr
 
+# Подключаем клиента для текстовой модели
 client = InferenceClient(
     "mistralai/Mistral-7B-Instruct-v0.3"
 )
 
+# Подключаем клиента для генерации изображений
+image_client = InferenceClient(
+    "stabilityai/stable-diffusion-2"
+)
+
+# Форматируем текстовый запрос
 def format_prompt(message, history, genre):
     prompt = f"Ты — персонаж в текстовой игре, тебя зовут Рассказчик, взаимодействующий с игроком на русском языке. " \
              f"Ты находишься в жанре игры: {genre}. Каждый твой ответ должен продолжать сюжет игры, принимая во внимание прошлые действия игрока и текущую ситуацию. " \
@@ -20,6 +27,7 @@ def format_prompt(message, history, genre):
     prompt += f"[INST] {message} [/INST]"
     return prompt
 
+# Генерация текста
 def generate(
     prompt, history, system_prompt, genre, temperature=0.9, max_new_tokens=256, top_p=0.95, repetition_penalty=1.0,
 ):
@@ -46,20 +54,26 @@ def generate(
         yield output
     return output
 
+# Генерация изображений
+def generate_image(description):
+    image = image_client.text_to_image(description, width=512, height=512)
+    return image.url
+
+# Дополнительные настройки для Gradio
 additional_inputs = [
     gr.Textbox(
         label="System Prompt",
         max_lines=1,
         interactive=True,
-        visible=False  # Hides the system prompt input
+        visible=False  # Скрыть поле ввода системного запроса
     ),
     gr.Radio(
         label="Game Genre",
         choices=["Investigation", "Fantasy", "Sci-Fi", "Horror", "Adventure", "Mystery"],
-        value="Horror",  # Default genre
+        value="Horror",  # Жанр по умолчанию
         interactive=True,
         info="Select the genre of the game",
-        visible=True  # Hides the genre selection
+        visible=True  # Отображение выбора жанра
     ),
     gr.Slider(
         label="Temperature",
@@ -69,7 +83,7 @@ additional_inputs = [
         step=0.05,
         interactive=True,
         info="Higher values produce more diverse outputs",
-        visible=False  # Hides the temperature slider
+        visible=False  # Скрыть слайдер температуры
     ),
     gr.Slider(
         label="Max new tokens",
@@ -79,7 +93,7 @@ additional_inputs = [
         step=64,
         interactive=True,
         info="The maximum numbers of new tokens",
-        visible=False  # Hides the max new tokens slider
+        visible=False  # Скрыть слайдер максимального числа токенов
     ),
     gr.Slider(
         label="Top-p (nucleus sampling)",
@@ -89,7 +103,7 @@ additional_inputs = [
         step=0.05,
         interactive=True,
         info="Higher values sample more low-probability tokens",
-        visible=False  # Hides the top-p slider
+        visible=False  # Скрыть слайдер top-p
     ),
     gr.Slider(
         label="Repetition penalty",
@@ -99,12 +113,19 @@ additional_inputs = [
         step=0.05,
         interactive=True,
         info="Penalize repeated tokens",
-        visible=False  # Hides the repetition penalty slider
+        visible=False  # Скрыть слайдер штрафа за повторения
+    ),
+    gr.Textbox(
+        label="Image Description",
+        placeholder="Describe the scene or character you want to generate.",
+        interactive=True
     )
 ]
 
+# Пример выбора
 examples = [["Подробнее"], ["Варианты"]]
 
+# Интерфейс Gradio
 gr.ChatInterface(
     fn=generate,
     chatbot=gr.Chatbot(show_label=False, show_share_button=True, show_copy_button=True, likeable=True, layout="panel"),
@@ -113,3 +134,11 @@ gr.ChatInterface(
     examples=examples,
     concurrency_limit=20,
 ).launch(show_api=False)
+
+# Интерфейс для генерации картинок
+gr.Interface(
+    fn=generate_image,
+    inputs="text",
+    outputs="image",
+    title="Image Generator"
+).launch()