fruitpicker01 commited on
Commit
aa349c2
1 Parent(s): 6b1f030

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +12 -2
app.py CHANGED
@@ -90,11 +90,21 @@ def generate_message_gigachat_pro(prompt):
90
  except Exception as e:
91
  return f"Ошибка при обращении к GigaChat-Pro: {e}"
92
 
93
- # Функция для повторной генерации сообщения, пока оно не станет короче 250 знаков
 
 
 
 
 
 
94
  def generate_message_gigachat_pro_with_retry(prompt):
95
  for _ in range(10):
96
  message = generate_message_gigachat_pro(prompt)
97
- if len(message) <= 250:
 
 
 
 
98
  return message
99
  return message
100
 
 
90
  except Exception as e:
91
  return f"Ошибка при обращении к GigaChat-Pro: {e}"
92
 
93
+ # Функция для замены сокращений с 'k' на тысячи
94
+ def replace_k_with_thousands(message):
95
+ message = re.sub(r'(\d+)k', r'\1 000', message, flags=re.IGNORECASE)
96
+ return message
97
+
98
+ # Функция для повторной генерации сообщения, пока оно не станет короче 250 знаков и не меньше 190 знаков
99
+ # Добавляем проверку в функцию перегенерации
100
  def generate_message_gigachat_pro_with_retry(prompt):
101
  for _ in range(10):
102
  message = generate_message_gigachat_pro(prompt)
103
+ message_length = len(message)
104
+ # Выполняем замены и проверки
105
+ message = replace_k_with_thousands(message)
106
+ checks = perform_checks(message)
107
+ if 190 <= message_length <= 250 and all(checks.values()):
108
  return message
109
  return message
110