Update app.py

app.py

@@ -94,6 +94,7 @@ def generate_message_gpt4o(prompt, temperature=1):
     except Exception as e:
         return f"Ошибка при обращении к ChatGPT-4o-Latest: {e}"
 
+
 def clean_message(message):
     # Если сообщение не заканчивается на точку или восклицательный знак, обрезаем его до последней точки
     if not message.endswith(('.', '!', '?')):
@@ -163,43 +164,9 @@ def generate_message_gigachat_plus_with_retry(prompt):
             return message
     return message
 
-# Создание классов для каждой модели
-class GPT4oModel:
-    def __init__(self, temperature=1):
-        self.temperature = temperature
-
-    def generate_message(self, prompt):
-        return generate_message_gpt4o(prompt, temperature=self.temperature)
-
-class GigaChatProModel:
-    def __init__(self, temperature=0.87):
-        self.temperature = temperature
-
-    def generate_message(self, prompt):
-        return generate_message_gigachat_pro(prompt, temperature=self.temperature)
-
-class GigaChatLiteModel:
-    def __init__(self, temperature=0.87):
-        self.temperature = temperature
-
-    def generate_message(self, prompt):
-        return generate_message_gigachat_lite(prompt, temperature=self.temperature)
-
-class GigaChatPlusModel:
-    def __init__(self, temperature=0.87):
-        self.temperature = temperature
-
-    def generate_message(self, prompt):
-        return generate_message_gigachat_plus(prompt, temperature=self.temperature)
 
 # Обновляем генерацию сообщений для отображения в интерфейсе
