Update app.py

app.py

@@ -53,16 +53,109 @@ for sheet_name, df in data.items():
 current_request_index = -1  # Изначально указывает на последний запрос
 
 
-def save_user_request_to_github(description, advantages, personalization_params):
-
+def save_user_request_to_github(
+    description, 
+    advantages, 
+    personalization_params, 
+    prompt_display, 
+    output_text_gpt4o, 
+    output_text_gigachat_pro, 
+    output_text_gigachat_lite, 
+    output_text_gigachat_plus, 
+    output_text_meta_llama_405b, 
+    output_text_meta_llama_70b, 
+    output_text_meta_llama_8b, 
+    output_text_gemma_27b, 
+    output_text_gemma_9b, 
+    output_text_mistral, 
+    personalized_output_text_gpt4o, 
+    personalized_output_text_gigachat_pro, 
+    personalized_output_text_gigachat_lite, 
+    personalized_output_text_gigachat_plus, 
+    personalized_output_text_meta_llama_405b, 
+    personalized_output_text_meta_llama_70b, 
+    personalized_output_text_meta_llama_8b, 
+    personalized_output_text_gemma_27b, 
+    personalized_output_text_gemma_9b, 
+    personalized_output_text_mistral, 
+    comment_gpt4o, 
+    comment_gigachat_pro, 
+    comment_gigachat_lite, 
+    comment_gigachat_plus, 
+    comment_meta_llama_405b, 
+    comment_meta_llama_70b, 
+    comment_meta_llama_8b, 
+    comment_gemma_27b, 
+    comment_gemma_9b, 
+    comment_mistral, 
+    corrected_gpt4o, 
+    corrected_gigachat_pro, 
+    corrected_gigachat_lite, 
+    corrected_gigachat_plus, 
+    corrected_meta_llama_405b, 
+    corrected_meta_llama_70b, 
+    corrected_meta_llama_8b, 
+    corrected_gemma_27b, 
+    corrected_gemma_9b, 
+    corrected_mistral
+):
     global current_request_index  # Используем глобальную переменную
     current_request_index = -1  # Сбрасываем позицию к последнему запросу
-    
+
     # Собираем все данные в один словарь
     data_to_save = {
         "description": description,
         "advantages": advantages,
         "personalization_params": personalization_params,
+        "prompt_display": prompt_display,  # Добавляем промпт
+        "output_texts": {  # Добавляем сгенерированные сообщения
+            "gpt4o": output_text_gpt4o,
+            "gigachat_pro": output_text_gigachat_pro,
+            "gigachat_lite": output_text_gigachat_lite,
+            "gigachat_plus": output_text_gigachat_plus,
+            "meta_llama_405b": output_text_meta_llama_405b,
+            "meta_llama_70b": output_text_meta_llama_70b,
+            "meta_llama_8b": output_text_meta_llama_8b,
+            "gemma_27b": output_text_gemma_27b,
+            "gemma_9b": output_text_gemma_9b,
+            "mistral": output_text_mistral
+        },
+        "personalized_texts": {  # Добавляем персонализированные сообщения
+            "gpt4o": personalized_output_text_gpt4o,
+            "gigachat_pro": personalized_output_text_gigachat_pro,
+            "gigachat_lite": personalized_output_text_gigachat_lite,
+            "gigachat_plus": personalized_output_text_gigachat_plus,
+            "meta_llama_405b": personalized_output_text_meta_llama_405b,
+            "meta_llama_70b": personalized_output_text_meta_llama_70b,
+            "meta_llama_8b": personalized_output_text_meta_llama_8b,
+            "gemma_27b": personalized_output_text_gemma_27b,
+            "gemma_9b": personalized_output_text_gemma_9b,
+            "mistral": personalized_output_text_mistral
+        },
+        "comments": {  # Добавляем комментарии
+            "gpt4o": comment_gpt4o,
+            "gigachat_pro": comment_gigachat_pro,
+            "gigachat_lite": comment_gigachat_lite,
+            "gigachat_plus": comment_gigachat_plus,
+            "meta_llama_405b": comment_meta_llama_405b,
+            "meta_llama_70b": comment_meta_llama_70b,
+            "meta_llama_8b": comment_meta_llama_8b,
+            "gemma_27b": comment_gemma_27b,
+            "gemma_9b": comment_gemma_9b,
+            "mistral": comment_mistral
+        },
+        "corrected_texts": {  # Добавляем откорректированные сообщения
+            "gpt4o": corrected_gpt4o,
+            "gigachat_pro": corrected_gigachat_pro,
+            "gigachat_lite": corrected_gigachat_lite,
+            "gigachat_plus": corrected_gigachat_plus,
+            "meta_llama_405b": corrected_meta_llama_405b,
+            "meta_llama_70b": corrected_meta_llama_70b,
+            "meta_llama_8b": corrected_meta_llama_8b,
+            "gemma_27b": corrected_gemma_27b,
+            "gemma_9b": corrected_gemma_9b,
+            "mistral": corrected_mistral
+        },
         "timestamp": time.time()
     }
 
