Update app.py

app.py

@@ -515,6 +515,101 @@ def on_load_previous():
             gender_val, generation_val, psychotype_val, business_stage_val, industry_val, opf_val,
             chosen_approach_val, p1, p2, "", "", "")
 
+def save_preferred_sms_to_github(
+    selected_product, description, product_name, benefits, key_message,
+    gender, generation, psychotype, business_stage, industry, opf,
+    chosen_approach, presence_in_db,
+    model_1_name, prompt_1, sms_1,
+    model_2_name, prompt_2, sms_2,
+    chosen_sms
+):
+    """
+    Сохраняет выбранные поля в отдельный JSON-файл на GitHub,
+    чтобы не смешивать с предыдущими действиями, сохраняем под другим названием.
+    """
+    # Собираем все данные
+    data_to_save = {
+        "timestamp": time.time(),
+        "product_dropdown": selected_product,
+        "description": description,
+        "product_name": product_name,
+        "benefits": benefits,
+        "key_message": key_message,
+        "gender": gender,
+        "generation": generation,
+        "psychotype": psychotype,
+        "business_stage": business_stage,
+        "industry": industry,
+        "opf": opf,
+        "chosen_approach": chosen_approach,
+        "comment": presence_in_db,
+        "model_1_name": model_1_name,
+        "prompt_1": prompt_1,
+        "sms_1": sms_1,
+        "model_2_name": model_2_name,
+        "prompt_2": prompt_2,
+        "sms_2": sms_2,
+        "preferred_sms": chosen_sms  # "sms_1" или "sms_2"
+    }
+
+    file_content_encoded = base64.b64encode(json.dumps(data_to_save).encode()).decode()
+    filename = f"preferred_sms_{int(time.time())}.json"
+    url = f"https://api.github.com/repos/{repo}/contents/{filename}"
+    headers = {
+        "Authorization": f"token {token}",
+        "Content-Type": "application/json"
+    }
+    data = {
+        "message": f"Добавлен файл {filename} со сведениями о предпочтённом SMS",
+        "content": file_content_encoded
+    }
+    response = requests.put(url, headers=headers, data=json.dumps(data))
+    if response.status_code == 201:
+        print(f"Файл {filename} успешно сохранен.")
+    else:
+        print(f"Ошибка при сохранении данных: {response.status_code}, {response.text}")
+
+def on_prefer_sms_1(
+    selected_product, description, product_name, benefits, key_message,
+    gender, generation, psychotype, business_stage, industry, opf,
+    chosen_approach, presence_in_db,
+    model_1_name, prompt_1, sms_1,
+    model_2_name, prompt_2, sms_2
+):
+    """
+    Вызывается при нажатии кнопки «Я предпочитаю это SMS» (для SMS 1).
+    """
+    save_preferred_sms_to_github(
+        selected_product, description, product_name, benefits, key_message,
+        gender, generation, psychotype, business_stage, industry, opf,
+        chosen_approach, presence_in_db,
+        model_1_name, prompt_1, sms_1,
+        model_2_name, prompt_2, sms_2,
+        chosen_sms="sms_1"
+    )
+    return "Предпочтение SMS 1 сохранено в GitHub"
+
+def on_prefer_sms_2(
+    selected_product, description, product_name, benefits, key_message,
+    gender, generation, psychotype, business_stage, industry, opf,
+    chosen_approach, presence_in_db,
+    model_1_name, prompt_1, sms_1,
+    model_2_name, prompt_2, sms_2
+):
+    """
+    Вызывается при нажатии кнопки «Я предпочитаю это SMS» (для SMS 2).
+    """
+    save_preferred_sms_to_github(
+        selected_product, description, product_name, benefits, key_message,
+        gender, generation, psychotype, business_stage, industry, opf,
+        chosen_approach, presence_in_db,
+        model_1_name, prompt_1, sms_1,
+        model_2_name, prompt_2, sms_2,
+        chosen_sms="sms_2"
+    )
+    return "Предпочтение SMS 2 сохранено в GitHub"
+    
+
 with gr.Blocks(theme="default") as demo:
     gr.Markdown("**Процент созданных SMS по выбранному продукту**")
     progress_bar_html = """