fruitpicker01 commited on
Commit
9a049fe
·
verified ·
1 Parent(s): 9862383

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -10
app.py CHANGED
@@ -144,16 +144,16 @@ def generate_messages(description, advantages, *selected_values):
144
  # Возвращаем результаты для дальнейшей персонализации
145
  return results
146
 
147
- # Функция для выполнения персонализации
148
- def personalize_all_texts(*selected_values):
149
- personalized_texts = []
150
  for i, feature in enumerate(features.keys()):
151
  if selected_values[i]:
152
- text = features[feature].get(selected_values[i], "Текст не найден")
153
- personalized_texts.append(f"Персонализированный текст для {feature}: {text}")
154
- else:
155
- personalized_texts.append(f"Персонализированный текст для {feature}: не выбран")
156
- return personalized_texts
157
 
158
  # Создание интерфейса Gradio
159
  with gr.Blocks() as demo:
@@ -210,9 +210,9 @@ with gr.Blocks() as demo:
210
 
211
  with gr.Row():
212
  personalize_btn.click(
213
- personalize_all_texts,
214
  inputs=selections,
215
- outputs=[gr.Textbox(label=f"Персонализированный текст для {feature}", lines=3) for feature in features.keys()]
216
  )
217
 
218
  demo.launch()
 
144
  # Возвращаем результаты для дальнейшей персонализации
145
  return results
146
 
147
+ # Функция для генерации персонализированного промпта
148
+ def generate_personalization_prompt(*selected_values):
149
+ prompt = "Адаптируй текст с учетом следующих особенностей:\n"
150
  for i, feature in enumerate(features.keys()):
151
  if selected_values[i]:
152
+ try:
153
+ prompt += f"{features[feature][selected_values[i]]}\n"
154
+ except KeyError:
155
+ return f"Ошибка: выбранное значение {selected_values[i]} не найдено в данных."
156
+ return prompt.strip()
157
 
158
  # Создание интерфейса Gradio
159
  with gr.Blocks() as demo:
 
210
 
211
  with gr.Row():
212
  personalize_btn.click(
213
+ generate_personalization_prompt,
214
  inputs=selections,
215
+ outputs=[gr.Textbox(label="Промпт для персонализации", lines=6)]
216
  )
217
 
218
  demo.launch()