Update app.py

app.py

@@ -72,61 +72,6 @@ def generate_standard_prompt(description, advantages, *selected_values):
     )
     return prompt
 
-# Функции для генерации сообщений с персонализацией
-def create_personalization_prompt(message, *selected_values):
-    prompt = f"Пожалуйста, адаптируйте данное сообщение: '{message}' в соответствии со следующими требованиями:\n"
-    for i, feature in enumerate(features.keys()):
-        if selected_values[i]:
-            try:
-                prompt += f"- {features[feature][selected_values[i]]}\n"
-            except KeyError:
-                return f"Ошибка: выбранное значение {selected_values[i]} не найдено в данных."
-    return prompt.strip()
-
-def generate_personalized_message_gpt4o(message, personalization_prompt):
-    try:
-        headers = {
-            "Content-Type": "application/json",
-            "Authorization": f"Bearer {openai_api_key}"
-        }
-        data = {
-            "model": "chatgpt-4o-latest",
-            "messages": [{"role": "system", "content": f"{personalization_prompt}"}],
-            "max_tokens": 700
-        }
-        response = requests.post("https://api.openai.com/v1/chat/completions", json=data, headers=headers)
-        response_data = response.json()
-        return response_data["choices"][0]["message"]["content"].strip()
-    except Exception as e:
-        return f"Ошибка при обращении к ChatGPT-4o-Latest: {e}"
-
-def generate_personalized_message_gigachat_pro(message, personalization_prompt):
-    try:
-        messages = [SystemMessage(content=f"{personalization_prompt}")]
-        res = chat_pro(messages)
-        return res.content.strip()
-    except Exception as e:
-        return f"Ошибка при обращении к GigaChat-Pro: {e}"
-
-def generate_personalized_message_gigachat_lite(message, personalization_prompt):
-    try:
-        time.sleep(2)
-        messages = [SystemMessage(content=f"{personalization_prompt}")]
-        res = chat_lite(messages)
-        return res.content.strip()
-    except Exception as e:
-        return f"Ошибка при обращении к GigaChat-Lite: {e}"
-
-def generate_personalized_message_gigachat_plus(message, personalization_prompt):
-    try:
-        time.sleep(2)
-        messages = [SystemMessage(content=f"{personalization_prompt}")]
-        res = chat_plus(messages)
-        return res.content.strip()
-    except Exception as e:
-        return f"Ошибка при обращении к GigaChat-Lite+: {e}"
-
-
 # Функция для генерации сообщения с помощью ChatGPT-4o-Latest
 def generate_message_gpt4o(prompt):
     try:
@@ -167,7 +112,6 @@ def generate_message_gigachat_lite(prompt):
 # Функция для генерации сообщения с помощью GigaChat-Lite+
 def generate_message_gigachat_plus(prompt):
     try:
-        time.sleep(2)
         messages = [SystemMessage(content=prompt)]
         res = chat_plus(messages)
         return res.content.strip()
@@ -176,50 +120,29 @@ def generate_message_gigachat_plus(prompt):
 
 # Функция для обработки нажатия кнопки Submit и последовательного отображения результатов
 def generate_messages(description, advantages, *selected_values):
-    # Генерация стандартного промпта с использованием вашей функции
-    prompt = generate_standard_prompt(description, advantages, *selected_values)
+    standard_prompt = generate_standard_prompt(description, advantages, *selected_values)
 
-    # Генерация стандартных сообщений от разных моделей
-    message_gpt4o = generate_message_gpt4o(prompt)
-    yield prompt, message_gpt4o, "", "", "", "", "", "", "", "", "", "", ""
+    results = {
+        "prompt": standard_prompt,
+        "gpt4o": None,
+        "gigachat_pro": None,
+        "gigachat_lite": None,
+        "gigachat_plus": None
+    }
 
-    message_gigachat_pro = generate_message_gigachat_pro(prompt)
-    yield prompt, message_gpt4o, message_gigachat_pro, "", "", "", "", "", "", "", "", "", ""
+    yield results["prompt"], "", "", "", ""
 
