Update app.py

app.py

@@ -53,109 +53,16 @@ for sheet_name, df in data.items():
 current_request_index = -1  # Изначально указывает на последний запрос
 
 
-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
-):
+def save_user_request_to_github(description, advantages, personalization_params):
+
     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()
     }
 
@@ -201,7 +108,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
@@ -220,65 +127,16 @@ 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)
-            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", "")
-            )
+            personalization_params = data.get('personalization_params', [None] * 6)  # Убедитесь, что размер списка соответствует количеству полей
+
+            # Возвращаем данные по отдельности для каждого компонента Gradio
+            return description, advantages, *personalization_params
         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
 
 
 # Функция для генерации стандартного промпта
@@ -1239,52 +1097,7 @@ with gr.Blocks() as demo:
     load_btn.click(
         fn=lambda: load_previous_user_request_from_github(),
         inputs=[],
-        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
-        ]
+        outputs=[description_input, advantages_input, *selections]
     )