Update app.py

app.py

@@ -301,41 +301,43 @@ 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(message, model_name):
-    # Формируем уникальное имя файла на основе текущего времени
-    timestamp = int(time.time())
-    file_name = f"file_{timestamp}.json"
-    
-    # Данные для записи
-    file_content = {
+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):
+    # Собираем все данные в один словарь
+    data_to_save = {
         "Модель": model_name,
-        "Сообщение": message
+        "Персонализированное сообщение": personalized_message,
+        "Комментарий": comment,
+        "Откорректированное сообщение": corrected_message,
+        "Описание предложения": description,
+        "Преимущества": advantages,
+        "Неперсонализированный промпт": non_personalized_prompt,
+        "Неперсонализированное сообщение": non_personalized_message,
+        "Пол": gender,
+        "Поколение": generation,
+        "Психотип": psychotype,
+        "Стадия бизнеса": business_stage,
+        "Отрасль": industry,
+        "ОПФ": legal_form
     }
 
     # Преобразуем контент в JSON-строку и кодируем в base64
-    file_content_encoded = base64.b64encode(json.dumps(file_content).encode()).decode()
+    file_content_encoded = base64.b64encode(json.dumps(data_to_save).encode()).decode()
 
     # Параметры для GitHub API
     repo = "fruitpicker01/Storage_1"
-    path = file_name
+    path = f"file_{int(time.time())}.json"
     url = f"https://api.github.com/repos/{repo}/contents/{path}"
     headers = {
         "Authorization": f"token {token}",
         "Content-Type": "application/json"
     }
     data = {
-        "message": f"Добавлен новый файл {file_name}",
+        "message": f"Добавлен новый файл {path}",
         "content": file_content_encoded
     }
 
     # Отправка POST-запроса на GitHub API для создания файла в репозитории
     response = requests.put(url, headers=headers, data=json.dumps(data))
-    
-    if response.status_code == 201:
-        print(f"Файл {file_name} успешно добавлен в репозиторий.")
-    else:
-        print(f"Ошибка при добавлении файла: {response.status_code}")
-        print(response.json())
 
 
 # Создание интерфейса Gradio
@@ -427,11 +429,89 @@ with gr.Blocks() as demo:
         save_gigachat_plus_btn = gr.Button("👍")
 
         # Привязка кнопок к функциям сохранения
-        save_gpt4o_btn.click(fn=lambda x: save_to_github(x, "GPT-4o"), inputs=personalized_output_text_gpt4o, outputs=None)
-        save_gigachat_pro_btn.click(fn=lambda x: save_to_github(x, "GigaChat-Pro"), inputs=personalized_output_text_gigachat_pro, outputs=None)
-        save_gigachat_lite_btn.click(fn=lambda x: save_to_github(x, "GigaChat-Lite"), inputs=personalized_output_text_gigachat_lite, outputs=None)
-        save_gigachat_plus_btn.click(fn=lambda x: save_to_github(x, "GigaChat-Lite+"), inputs=personalized_output_text_gigachat_plus, outputs=None)
+        save_gpt4o_btn.click(
+            fn=lambda personalized_message, comment, corrected_message, description, advantages, non_personalized_prompt, non_personalized_message, gender, generation, psychotype, business_stage, industry, legal_form: 
+                save_to_github(personalized_message, "GPT-4o", comment, corrected_message, description, advantages, non_personalized_prompt, non_personalized_message, gender, generation, psychotype, business_stage, industry, legal_form),
+            inputs=[
+                personalized_output_text_gpt4o,
+                comment_gpt4o,
+                corrected_gpt4o,
+                description_input,
+                advantages_input,
+                prompt_display,
+                output_text_gpt4o,
+                selections[0],  # Пол
+                selections[1],  # Поколение
+                selections[2],  # Психотип
+                selections[3],  # Стадия бизнеса
+                selections[4],  # Отрасль
+                selections[5]   # ОПФ
+            ],
+            outputs=None
+        )
+
+        save_gigachat_pro_btn.click(
+            fn=lambda personalized_message, comment, corrected_message, description, advantages, non_personalized_prompt, non_personalized_message, gender, generation, psychotype, business_stage, industry, legal_form: 
+                save_to_github(personalized_message, "GigaChat-Pro", comment, corrected_message, description, advantages, non_personalized_prompt, non_personalized_message, gender, generation, psychotype, business_stage, industry, legal_form),
+            inputs=[
+                personalized_output_text_gigachat_pro,
+                comment_gigachat_pro,
+                corrected_gigachat_pro,
+                description_input,
+                advantages_input,
+                prompt_display,
+                output_text_gigachat_pro,
+                selections[0],  # Пол
+                selections[1],  # Поколение
+                selections[2],  # Психотип
+                selections[3],  # Стадия бизнеса
+                selections[4],  # Отрасль
+                selections[5]   # ОПФ
+            ],
+            outputs=None
+        )
+
+        save_gigachat_lite_btn.click(
+            fn=lambda personalized_message, comment, corrected_message, description, advantages, non_personalized_prompt, non_personalized_message, gender, generation, psychotype, business_stage, industry, legal_form: 
+                save_to_github(personalized_message, "GigaChat-Lite", comment, corrected_message, description, advantages, non_personalized_prompt, non_personalized_message, gender, generation, psychotype, business_stage, industry, legal_form),
+            inputs=[
+                personalized_output_text_gigachat_lite,
+                comment_gigachat_lite,
+                corrected_gigachat_lite,
+                description_input,
+                advantages_input,
+                prompt_display,
+                output_text_gigachat_lite,
+                selections[0],  # Пол
+                selections[1],  # Поколение
+                selections[2],  # Психотип
+                selections[3],  # Стадия бизнеса
+                selections[4],  # Отрасль
+                selections[5]   # ОПФ
+            ],
+            outputs=None
+        )
 
+        save_gigachat_plus_btn.click(
+            fn=lambda personalized_message, comment, corrected_message, description, advantages, non_personalized_prompt, non_personalized_message, gender, generation, psychotype, business_stage, industry, legal_form: 
+                save_to_github(personalized_message, "GigaChat-Lite+", comment, corrected_message, description, advantages, non_personalized_prompt, non_personalized_message, gender, generation, psychotype, business_stage, industry, legal_form),
+            inputs=[
+                personalized_output_text_gigachat_plus,
+                comment_gigachat_plus,
+                corrected_gigachat_plus,
+                description_input,
+                advantages_input,
+                prompt_display,
+                output_text_gigachat_plus,
+                selections[0],  # Пол
+                selections[1],  # Поколение
+                selections[2],  # Психотип
+                selections[3],  # Стадия бизнеса
+                selections[4],  # Отрасль
+                selections[5]   # ОПФ
+            ],
+            outputs=None
+        )
 
     # Использование сохраненных переменных в следующем блоке
     with gr.Row():