-def generate_messages(description, advantages, *selected_values, gpt4o_temp, gigachat_pro_temp, gigachat_lite_temp, gigachat_plus_temp):
-    # Создаем экземпляры классов с переданными значениями температур
-    gpt4o_model = GPT4oModel(temperature=gpt4o_temp)
-    gigachat_pro_model = GigaChatProModel(temperature=gigachat_pro_temp)
-    gigachat_lite_model = GigaChatLiteModel(temperature=gigachat_lite_temp)
-    gigachat_plus_model = GigaChatPlusModel(temperature=gigachat_plus_temp)
-
+def generate_messages(description, advantages, *selected_values):
     standard_prompt = generate_standard_prompt(description, advantages, *selected_values)
 
     results = {
@@ -212,32 +179,34 @@ def generate_messages(description, advantages, *selected_values, gpt4o_temp, gig
 
     yield results["prompt"], "", "", "", "", "Генерация стандартного промпта завершена"
 
-    results["gpt4o"] = gpt4o_model.generate_message(standard_prompt)
+    results["gpt4o"] = generate_message_gpt4o_with_retry(standard_prompt)
     gpt4o_length = len(results["gpt4o"])
     gpt4o_display = f"{results['gpt4o']}\n\n------\nКоличество знаков: {gpt4o_length}"
     yield results["prompt"], gpt4o_display, "", "", "", "Сообщение GPT-4o сгенерировано"
 
-    results["gigachat_pro"] = gigachat_pro_model.generate_message(standard_prompt)
+    results["gigachat_pro"] = generate_message_gigachat_pro_with_retry(standard_prompt)
     gigachat_pro_length = len(results["gigachat_pro"])
     gigachat_pro_display = f"{results['gigachat_pro']}\n\n------\nКоличество знаков: {gigachat_pro_length}"
     yield results["prompt"], gpt4o_display, gigachat_pro_display, "", "", "Сообщение GigaChat-Pro сгенерировано"
 
     time.sleep(2)
 
-    results["gigachat_lite"] = gigachat_lite_model.generate_message(standard_prompt)
+    results["gigachat_lite"] = generate_message_gigachat_lite_with_retry(standard_prompt)
     gigachat_lite_length = len(results["gigachat_lite"])
     gigachat_lite_display = f"{results['gigachat_lite']}\n\n------\nКоличество знаков: {gigachat_lite_length}"
     yield results["prompt"], gpt4o_display, gigachat_pro_display, gigachat_lite_display, "", "Сообщение GigaChat-Lite сгенерировано"
 
     time.sleep(2)
 
-    results["gigachat_plus"] = gigachat_plus_model.generate_message(standard_prompt)
+    results["gigachat_plus"] = generate_message_gigachat_plus_with_retry(standard_prompt)
     gigachat_plus_length = len(results["gigachat_plus"])
     gigachat_plus_display = f"{results['gigachat_plus']}\n\n------\nКоличество знаков: {gigachat_plus_length}"
     yield results["prompt"], gpt4o_display, gigachat_pro_display, gigachat_lite_display, gigachat_plus_display, "Все сообщения сгенерированы"
 
     return results
 
+
+
 # Функция для генерации персонализированного промпта
 def generate_personalization_prompt(*selected_values):
     prompt = "Адаптируй, не превышая длину сообщения в 250 знаков с пробелами, текст с учетом следующих особенностей:\n"
@@ -282,6 +251,7 @@ def perform_personalization_gigachat_with_retry(standard_message, personalizatio
             return message
     return message
 
+
 # Обновляем блок персонализации
 def personalize_messages_with_yield(gpt4o_message, gigachat_pro_message, gigachat_lite_message, gigachat_plus_message, *selected_values):
     personalization_prompt = generate_personalization_prompt(*selected_values)
@@ -307,6 +277,7 @@ def personalize_messages_with_yield(gpt4o_message, gigachat_pro_message, gigacha
     gigachat_plus_display = f"{personalized_message_gigachat_plus}\n\n------\nКоличество знаков: {gigachat_plus_length}"
     yield personalization_prompt, gpt4o_display, gigachat_pro_display, gigachat_lite_display, gigachat_plus_display, "Все персонализированные сообщения сгенерированы"
 
+
 # Функция для генерации промпта проверки текста
 def generate_error_check_prompt():
     prompt = (
@@ -351,6 +322,7 @@ def generate_error_check_prompt():
     )
     return prompt
 
+
 # Функция для выполнения проверки текста с использованием yield
 def check_errors_with_yield(*personalized_messages):
     if len(personalized_messages) < 4:
@@ -378,6 +350,7 @@ def check_errors_with_yield(*personalized_messages):
         
     yield error_check_prompt, error_message_gpt4o, error_message_gigachat_pro, error_message_gigachat_lite, error_message_gigachat_plus, "Все результаты проверки сгенерированы"
 
+
 def save_to_github(personalized_message, model_name, comment, corrected_message, description, advantages, non_personalized_prompt, non_personalized_message, gender, generation, psychotype, business_stage, industry, legal_form, temperature):
     # Собираем все данные в один словарь
     data_to_save = {
@@ -423,6 +396,7 @@ def save_to_github(personalized_message, model_name, comment, corrected_message,
         print(f"Ошибка при загрузке файла на GitHub: {response.status_code}")
         print(f"Ответ сервера: {response.json()}")
 
+
 # Создание интерфейса Gradio
 with gr.Blocks() as demo:
     gr.Markdown("# Генерация SMS-сообщений по заданным признакам")
@@ -486,6 +460,7 @@ with gr.Blocks() as demo:
             outputs=[prompt_display, output_text_gpt4o, output_text_gigachat_pro, output_text_gigachat_lite, output_text_gigachat_plus]
         )
 
+
     with gr.Row():
         personalize_btn = gr.Button("2. Выполнить персонализацию (нажимать только после кнопки 1)", elem_id="personalize_button")
         
@@ -516,6 +491,7 @@ with gr.Blocks() as demo:
         corrected_gigachat_lite = gr.Textbox(label="", lines=3)
         corrected_gigachat_plus = gr.Textbox(label="", lines=3)
 
+    
     # Отдельная строка для кнопок с использованием пустой колонки
     with gr.Row():
         gr.Button("Жми 👍 для сохранения удачного SMS в базу =>")
@@ -613,6 +589,7 @@ with gr.Blocks() as demo:
             outputs=None
         )
 
+
     # Использование сохраненных переменных в следующем блоке
     with gr.Row():
         check_errors_btn = gr.Button("3. Проверить текст (нажимать только после кнопки 2) - экспериментальная фича, качество пока крайне низкое", elem_id="check_errors_button")
@@ -630,4 +607,5 @@ with gr.Blocks() as demo:
             ]
         )
 
+
     demo.launch()