-    message_gigachat_lite = generate_message_gigachat_lite(prompt)
-    yield prompt, message_gpt4o, message_gigachat_pro, message_gigachat_lite, "", "", "", "", "", "", "", ""
+    results["gpt4o"] = generate_message_gpt4o(standard_prompt)
+    yield results["prompt"], results["gpt4o"], "", "", ""
 
-    message_gigachat_plus = generate_message_gigachat_plus(prompt)
-    yield prompt, message_gpt4o, message_gigachat_pro, message_gigachat_lite, message_gigachat_plus, "", "", "", "", "", "", "", ""
+    results["gigachat_pro"] = generate_message_gigachat_pro(standard_prompt)
+    yield results["prompt"], results["gpt4o"], results["gigachat_pro"], "", ""
 
-    # Создание персонализированн��х промптов на основе сгенерированных сообщений и выбранных признаков
-    personalization_prompt_gpt4o = f"Пожалуйста, адаптируйте следующее сообщение: '{message_gpt4o}' под следующие требования:\n"
-    personalization_prompt_gigachat_pro = f"Пожалуйста, адаптируйте следующее сообщение: '{message_gigachat_pro}' под следующие требования:\n"
-    personalization_prompt_gigachat_lite = f"Пожалуйста, адаптируйте следующее сообщение: '{message_gigachat_lite}' под следующие требования:\n"
-    personalization_prompt_gigachat_plus = f"Пожалуйста, адаптируйте следующее сообщение: '{message_gigachat_plus}' под следующие требования:\n"
-
-    # Добавляем требования, выбранные пользователем
-    for i, feature in enumerate(features.keys()):
-        if selected_values[i]:
-            requirement = f"- {features[feature][selected_values[i]]}\n"
-            personalization_prompt_gpt4o += requirement
-            personalization_prompt_gigachat_pro += requirement
-            personalization_prompt_gigachat_lite += requirement
-            personalization_prompt_gigachat_plus += requirement
-
-    # Генерация и отображение персонализированных сообщений
-    personalized_message_gpt4o = generate_personalized_message_gpt4o(message_gpt4o, personalization_prompt_gpt4o)
-    yield prompt, message_gpt4o, message_gigachat_pro, message_gigachat_lite, message_gigachat_plus, personalized_message_gpt4o, "", "", "", personalization_prompt_gpt4o, "", "", ""
-
-    personalized_message_gigachat_pro = generate_personalized_message_gigachat_pro(message_gigachat_pro, personalization_prompt_gigachat_pro)
-    yield prompt, message_gpt4o, message_gigachat_pro, message_gigachat_lite, message_gigachat_plus, personalized_message_gpt4o, personalized_message_gigachat_pro, "", "", "", personalization_prompt_gigachat_pro, "", ""
-
-    personalized_message_gigachat_lite = generate_personalized_message_gigachat_lite(message_gigachat_lite, personalization_prompt_gigachat_lite)
-    yield prompt, message_gpt4o, message_gigachat_pro, message_gigachat_lite, message_gigachat_plus, personalized_message_gpt4o, personalized_message_gigachat_pro, personalized_message_gigachat_lite, "", "", "", personalization_prompt_gigachat_lite, ""
-
-    personalized_message_gigachat_plus = generate_personalized_message_gigachat_plus(message_gigachat_plus, personalization_prompt_gigachat_plus)
-    yield prompt, message_gpt4o, message_gigachat_pro, message_gigachat_lite, message_gigachat_plus, personalized_message_gpt4o, personalized_message_gigachat_pro, personalized_message_gigachat_lite, personalized_message_gigachat_plus, "", "", "", personalization_prompt_gigachat_plus
+    results["gigachat_lite"] = generate_message_gigachat_lite(standard_prompt)
+    yield results["prompt"], results["gpt4o"], results["gigachat_pro"], results["gigachat_lite"], ""
 
+    results["gigachat_plus"] = generate_message_gigachat_plus(standard_prompt)
+    yield results["prompt"], results["gpt4o"], results["gigachat_pro"], results["gigachat_lite"], results["gigachat_plus"]
 
 # Создание интерфейса Gradio
 with gr.Blocks() as demo:
