Update app.py

app.py

@@ -1,5 +1,14 @@
 import gradio as gr
 import pandas as pd
+import os
+import random
+from openai import OpenAI
+
+XAI_API_KEY = os.getenv("XAI_API_KEY")
+client = OpenAI(
+    api_key=XAI_API_KEY,
+    base_url="https://api.x.ai/v1",
+)
 
 # Функция для загрузки данных из вкладок Excel
 def load_dropdown_data(file_path, sheet_name, column_name):
@@ -98,9 +107,9 @@ def format_instruction_string(instr):
     terms = [t.strip() for t in instr.split(',') if t.strip()]
     return " или ".join(terms) if terms else ""
 
-def generate_prompts(description, product_name, benefits, key_message, chosen_approach,
-                     gender, generation, psychotype, business_stage, industry, opf):
-    # Если подход не определён
+def generate_display_prompts(description, product_name, benefits, key_message, chosen_approach,
+                             gender, generation, psychotype, business_stage, industry, opf):
+    # Формируем промпты для отображения (с объединением всех подходов)
     if chosen_approach == "Подход не найден для выбранных параметров.":
         return ("Для формирования промпта выберите хотя бы один личный персональный параметр для определения подхода",
                 "Для формирования промпта выберите хотя бы один личный персональный параметр для определения подхода")
