Update app.py

app.py

@@ -4,7 +4,7 @@ import os
 import pandas as pd
 import time
 from langchain.schema import SystemMessage
-#from langchain_community.chat_models.gigachat import GigaChat
+from langchain_community.chat_models.gigachat import GigaChat
 from openpyxl import load_workbook
 import plotly.graph_objects as go
 import random
@@ -29,25 +29,25 @@ client_mistral_check = Mistral(api_key=MISTRAL_API_KEY)
 morph = pymorphy3.MorphAnalyzer()
 
 # Авторизация в GigaChat Pro
-#gc_key = os.getenv('GC_KEY')
-#chat_pro = GigaChat(credentials=gc_key, model='GigaChat', max_tokens=68, temperature=1.15, verify_ssl_certs=False)
-#chat_pro = GigaChat(
-#    credentials=gc_key,
-#    model='GigaChat-Pro-preview',
-#    base_url='https://gigachat-preview.devices.sberbank.ru/api/v1/',
-#    max_tokens=68,
-#    temperature=1.15,
-#    verify_ssl_certs=False
-#)
-
-#chat_pro_check = GigaChat(
-#    credentials=gc_key,
-#    model='GigaChat-Pro-preview',
-#    base_url='https://gigachat-preview.devices.sberbank.ru/api/v1/',
-#    max_tokens=3000,
-#    temperature=0.8,
-#    verify_ssl_certs=False
-#)
+gc_key = os.getenv('GC_KEY')
+# chat_pro = GigaChat(credentials=gc_key, model='GigaChat', max_tokens=68, temperature=1.15, verify_ssl_certs=False)
+chat_pro = GigaChat(
+    credentials=gc_key,
+    model='GigaChat-Pro-preview',
+    base_url='https://gigachat-preview.devices.sberbank.ru/api/v1/',
+    max_tokens=68,
+    temperature=1.15,
+    verify_ssl_certs=False
+)
+
+chat_pro_check = GigaChat(
+    credentials=gc_key,
+    model='GigaChat-Pro-preview',
+    base_url='https://gigachat-preview.devices.sberbank.ru/api/v1/',
+    max_tokens=3000,
+    temperature=0.8,
+    verify_ssl_certs=False
+)
 
 approach_stats = {
     "Начни сообщение с призыва к действию с продуктом.": {"failed_checks": defaultdict(int), "total_attempts": 0},
@@ -181,62 +181,79 @@ def clean_message(message):
 #    except Exception as e:
 #        return f"Ошибка при обращении к GigaChat-Pro: {e}"
 
-def generate_message_mistral_generate(prompt, max_retries=5):
-    retries = 0
-    while retries < max_retries:
-        try:
-            chat_response = client_mistral_generate.chat.complete(
-                model="mistral-large-latest",
-                temperature=1.0,
-                min_tokens=81,
-                max_tokens=108,
-                messages=[
-                    {
-                        "role": "user",
-                        "content": prompt
-                    },
-                ]
-            )
-            cleaned_message = clean_message(chat_response.choices[0].message.content.strip())
-            return cleaned_message
-        except Exception as e:
-            error_message = str(e)
-            if "Status 429" in error_message or "Server disconnected without sending a response" in error_message:
-                wait_time = 3
-                print(f"Превышен лимит запросов или сервер не ответил. Ожидание {wait_time} секунд перед повторной попыткой...")
-                time.sleep(wait_time)
-                retries += 1
-            else:
-                print(f"Ошибка при обращении к Mistral: {e}")
-                return None
-    print("Не удалось получить ответ от Mistral после максимального числа попыток.")
-    return None
-
-def generate_message_mistral_check(prompt, max_retries=5):
-    retries = 0
-    while retries < max_retries:
-        try:
-            chat_response = client_mistral_check.chat.complete(
-                model="mistral-large-latest",
-                temperature=0.2,
-                messages=[
-                    {
-                        "role": "user",
-                        "content": prompt
-                    },
-                ]
-            )
-            cleaned_message = clean_message(chat_response.choices[0].message.content.strip())
-            return cleaned_message
-        except Exception as e:
-                if "Status 429" in str(e) or "Server disconnected without sending a response" in str(e):
-                    wait_time = 3  # Можно установить фиксированную задержку
-                    print(f"Превышен лимит запросов. Ожидание {wait_time} секунд перед повторной попыткой...")
-                    time.sleep(wait_time)
-                    retries += 1
-                else:
-                    print(f"Ошибка при обращении к Mistral: {e}")
-                    return None
+#def generate_message_mistral_generate(prompt, max_retries=5):
+def generate_message_mistral_generate(prompt):
+    try:
+        messages = [SystemMessage(content=prompt)]
+        res = chat_pro(messages)
+        cleaned_message = clean_message(res.content.strip())
+        return cleaned_message
+    except Exception as e:
+        return f"Ошибка при обращении к GigaChat-Pro: {e}"
+    
+#    retries = 0
+#    while retries < max_retries:
+#        try:
+#            chat_response = client_mistral_generate.chat.complete(
+#                model="mistral-large-latest",
+#                temperature=1.0,
+#                min_tokens=81,
+#                max_tokens=108,
+#                messages=[
+#                    {
+#                        "role": "user",
+#                        "content": prompt
+#                    },
+#                ]
+#            )
+#            cleaned_message = clean_message(chat_response.choices[0].message.content.strip())
+#            return cleaned_message
+#        except Exception as e:
+#            error_message = str(e)
+#            if "Status 429" in error_message or "Server disconnected without sending a response" in error_message:
+#                wait_time = 3
+#                print(f"Превышен лимит запросов или сервер не ответил. Ожидание {wait_time} секунд перед повторной попыткой...")
+#                time.sleep(wait_time)
+#                retries += 1
+#            else:
+#                print(f"Ошибка при обращении к Mistral: {e}")
+#                return None
+#    print("Не удалось получить ответ от Mistral после максимального числа попыток.")
+#    return None
+
+#def generate_message_mistral_check(prompt, max_retries=5):
+def generate_message_mistral_check(prompt)    
+    try:
+        messages = [SystemMessage(content=prompt)]
+        res2 = chat_pro_check(messages)
+        cleaned_message = clean_message(res2.content.strip())
+        return cleaned_message
+    except Exception as e:
+        return f"Ошибка при обращении к GigaChat-Pro: {e}"
+#    retries = 0
+#    while retries < max_retries:
+#        try:
+#            chat_response = client_mistral_check.chat.complete(
+#                model="mistral-large-latest",
+#                temperature=0.2,
+#                messages=[
+#                    {
+#                        "role": "user",
+#                        "content": prompt
+#                    },
+#                ]
+#            )
+#            cleaned_message = clean_message(chat_response.choices[0].message.content.strip())
+#            return cleaned_message
+#        except Exception as e:
+#                if "Status 429" in str(e) or "Server disconnected without sending a response" in str(e):
+#                    wait_time = 3  # Можно установить фиксированную задержку
+#                    print(f"Превышен лимит запросов. Ожидание {wait_time} секунд перед повторной попыткой...")
+#                    time.sleep(wait_time)
+#                    retries += 1
+#                else:
+#                    print(f"Ошибка при обращении к Mistral: {e}")
+#                    return None
 
 #def generate_check_gigachat_pro(prompt):
 #    try: