Update app.py

app.py

@@ -17,7 +17,6 @@ import string
 import io
 from datetime import datetime, timedelta
 import plotly.express as px
-from gradio_client import Client
 
 morph = pymorphy2.MorphAnalyzer()
 
@@ -33,8 +32,7 @@ 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)
-client_vikhr = Client("https://6429e2570d3517d2f7.gradio.live/")
+chat_plus = GigaChat(credentials=gc_key, model='GigaChat-Plus', max_tokens=68, temperature=1.15, verify_ssl_certs=False)
 
 # Загрузка данных из Excel-файла
 try:
@@ -1058,28 +1056,15 @@ 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:
-        result = client_vikhr.predict(
-            message=prompt,
-            system_prompt="Вы – опытный маркетолог-копирайтер, специализирующийся на создании SMS-сообщений для банковских продуктов. У вас глубокое понимание психологии разных поколений, особенностей коммуникации с людьми разного пола и на разных стадиях развития бизнеса",
-            api_name="/chat"
-        )
-        cleaned_message = clean_message(result.strip())
+        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"Ошибка при обращении к Vikhr-Nemo-12B: {e}"
-
+        return f"Ошибка при обращении к GigaChat-Plus: {e}"
 
 def generate_message_meta_llama_3_1_405b(prompt):
     try: