Spaces:
Sleeping
Sleeping
fruitpicker01
commited on
Commit
•
aa349c2
1
Parent(s):
6b1f030
Update app.py
Browse files
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 |
-
# Функция для
|
|
|
|
|
|
|
|
|
|
|
|
|
94 |
def generate_message_gigachat_pro_with_retry(prompt):
|
95 |
for _ in range(10):
|
96 |
message = generate_message_gigachat_pro(prompt)
|
97 |
-
|
|
|
|
|
|
|
|
|
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 |
|