fruitpicker01 commited on
Commit
307fbe8
1 Parent(s): d8d03be

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +14 -9
app.py CHANGED
@@ -141,7 +141,6 @@ def generate_standard_prompt(description, advantages, key_message, *selected_val
141
 
142
 
143
  # Функция для создания задания для редактора с добавлением prefix и suffix
144
- # Функция для создания задания для редактора
145
  def generate_personalization_prompt(key_message, *selected_values, prefix, suffix):
146
  prompt = "Адаптируй, не превышая длину сообщения в 250 знаков с пробелами, текст с учетом следующих особенностей:\n"
147
  gender, generation, psychotype = selected_values[0], selected_values[1], selected_values[2]
@@ -192,14 +191,20 @@ def generate_personalization_prompt(key_message, *selected_values, prefix, suffi
192
 
193
  return prompt.strip()
194
 
195
-
196
- # Функция для удаления префиксов и суффиксов перед выводом на экран
197
- def remove_prefix_suffix(prompt, prefixes, suffixes):
198
  for prefix in prefixes:
199
  prompt = prompt.replace(prefix, "")
200
  for suffix in suffixes:
201
  prompt = prompt.replace(suffix, "")
202
- return prompt.strip()
 
 
 
 
 
 
203
 
204
  # Функция для постепенной генерации всех сообщений через yield
205
  def generate_all_messages(desc, benefits, key_message, gender, generation, psychotype, business_stage, industry, opf):
@@ -240,12 +245,12 @@ def generate_all_messages(desc, benefits, key_message, gender, generation, psych
240
  prefix=prefixes[i], suffix=suffixes[i]
241
  )
242
 
243
- # Удаляем префиксы и суффиксы перед выводом на экран
244
- display_personalization_prompt = remove_prefix_suffix(personalization_prompt, prefixes, suffixes)
245
 
246
  # Выводим неперсонализированное сообщение и задание для редактора
247
  yield (
248
- standard_prompt, display_personalization_prompt, # Задание для редактора без префиксов и суффиксов
249
  non_personalized_messages[0] if i >= 0 else None,
250
  personalized_messages[0] if len(personalized_messages) > 0 else None,
251
  non_personalized_messages[1] if i >= 1 else None,
@@ -263,7 +268,7 @@ def generate_all_messages(desc, benefits, key_message, gender, generation, psych
263
 
264
  # Выводим персонализированное сообщение
265
  yield (
266
- standard_prompt, display_personalization_prompt, # Задание для редактора без префиксов и суффиксов
267
  non_personalized_messages[0] if len(non_personalized_messages) > 0 else None,
268
  personalized_messages[0] if len(personalized_messages) > 0 else None,
269
  non_personalized_messages[1] if len(non_personalized_messages) > 1 else None,
 
141
 
142
 
143
  # Функция для создания задания для редактора с добавлением prefix и suffix
 
144
  def generate_personalization_prompt(key_message, *selected_values, prefix, suffix):
145
  prompt = "Адаптируй, не превышая длину сообщения в 250 знаков с пробелами, текст с учетом следующих особенностей:\n"
146
  gender, generation, psychotype = selected_values[0], selected_values[1], selected_values[2]
 
191
 
192
  return prompt.strip()
193
 
194
+ # Функция для удаления префиксов, суффиксов и пустых строк перед выводом на экран
195
+ def clean_prompt_for_display(prompt, prefixes, suffixes):
196
+ # Удаляем префиксы и суффиксы
197
  for prefix in prefixes:
198
  prompt = prompt.replace(prefix, "")
199
  for suffix in suffixes:
200
  prompt = prompt.replace(suffix, "")
201
+
202
+ # Удаляем пустые строки
203
+ lines = prompt.split('\n')
204
+ non_empty_lines = [line for line in lines if line.strip() != '']
205
+ cleaned_prompt = '\n'.join(non_empty_lines)
206
+
207
+ return cleaned_prompt.strip()
208
 
209
  # Функция для постепенной генерации всех сообщений через yield
210
  def generate_all_messages(desc, benefits, key_message, gender, generation, psychotype, business_stage, industry, opf):
 
245
  prefix=prefixes[i], suffix=suffixes[i]
246
  )
247
 
248
+ # Удаляем префиксы, суффиксы и пустые строки перед выводом на экран
249
+ display_personalization_prompt = clean_prompt_for_display(personalization_prompt, prefixes, suffixes)
250
 
251
  # Выводим неперсонализированное сообщение и задание для редактора
252
  yield (
253
+ standard_prompt, display_personalization_prompt, # Задание для редактора без префиксов, суффиксов и пустых строк
254
  non_personalized_messages[0] if i >= 0 else None,
255
  personalized_messages[0] if len(personalized_messages) > 0 else None,
256
  non_personalized_messages[1] if i >= 1 else None,
 
268
 
269
  # Выводим персонализированное сообщение
270
  yield (
271
+ standard_prompt, display_personalization_prompt, # Задание для редактора без префиксов, суффиксов и пустых строк
272
  non_personalized_messages[0] if len(non_personalized_messages) > 0 else None,
273
  personalized_messages[0] if len(personalized_messages) > 0 else None,
274
  non_personalized_messages[1] if len(non_personalized_messages) > 1 else None,