Update app.py

app.py

@@ -637,7 +637,157 @@ def on_prefer_sms_2(
         chosen_sms="sms_2"
     )
     return "Предпочтение SMS 2 сохранено в GitHub"
-    
+
+def save_sms_to_db(
+    selected_product,
+    description,
+    product_name,
+    benefits,
+    key_message,
+    gender,
+    generation,
+    psychotype,
+    business_stage,
+    industry,
+    opf,
+    chosen_approach,
+    presence_in_db,
+    model_name,
+    prompt_text,
+    final_prompt,  # добавили финальный промпт
+    sms_text,
+    comment_sms,
+    corrected_sms
+):
+    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": model_name,
+        "prompt": prompt_text,
+        "final_prompt": final_prompt,   # сохраняем финальный промпт
+        "sms": sms_text,
+        "comment_sms": comment_sms,
+        "corrected_sms": corrected_sms
+    }
+
+    file_content_encoded = base64.b64encode(json.dumps(data_to_save).encode()).decode()
+    filename = f"saved_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
+    }
+    resp = requests.put(url, headers=headers, data=json.dumps(data))
+    if resp.status_code == 201:
+        print(f"Файл {filename} успешно сохранён (save_sms_to_db).")
+    else:
+        print(f"Ошибка при сохранении (save_sms_to_db): {resp.status_code}, {resp.text}")
+
+def on_save_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,
+    final_prompt_1,
+    sms_1,
+    comment_sms_1,
+    corrected_sms_1
+):
+    save_sms_to_db(
+        selected_product=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,
+        presence_in_db=presence_in_db,
+        model_name=model_1_name,
+        prompt_text=prompt_1,
+        final_prompt=final_prompt_1,
+        sms_text=sms_1,
+        comment_sms=comment_sms_1,
+        corrected_sms=corrected_sms_1
+    )
+    return "SMS 1 сохранено в базу"
+
+
+def on_save_sms_2(
+    selected_product,
+    description,
+    product_name,
+    benefits,
+    key_message,
+    gender,
+    generation,
+    psychotype,
+    business_stage,
+    industry,
+    opf,
+    chosen_approach,
+    presence_in_db,
+    model_2_name,
+    prompt_2,
+    final_prompt_2,
+    sms_2,
+    comment_sms_2,
+    corrected_sms_2
+):
+    save_sms_to_db(
+        selected_product=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,
+        presence_in_db=presence_in_db,
+        model_name=model_2_name,
+        prompt_text=prompt_2,
+        final_prompt=final_prompt_2,
+        sms_text=sms_2,
+        comment_sms=comment_sms_2,
+        corrected_sms=corrected_sms_2
+    )
+    return "SMS 2 сохранено в базу"
+
 
 with gr.Blocks(theme="default") as demo:
     gr.Markdown("**Процент созданных SMS по выбранному продукту**")
@@ -867,5 +1017,57 @@ with gr.Blocks(theme="default") as demo:
         outputs=[]
     )
 
+    save_sms_1_btn.click(
+        fn=on_save_sms_1,
+        inputs=[
+            product_dropdown,
+            description,
+            product_name,
+            benefits,
+            key_message,
+            gender_dropdown,
+            generation_dropdown,
+            psychotype_dropdown,
+            business_stage_dropdown,
+            industry_dropdown,
+            opf_dropdown,
+            chosen_approach,
+            presence_in_db,
+            model_1_name,
+            prompt_1,
+            final_prompt_1_state,
+            sms_1,
+            comment_sms_1,
+            corrected_sms_1
+        ],
+        outputs=[]
+    )
+    
+    save_sms_2_btn.click(
+        fn=on_save_sms_2,
+        inputs=[
+            product_dropdown,
+            description,
+            product_name,
+            benefits,
+            key_message,
+            gender_dropdown,
+            generation_dropdown,
+            psychotype_dropdown,
+            business_stage_dropdown,
+            industry_dropdown,
+            opf_dropdown,
+            chosen_approach,
+            presence_in_db,
+            model_2_name,
+            prompt_2,
+            final_prompt_2_state,
+            sms_2,
+            comment_sms_2,
+            corrected_sms_2
+        ],
+        outputs=[]
+    )
+
 
     demo.queue().launch()