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
@@ -181,79 +181,79 @@ def clean_message(message):
 #    except Exception as e:
 #        return f"Ошибка при обращении к GigaChat-Pro: {e}"
 
-#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}"
+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
+    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: