Update app.py

app.py

@@ -109,7 +109,6 @@ def generate_message_gigachat_lite(prompt):
 
 def generate_message_gigachat_plus(prompt):
     try:
-        time.sleep(2)
         messages = [SystemMessage(content=prompt)]
         res = chat_plus(messages)
         return res.content.strip()
@@ -161,6 +160,15 @@ def perform_personalization(standard_message, personalization_prompt):
     full_prompt = f"{personalization_prompt}\n\nТекст для адаптации:\n{standard_message}"
     return generate_message_gpt4o(full_prompt)
 
+def perform_personalization_gigachat(standard_message, personalization_prompt, model):
+    full_prompt = f"{personalization_prompt}\n\nТекст для адаптации:\n{standard_message}"
+    if model == "gigachat_pro":
+        return generate_message_gigachat_pro(full_prompt)
+    elif model == "gigachat_lite":
+        return generate_message_gigachat_lite(full_prompt)
+    elif model == "gigachat_plus":
+        return generate_message_gigachat_plus(full_prompt)
+
 # Создание интерфейса Gradio
 with gr.Blocks() as demo:
     gr.Markdown("# Генерация SMS-сообщений по заданным признакам")
@@ -216,9 +224,21 @@ with gr.Blocks() as demo:
         
     with gr.Row():
         personalize_btn.click(
-            lambda standard_message, *selected_values: perform_personalization(standard_message, generate_personalization_prompt(*selected_values)),
+            lambda standard_message, *selected_values: (
+                generate_personalization_prompt(*selected_values),
+                perform_personalization(standard_message, generate_personalization_prompt(*selected_values)),
+                perform_personalization_gigachat(standard_message, generate_personalization_prompt(*selected_values), "gigachat_pro"),
+                perform_personalization_gigachat(standard_message, generate_personalization_prompt(*selected_values), "gigachat_lite"),
+                perform_personalization_gigachat(standard_message, generate_personalization_prompt(*selected_values), "gigachat_plus")
+            ),
             inputs=[output_text_gpt4o] + selections,
-            outputs=[gr.Textbox(label="Персонализированное сообщение GPT-4o", lines=6)]
+            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)
+            ]
         )
 
     demo.launch()