@@ -108,7 +201,7 @@ def load_previous_user_request_from_github():
 
         if not json_files:
             print("Нет сохраненных запросов.")
-            return "", "", None, None, None, None, None, None
+            return "", "", None, None, None, None, None, None, "", "", "", "", "", "", "", "", "", "", "", "", "", ""
 
         # Определяем новый индекс для загрузки предыдущего файла
         current_request_index -= 1
@@ -127,16 +220,65 @@ def load_previous_user_request_from_github():
             data = json.loads(file_response.text)
             description = data.get('description', "")
             advantages = data.get('advantages', "")
-            personalization_params = data.get('personalization_params', [None] * 6)  # Убедитесь, что размер списка соответствует количеству полей
-
-            # Возвращаем данные по отдельности для каждого компонента Gradio
-            return description, advantages, *personalization_params
+            personalization_params = data.get('personalization_params', [None] * 6)
+            prompt_display = data.get('prompt_display', "")
+            output_texts = data.get('output_texts', {})
+            personalized_texts = data.get('personalized_texts', {})
+            comments = data.get('comments', {})
+            corrected_texts = data.get('corrected_texts', {})
+
+            return (
+                description, 
+                advantages, 
+                *personalization_params,
+                prompt_display,
+                output_texts.get("gpt4o", ""),
+                output_texts.get("gigachat_pro", ""),
+                output_texts.get("gigachat_lite", ""),
+                output_texts.get("gigachat_plus", ""),
+                output_texts.get("meta_llama_405b", ""),
+                output_texts.get("meta_llama_70b", ""),
+                output_texts.get("meta_llama_8b", ""),
+                output_texts.get("gemma_27b", ""),
+                output_texts.get("gemma_9b", ""),
+                output_texts.get("mistral", ""),
+                personalized_texts.get("gpt4o", ""),
+                personalized_texts.get("gigachat_pro", ""),
+                personalized_texts.get("gigachat_lite", ""),
+                personalized_texts.get("gigachat_plus", ""),
+                personalized_texts.get("meta_llama_405b", ""),
+                personalized_texts.get("meta_llama_70b", ""),
+                personalized_texts.get("meta_llama_8b", ""),
+                personalized_texts.get("gemma_27b", ""),
+                personalized_texts.get("gemma_9b", ""),
+                personalized_texts.get("mistral", ""),
+                comments.get("gpt4o", ""),
+                corrected_texts.get("gpt4o", ""),
+                comments.get("gigachat_pro", ""),
+                corrected_texts.get("gigachat_pro", ""),
+                comments.get("gigachat_lite", ""),
+                corrected_texts.get("gigachat_lite", ""),
+                comments.get("gigachat_plus", ""),
+                corrected_texts.get("gigachat_plus", ""),
+                comments.get("meta_llama_405b", ""),
+                corrected_texts.get("meta_llama_405b", ""),
+                comments.get("meta_llama_70b", ""),
+                corrected_texts.get("meta_llama_70b", ""),
+                comments.get("meta_llama_8b", ""),
+                corrected_texts.get("meta_llama_8b", ""),
+                comments.get("gemma_27b", ""),
+                corrected_texts.get("gemma_27b", ""),
+                comments.get("gemma_9b", ""),
+                corrected_texts.get("gemma_9b", ""),
+                comments.get("mistral", ""),
+                corrected_texts.get("mistral", "")
+            )
         else:
             print(f"Ошибка при загрузке файла: {file_response.status_code}")
-            return "", "", None, None, None, None, None, None
+            return "", "", None, None, None, None, None, None, "", "", "", "", "", "", "", "", "", "", "", "", "", ""
     else:
         print(f"Ошибка при обращении к GitHub: {response.status_code}")