@@ -159,7 +168,6 @@ def generate_prompts(description, product_name, benefits, key_message, chosen_ap
 Убедись, что в SMS без изменений, синонимов и перестановок слов используется наименование продукта: «{product_name}».
 Убедись, что в SMS есть следующая ключевая информация: «{key_message}»."""
     else:
-        # Нет персональных параметров — нет обязательных терминов
         prompt = f"""Напиши три или четыре предложения. {approach_prefix}. Напиши рекламное SMS для следующего продукта:
 «{description}».
 Не изменяй название продукта: «{product_name}».
@@ -172,15 +180,117 @@ def generate_prompts(description, product_name, benefits, key_message, chosen_ap
 
     return prompt, prompt
 
+def generate_model_prompt(description, product_name, benefits, key_message,
+                          gender, generation, psychotype, business_stage, industry, opf,
+                          single_approach):
+    # Генерируем промпт для обращения к модели, используя только один выбранный подход (single_approach)
+    prefix = approach_dict[single_approach]["prefix"]
+    suffix = approach_dict[single_approach]["suffix"]
+
+    instructions_data = [
+        (gender, genders_data, "Пол"),
+        (generation, generations_data, "Поколение"),
+        (psychotype, psychotypes_data, "Психотип"),
+        (business_stage, business_stages_data, "Стадия бизнеса"),
+        (industry, industries_data, "Отрасль"),
+        (opf, opfs_data, "ОПФ")
+    ]
+
+    chosen_params_instructions = []
+    for (param_value, df, col) in instructions_data:
+        instr1 = get_instructions_for_param(param_value, df, col)
+        if instr1:
+            chosen_params_instructions.append(instr1)
+
+    if chosen_params_instructions:
+        lines = []
+        for i, instr_line in enumerate(chosen_params_instructions, start=1):
+            formatted_line = format_instruction_string(instr_line)
+            lines.append(f"{i}. {formatted_line}.")
+        mandatory_terms = "\n".join(lines)
+    else:
+        mandatory_terms = None
+
+    if mandatory_terms:
+        model_prompt = f"""Напиши три или четыре предложения. {prefix}. 
+Напиши рекламное SMS для следующего продукта:
+«{description}».
+Не изменяй название продукта: «{product_name}».
+Преимущества:
+«{benefits}».
+ОБЯЗАТЕЛЬНО используй в SMS КАЖДЫЙ из следующих терминов:
+{mandatory_terms}
+Убедись, что написал не меньше трех и не больше четырех предложений.
+{suffix}.
+Убедись, что УМЕСТНО использовал КАЖДЫЙ необходимый термин.
+Убедись, что в SMS без изменений, синонимов и перестановок слов используется наименование продукта: «{product_name}».
+Убедись, что в SMS есть следующая ключевая информация: «{key_message}»."""
+    else:
+        model_prompt = f"""Напиши три или четыре предложения. {prefix}. Напиши рекламное SMS для следующего продукта:
+«{description}».
+Не изменяй название продукта: «{product_name}».
+Преимущества:
+«{benefits}».
+Убедись, что написал не меньше трех и не больше четырех предложений.
+{suffix}.
+Убедись, что в SMS без изменений, синонимов и перестановок слов используется наименование продукта: «{product_name}».
+Убедись, что в SMS есть следующая ключевая информация: «{key_message}»."""
+
+    return model_prompt
+
+def call_model(model_prompt):
+    completion = client.chat.completions.create(
+        model="grok-2-1212",
+        messages=[
+            {"role": "system", "content": "You are a world-class expert in creating personalized SMS who returns only the SMS and nothing else."},
+            {"role": "user", "content": model_prompt},
+        ],
+    )
+    return completion.choices[0].message.content.strip()
+
 def update_all_params(description, product_name, benefits, key_message, 
                       gender, generation, psychotype, business_stage, industry, opf):
     # Сначала определяем подход
     chosen_approach = get_approaches(gender, generation, psychotype, approaches_data)
-    # Затем формируем промпты
-    prompt_1, prompt_2 = generate_prompts(description, product_name, benefits, key_message,
-                                          chosen_approach, gender, generation, psychotype,
-                                          business_stage, industry, opf)
-    return chosen_approach, prompt_1, prompt_2
+    # Затем формируем промпты для отображения
+    prompt_1, prompt_2 = generate_display_prompts(description, product_name, benefits, key_message,
+                                                  chosen_approach, gender, generation, psychotype,
+                                                  business_stage, industry, opf)
+
+    # Если подход не найден или нет персональных параметров – показываем тексты как есть без генерации SMS
+    if chosen_approach == "Подход не найден для выбранных параметров." or "Для формирования промпта выберите" in prompt_1:
+        sms_1 = ""
+        sms_2 = ""
+        return chosen_approach, prompt_1, prompt_2, sms_1, sms_2
+
+    # Если подход есть
+    approach_list = [a.strip() for a in chosen_approach.split(',') if a.strip()]
+
+    # Выбираем случайный подход для model_prompt_1
+    if len(approach_list) > 1:
+        chosen_single_approach_1 = random.choice(approach_list)
+    else:
+        chosen_single_approach_1 = approach_list[0]
+
+    # Выбираем случайный подход для model_prompt_2
+    if len(approach_list) > 1:
+        chosen_single_approach_2 = random.choice(approach_list)
+    else:
+        chosen_single_approach_2 = approach_list[0]
+
+    model_prompt_1 = generate_model_prompt(description, product_name, benefits, key_message,
+                                           gender, generation, psychotype, business_stage, industry, opf,
+                                           chosen_single_approach_1)
+
+    model_prompt_2 = generate_model_prompt(description, product_name, benefits, key_message,
+                                           gender, generation, psychotype, business_stage, industry, opf,
+                                           chosen_single_approach_2)
+
+    # Вызываем модель для получения SMS
+    sms_1 = call_model(model_prompt_1)
+    sms_2 = call_model(model_prompt_2)
+
+    return chosen_approach, prompt_1, prompt_2, sms_1, sms_2
 
 with gr.Blocks(theme="default") as demo:
     gr.Markdown("**Процент созданных SMS по выбранному продукту**")
@@ -225,12 +335,12 @@ with gr.Blocks(theme="default") as demo:
     
     with gr.Row():
         with gr.Column():
-            model_1_name = gr.Textbox(label="Модель 1", value="Скрыто для слепого тестирования", interactive=False)
+            model_1_name = gr.Textbox(label="Модель 1", value="Grok-2-1212", interactive=False)
             prompt_1 = gr.Textbox(label="Промпт 1", value="", interactive=False, lines=10)
             sms_1 = gr.Textbox(label="SMS 1", lines=3, value="")
         
         with gr.Column():
-            model_2_name = gr.Textbox(label="Модель 2", value="Скрыто для слепого тестирования", interactive=False)
+            model_2_name = gr.Textbox(label="Модель 2", value="Grok-2-1212", interactive=False)
             prompt_2 = gr.Textbox(label="Промпт 2", value="", interactive=False, lines=10)
             sms_2 = gr.Textbox(label="SMS 2", lines=3, value="")
     
@@ -252,7 +362,7 @@ with gr.Blocks(theme="default") as demo:
         save_sms_1_btn = gr.Button("Сохранить в базу")
         save_sms_2_btn = gr.Button("Сохранить в базу")
 
-    # Обновляем промпты и подход при изменении параметров клиента
+    # Обновляем промпты, подход и генерируем SMS при изменении параметров клиента
     client_params = [gender_dropdown, generation_dropdown, psychotype_dropdown, business_stage_dropdown, industry_dropdown, opf_dropdown]
     for cp in client_params:
         cp.change(
@@ -260,7 +370,7 @@ with gr.Blocks(theme="default") as demo:
             inputs=[description, product_name, benefits, key_message, 
                     gender_dropdown, generation_dropdown, psychotype_dropdown,
                     business_stage_dropdown, industry_dropdown, opf_dropdown],
-            outputs=[chosen_approach, prompt_1, prompt_2]
+            outputs=[chosen_approach, prompt_1, prompt_2, sms_1, sms_2]
         )
 
     demo.launch()