Update app.py

app.py

@@ -95,7 +95,20 @@ def replace_k_with_thousands(message):
     # Замена и для 'k' и для 'К', с учётом регистра
     message = re.sub(r'(\d+)[kкКK]', r'\1 000', message, flags=re.IGNORECASE)
     return message
- 
+
+# Функция для корректировки использования тире и дефисов
+def correct_dash_usage(text):
+    # Step 1: Replace any dash with long dash if surrounded by spaces
+    text = re.sub(r'\s[-–—]\s', ' — ', text)
+
+    # Step 2: Replace any dash with short dash if surrounded by numbers without spaces
+    text = re.sub(r'(?<=\d)[-–—](?=\d)', '–', text)
+
+    # Step 3: Replace any dash with hyphen if surrounded by letters or a combination of letters and digits
+    text = re.sub(r'(?<=[a-zA-Zа-яА-Я0-9])[-–—](?=[a-zA-Zа-яА-Я0-9])', '-', text)
+
+    return text
+
 # Функция для добавления ошибок в промпт для перегенерации
 def append_errors_to_prompt(prompt, checks):
     errors = []
@@ -145,6 +158,8 @@ def generate_message_gigachat_pro_with_retry(prompt):
 
         # Генерация сообщения
         message = generate_message_gigachat_pro(prompt)
+        message = replace_k_with_thousands(message)
+        message = correct_dash_usage(message)
         message_length = len(message)
 
         # Проверка длины сообщения. Если не проходит, сразу перегенерируем
@@ -154,7 +169,6 @@ def generate_message_gigachat_pro_with_retry(prompt):
             continue
 
         # Выполняем замены и проверки
-        message = replace_k_with_thousands(message)
         checks = perform_checks(message)
 
         # Обновляем последнее сообщение