@@ -227,7 +150,6 @@ with gr.Blocks() as demo:
     
     with gr.Row():
         with gr.Column(scale=1):
-            # Поле для ввода описания предложения
             description_input = gr.Textbox(
                 label="Описание предложения (предзаполненный пример можно поменять на свой)", 
                 lines=6,
@@ -241,8 +163,6 @@ with gr.Blocks() as demo:
                     "4. В течение следующего месяца пользоваться ей бесплатно."
                 )
             )
-            
-            # Поле для ввода преимуществ
             advantages_input = gr.Textbox(
                 label="Преимущества (предзаполненный пример можно поменять на свой)", 
                 lines=4,
@@ -254,8 +174,6 @@ with gr.Blocks() as demo:
                     "Мгновенные переводы на карты любых банков."
                 )
             )
-            
-            # Поля для выбора признаков
             selections = []
             for feature in features.keys():
                 selections.append(gr.Dropdown(choices=[None] + list(features[feature].keys()), label=f"Выберите {feature}"))
@@ -263,42 +181,16 @@ with gr.Blocks() as demo:
             submit_btn = gr.Button("Submit")  # Оранжевая кнопка по умолчанию
 
         with gr.Column(scale=2):
-            prompt_display = gr.Textbox(label="Сгенерированный промпт", lines=4, interactive=False)
+            prompt_display = gr.Textbox(label="Стандартный промпт", lines=15, interactive=False)
             output_text_gpt4o = gr.Textbox(label="Создание сообщения GPT-4o", lines=3)
             output_text_gigachat_pro = gr.Textbox(label="Создание сообщения GigaChat-Pro", lines=3)
             output_text_gigachat_lite = gr.Textbox(label="Создание сообщения GigaChat-Lite", lines=3)
             output_text_gigachat_plus = gr.Textbox(label="Создание сообщения GigaChat-Lite+", lines=3)
-
-            # Добавляем поля для персонализированных сообщений
-            personalized_output_text_gpt4o = gr.Textbox(label="Персонализированное сообщение GPT-4o", lines=3)
-            personalized_output_text_gigachat_pro = gr.Textbox(label="Персонализированное сообщение GigaChat-Pro", lines=3)
-            personalized_output_text_gigachat_lite = gr.Textbox(label="Персонализированное сообщение GigaChat-Lite", lines=3)
-            personalized_output_text_gigachat_plus = gr.Textbox(label="Персонализированное сообщение GigaChat-Lite+", lines=3)
-
-            # Добавляем поля для персонализированных промптов
-            personalization_prompt_gpt4o_display = gr.Textbox(label="Промпт для персонализации GPT-4o", lines=6, interactive=False)
-            personalization_prompt_gigachat_pro_display = gr.Textbox(label="Промпт для персонализации GigaChat-Pro", lines=6, interactive=False)
-            personalization_prompt_gigachat_lite_display = gr.Textbox(label="Промпт для персонализации GigaChat-Lite", lines=6, interactive=False)
-            personalization_prompt_gigachat_plus_display = gr.Textbox(label="Промпт для персонализации GigaChat-Lite+", lines=6, interactive=False)
-
+            
             submit_btn.click(
-                generate_messages,
-                inputs=[description_input, advantages_input] + selections,
-                outputs=[
-                    prompt_display,
-                    output_text_gpt4o, 
-                    output_text_gigachat_pro, 
-                    output_text_gigachat_lite, 
-                    output_text_gigachat_plus,
-                    personalized_output_text_gpt4o,
-                    personalized_output_text_gigachat_pro,
-                    personalized_output_text_gigachat_lite,
-                    personalized_output_text_gigachat_plus,
-                    personalization_prompt_gpt4o_display,
-                    personalization_prompt_gigachat_pro_display,
-                    personalization_prompt_gigachat_lite_display,
-                    personalization_prompt_gigachat_plus_display
-                ]
+                generate_messages, 
+                inputs=[description_input, advantages_input] + selections, 
+                outputs=[prompt_display, output_text_gpt4o, output_text_gigachat_pro, output_text_gigachat_lite, output_text_gigachat_plus]
             )
 
     demo.launch()