Update app.py

app.py

@@ -16,7 +16,7 @@ token = os.getenv('GITHUB_TOKEN')
 
 # Функция аутентификации в сервисе GigaChat с заданной температурой
 def authenticate_gigachat(model, max_tokens, temperature):
-    return GigaChat(credentials=gc_key, model=model, max_tokens=max_tokens, temperature=temperature, verify_ssl_certs=False)
+    return GigaChat(credentials=gc_key, model=model, max_tokens=max_tokens, temperature=float(temperature), verify_ssl_certs=False)
 
 # Функция аутентификации для GPT-4o
 def authenticate_gpt4o(max_tokens, temperature):
@@ -27,7 +27,7 @@ def authenticate_gpt4o(max_tokens, temperature):
     return {
         "model": "chatgpt-4o-latest",
         "max_tokens": max_tokens,
-        "temperature": temperature,
+        "temperature": float(temperature),
         "headers": headers
     }
 
@@ -230,160 +230,32 @@ with gr.Blocks() as demo:
                 description_input, 
                 advantages_input, 
                 *selections,
-                authenticate_gpt4o(101, gpt4o_temperature),
-                authenticate_gigachat('GigaChat-Pro', 68, gigachat_pro_temperature),
-                authenticate_gigachat('GigaChat', 68, gigachat_lite_temperature),
-                authenticate_gigachat('GigaChat-Plus', 68, gigachat_plus_temperature)
+                gpt4o_temperature,  # Передаем температуру как компонент
+                gigachat_pro_temperature,  # Передаем температуру как компонент
+                gigachat_lite_temperature,  # Передаем температуру как компонент
+                gigachat_plus_temperature  # Передаем температуру как компонент
             ], 
             outputs=[prompt_display, output_text_gpt4o, output_text_gigachat_pro, output_text_gigachat_lite, output_text_gigachat_plus]
         )
 
 
-    with gr.Row():
-        personalize_btn = gr.Button("2. Выполнить персонализацию (нажимать только после кнопки 1)", elem_id="personalize_button")
-        
-    with gr.Row():
-        personalize_btn.click(
-            personalize_messages_with_yield,
-            inputs=[output_text_gpt4o, output_text_gigachat_pro, output_text_gigachat_lite, output_text_gigachat_plus] + selections,
-            outputs=[
-                gr.Textbox(label="Промпт для персонализации", lines=6, interactive=False),
-                personalized_output_text_gpt4o := gr.Textbox(label="Персонализированное сообщение GPT-4o", lines=6, interactive=False),
-                personalized_output_text_gigachat_pro := gr.Textbox(label="Персонализированное сообщение GigaChat-Pro", lines=6, interactive=False),
-                personalized_output_text_gigachat_lite := gr.Textbox(label="Персонализированное сообщение GigaChat-Lite", lines=6, interactive=False),
-                personalized_output_text_gigachat_plus := gr.Textbox(label="Персонализированное сообщение GigaChat-Lite+", lines=6, interactive=False)
-            ]
-        )
-
-    with gr.Row():
-        gr.Markdown("*Комментарий (опционально):*")
-        comment_gpt4o = gr.Textbox(label="", lines=3)
-        comment_gigachat_pro = gr.Textbox(label="", lines=3)
-        comment_gigachat_lite = gr.Textbox(label="", lines=3)
-        comment_gigachat_plus = gr.Textbox(label="", lines=3)
-
-    with gr.Row():
-        gr.Markdown("*Откорректированное сообщение (опционально):*")
-        corrected_gpt4o = gr.Textbox(label="", lines=3)
-        corrected_gigachat_pro = gr.Textbox(label="", lines=3)
-        corrected_gigachat_lite = gr.Textbox(label="", lines=3)
-        corrected_gigachat_plus = gr.Textbox(label="", lines=3)
-
-    
-    # Отдельная строка для кнопок с использованием пустой колонки
-    with gr.Row():
-        gr.Button("Жми 👍 для сохранения удачного SMS в базу =>")
-        save_gpt4o_btn = gr.Button("👍")
-        save_gigachat_pro_btn = gr.Button("👍")
-        save_gigachat_lite_btn = gr.Button("👍")
-        save_gigachat_plus_btn = gr.Button("👍")
-
-        # Привязка кнопок к функциям сохранения
-        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, temperature: 
-                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, temperature),
-            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],  # ОПФ
-                gpt4o_temperature  # Передача температуры
-            ],
-            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, temperature: 
-                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, temperature),
-            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],  # ОПФ
-                gigachat_pro_temperature  # Передача температуры
-            ],
-            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, temperature: 
-                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, temperature),
-            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],  # ОПФ
-                gigachat_lite_temperature  # Передача температуры
-            ],
-            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, temperature: 
-                save_to_github(personalized_message, "GigaChat-Plus", comment, corrected_message, description, advantages, non_personalized_prompt, non_personalized_message, gender, generation, psychotype, business_stage, industry, legal_form, temperature),
+        submit_btn.click(
+            fn=lambda description, advantages, *selected_values, gpt4o_temperature, gigachat_pro_temperature, gigachat_lite_temperature, gigachat_plus_temperature: 
+                generate_messages(description, advantages, *selected_values,
+                                  authenticate_gpt4o(101, gpt4o_temperature),
+                                  authenticate_gigachat('GigaChat-Pro', 68, gigachat_pro_temperature),
+                                  authenticate_gigachat('GigaChat', 68, gigachat_lite_temperature),
+                                  authenticate_gigachat('GigaChat-Plus', 68, gigachat_plus_temperature)),
             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],  # ОПФ
-                gigachat_plus_temperature  # Передача температуры
-            ],
-            outputs=None
-        )
-
-
-    # Использование сохраненных переменных в следующем блоке
-    with gr.Row():
-        check_errors_btn = gr.Button("3. Проверить текст (нажимать только после кнопки 2) - экспериментальная фича, качество пока крайне низкое", elem_id="check_errors_button")
-        
-    with gr.Row():
-        check_errors_btn.click(
-            check_errors_with_yield,
-            inputs=[personalized_output_text_gpt4o, personalized_output_text_gigachat_pro, personalized_output_text_gigachat_lite, personalized_output_text_gigachat_plus],
-            outputs=[
-                gr.Textbox(label="Промпт для проверки текста", lines=6, interactive=False),
-                gr.Textbox(label="Результат проверки GPT-4o", lines=6),
-                gr.Textbox(label="Результат проверки GigaChat-Pro", lines=6),
-                gr.Textbox(label="Результат проверки GigaChat-Lite", lines=6),
-                gr.Textbox(label="Результат проверки GigaChat-Lite+", lines=6)
-            ]
+                description_input, 
+                advantages_input, 
+                *selections,
+                gpt4o_temperature,  # Передаем значение температуры
+                gigachat_pro_temperature,  # Передаем значение температуры
+                gigachat_lite_temperature,  # Передаем значение температуры
+                gigachat_plus_temperature  # Передаем значение температуры
+            ], 
+            outputs=[prompt_display, output_text_gpt4o, output_text_gigachat_pro, output_text_gigachat_lite, output_text_gigachat_plus]
         )
 
-
     demo.launch()