-        return "", "", None, None, None, None, None, None
+        return "", "", None, None, None, None, None, None, "", "", "", "", "", "", "", "", "", "", "", "", "", ""
 
 
 # Функция для генерации стандартного промпта
@@ -1097,7 +1239,52 @@ with gr.Blocks() as demo:
     load_btn.click(
         fn=lambda: load_previous_user_request_from_github(),
         inputs=[],
-        outputs=[description_input, advantages_input, *selections]
+        outputs=[
+            description_input,               # Описание предложения
+            advantages_input,                # Преимущества
+            *selections,                     # Параметры персонализации (Пол, Поколение, и т.д.)
+            prompt_display,                  # Неперсонализированный промпт
+            output_text_gpt4o,               # Неперсонализированное сообщение GPT-4o
+            output_text_gigachat_pro,         # Неперсонал��зированное сообщение GigaChat-Pro
+            output_text_gigachat_lite,        # Неперсонализированное сообщение GigaChat-Lite
+            output_text_gigachat_plus,        # Неперсонализированное сообщение GigaChat-Plus
+            output_text_meta_llama_405b,      # Неперсонализированное сообщение Meta-Llama-405B
+            output_text_meta_llama_70b,       # Неперсонализированное сообщение Meta-Llama-70B
+            output_text_meta_llama_8b,        # Неперсонализированное сообщение Meta-Llama-8B
+            output_text_gemma_27b,            # Неперсонализированное сообщение Gemma-27B
+            output_text_gemma_9b,             # Неперсонализированное сообщение Gemma-9B
+            output_text_mistral,              # Неперсонализированное сообщение Mistral
+            personalized_output_text_gpt4o,   # Персонализированное сообщение GPT-4o
+            personalized_output_text_gigachat_pro,  # Персонализированное сообщение GigaChat-Pro
+            personalized_output_text_gigachat_lite, # Персонализированное сообщение GigaChat-Lite
+            personalized_output_text_gigachat_plus, # Персонализированное сообщение GigaChat-Plus
+            personalized_output_text_meta_llama_405b, # Персонализированное сообщение Meta-Llama-405B
+            personalized_output_text_meta_llama_70b,  # Персонализированное сообщение Meta-Llama-70B
+            personalized_output_text_meta_llama_8b,   # Персонализированное сообщение Meta-Llama-8B
+            personalized_output_text_gemma_27b,       # Персонализированное сообщение Gemma-27B
+            personalized_output_text_gemma_9b,        # Персонализированное сообщение Gemma-9B
+            personalized_output_text_mistral,         # Персонализированное сообщение Mistral
+            comment_gpt4o,                  # Комментарий к GPT-4o
+            corrected_gpt4o,                # Откорректированное сообщение GPT-4o
+            comment_gigachat_pro,           # Комментарий к GigaChat-Pro
+            corrected_gigachat_pro,         # Откорректированное сообщение GigaChat-Pro
+            comment_gigachat_lite,          # Комментарий к GigaChat-Lite
+            corrected_gigachat_lite,        # Откорректированное сообщение GigaChat-Lite
+            comment_gigachat_plus,          # Комментарий к GigaChat-Plus
+            corrected_gigachat_plus,        # Откорректированное сообщение GigaChat-Plus
+            comment_meta_llama_405b,        # Комментарий к Meta-Llama-405B
+            corrected_meta_llama_405b,      # Откорректированное сообщение Meta-Llama-405B
+            comment_meta_llama_70b,         # Комментарий к Meta-Llama-70B
+            corrected_meta_llama_70b,       # Откорректированное сообщение Meta-Llama-70B
+            comment_meta_llama_8b,          # Комментарий к Meta-Llama-8B
+            corrected_meta_llama_8b,        # Откорректированное сообщение Meta-Llama-8B
+            comment_gemma_27b,              # Комментарий к Gemma-27B
+            corrected_gemma_27b,            # Откорректированное сообщение Gemma-27B
+            comment_gemma_9b,               # Комментарий к Gemma-9B
+            corrected_gemma_9b,             # Откорректированное сообщение Gemma-9B
+            comment_mistral,                # Комментарий к Mistral
+            corrected_mistral               # Откорректированное сообщение Mistral
+        ]
     )