Update app.py

app.py

@@ -17,6 +17,7 @@ import string
 import io
 from datetime import datetime, timedelta
 import plotly.express as px
+from gradio_client import Client
 
 morph = pymorphy2.MorphAnalyzer()
 
@@ -32,7 +33,8 @@ client = Together(api_key=TOGETHER_API_KEY)
 # Авторизация в сервисе GigaChat
 chat_pro = GigaChat(credentials=gc_key, model='GigaChat-Pro', max_tokens=68, temperature=1, verify_ssl_certs=False)
 chat_lite = GigaChat(credentials=gc_key, model='GigaChat', max_tokens=68, temperature=1.15, verify_ssl_certs=False)
-chat_plus = GigaChat(credentials=gc_key, model='GigaChat-Plus', max_tokens=68, temperature=1.15, verify_ssl_certs=False)
+# chat_plus = GigaChat(credentials=gc_key, model='GigaChat-Plus', max_tokens=68, temperature=1.15, verify_ssl_certs=False)
+client_vikhr = Client("https://6429e2570d3517d2f7.gradio.live/")
 
 # Загрузка данных из Excel-файла
 try:
@@ -1056,15 +1058,27 @@ def generate_message_gigachat_lite(prompt):
     except Exception as e:
         return f"Ошибка при обращении к GigaChat-Lite: {e}"
 
+#def generate_message_gigachat_plus(prompt):
+#    try:
+#        time.sleep(2)
+#        messages = [SystemMessage(content=prompt)]
+#        res = chat_plus(messages)
+#        cleaned_message = clean_message(res.content.strip())
+#        return cleaned_message
+#    except Exception as e:
+#        return f"Ошибка при обращении к GigaChat-Plus: {e}"
+
 def generate_message_gigachat_plus(prompt):
     try:
-        time.sleep(2)
-        messages = [SystemMessage(content=prompt)]
-        res = chat_plus(messages)
-        cleaned_message = clean_message(res.content.strip())
+        result = client_vikhr.predict(
+            message=prompt,
+            api_name="/chat"
+        )
+        cleaned_message = clean_message(result.strip())
         return cleaned_message
     except Exception as e:
-        return f"Ошибка при обращении к GigaChat-Plus: {e}"
+        return f"Ошибка при обращении к Vikhr-Nemo-12B: {e}"
+
 
 def generate_message_meta_llama_3_1_405b(prompt):
     try:
@@ -1469,34 +1483,6 @@ def personalize_and_save(
 
         save_user_request_to_github(description, advantages, key_message, approach, selected_values)
 
-
-def clear_fields():
-    return (
-        "", "", "", "", "",  # personalized outputs and prompts
-        "", "", "", "", "",  # comment fields
-        "", "", "", "", "",  # corrected message fields
-        "", "", "", "", "",  # анализ персонализации
-    )
-
-
-def clear_personalization_fields():
-    return (
-        "", "", "", "", "",  # personalized outputs
-        "", "", "", "", "",  # comment fields
-        "", "", "", "", "",  # corrected message fields
-        "", "", "", "", "",  # анализ персонализации
-        ""
-    )
-
-def clear_on_change():
-    return (
-        "", "", "", "", "",  # очистка всех полей для промптов и сообщений
-        "", "", "", "", "",  # комментарии
-        "", "", "", "", "",  # откорректированные сообщения
-        "", "", "", "", "",  # результаты проверок
-        "", "", "", "", "",
-    )
-
 def clear_on_change_pers():
     return (
         "", "", "", "", "",  # очистка всех полей для промптов и сообщений
@@ -1505,15 +1491,6 @@ def clear_on_change_pers():
         "", "", "", "", ""  # результаты проверок
     )
 
-def clear_unnecessary_fields():
-    return (
-        "", "", "", "", "",  # personalized outputs and prompts
-        "", "", "", "", "",  # comment fields
-        "", "", "", "", "",  # corrected message fields
-        "", "", "", "", "",  # оставшиеся поля
-        "", "",  # Дополнительное пустое значение
-        "", "", "", "", "",  # поля анализа персонализации
-    )
 
 def clear_outputs(*args):
     return ('',) * len(args)