Update app.py

app.py

@@ -33,7 +33,9 @@ token = os.getenv('GITHUB_TOKEN')
 client_mistral_generate = Mistral(api_key=MISTRAL_API_KEY)
 
 # Клиент для выполнения проверок
-client_mistral_check = Mistral(api_key=MISTRAL_API_KEY)
+client_mistral_check = Mistral(api_key=MISTRAL_API_KEY_2)
+client_mistral_check_2 = Mistral(api_key=MISTRAL_API_KEY_3)
+client_mistral_check_3 = Mistral(api_key=MISTRAL_API_KEY_4)
 
 morph = pymorphy3.MorphAnalyzer()
 
@@ -267,6 +269,80 @@ async def generate_message_mistral_check(prompt, max_retries=5):
     print("Failed to get response from Mistral after maximum retries.")
     return None
 
+async def generate_message_mistral_check_2(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 = await client_mistral_check_2.chat.complete_async(
+                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:
+            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"Rate limit exceeded or server did not respond. Waiting {wait_time} seconds before retrying...")
+                await asyncio.sleep(wait_time)
+                retries += 1
+            else:
+                print(f"Error calling Mistral: {e}")
+                return None
+    print("Failed to get response from Mistral after maximum retries.")
+    return None
+
+async def generate_message_mistral_check_3(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 = await client_mistral_check_3.chat.complete_async(
+                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:
+            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"Rate limit exceeded or server did not respond. Waiting {wait_time} seconds before retrying...")
+                await asyncio.sleep(wait_time)
+                retries += 1
+            else:
+                print(f"Error calling Mistral: {e}")
+                return None
+    print("Failed to get response from Mistral after maximum retries.")
+    return None
+
 #def generate_check_gigachat_pro(prompt):
 #    try:
 #        messages = [SystemMessage(content=prompt)]
@@ -1398,7 +1474,7 @@ async def check_abstract_claims(message, description, benefits):
     
     **Не добавляй никакого дополнительного текста. Перед ответом убедись, что отвечаешь только в формате JSON с закрывающими кавычками и скобками.**'''
 
-    response = await generate_message_mistral_check(prompt)
+    response = await generate_message_mistral_check_2(prompt)
 #    await asyncio.sleep(3)
     print("Mistral response:", response)
     result = parse_json_response(response)
@@ -1477,7 +1553,7 @@ async def check_offensive_phrases(message):
         return None
 
 # 28. Проверка на речевые клише, рекламные штампы и канцеляризмы
-async def check_cliches_and_bureaucratese(message, description, benefits):
+async def check_cliches_and_bureaucratese_3(message, description, benefits):
     print()
     print("Проверка 28: Проверка на речевые клише, рекламные штампы и канцеляризмы")
     print()
@@ -1625,43 +1701,43 @@ async def perform_checks(message, description, key_message, product_name, benefi
 
     # 2. Morphological checks using pymorphy3
     morphological_checks = [
+        ("no_word_repetitions", check_no_word_repetitions),
         ("forbidden_words", check_forbidden_words),
-        ("client_addressing", check_no_greeting),
-        ("promises", check_no_promises),
         ("double_verbs", check_no_double_verbs),
-        ("participles", check_no_participles),
         ("adverbial_participles", check_no_adverbial_participles),
+        ("derived_prepositions", check_no_derived_prepositions),
+        ("multiple_nouns", check_no_multiple_nouns),
+        ("introductory_phrases", check_no_introductory_phrases),
+        ("compound_sentences", check_no_compound_sentences),
         ("superlative_adjectives", check_no_superlative_adjectives),
+        ("client_addressing", check_no_greeting),
+        ("promises", check_no_promises),
+        ("amplifiers", check_no_amplifiers),
+        ("participles", check_no_participles),
         ("passive_voice", check_no_passive_voice),
         ("written_out_ordinals", check_no_written_out_ordinals),
         ("subordinate_clauses_chain", check_no_subordinate_clauses_chain),
         ("repeating_conjunctions", check_no_repeating_conjunctions),
-        ("introductory_phrases", check_no_introductory_phrases),
-        ("amplifiers", check_no_amplifiers),
         ("time_parasites", check_no_time_parasites),
-        ("multiple_nouns", check_no_multiple_nouns),
-        ("derived_prepositions", check_no_derived_prepositions),
-        ("compound_sentences", check_no_compound_sentences),
         ("dates_written_out", check_no_dates_written_out),
-        ("no_word_repetitions", check_no_word_repetitions),
     ]
 
     # 3. LLM checks: check_clickbait_phrases, check_abstract_claims, check_cliches_and_bureaucratese
     llm_checks_group1 = [
-        ("product_name_consistency", check_product_name_consistency),
         ("no_contradictions", check_no_contradictions),
-        ("clickbait_phrases", check_clickbait_phrases),
         ("abstract_claims", check_abstract_claims),
         ("cliches_and_bureaucratese", check_cliches_and_bureaucratese),
-        ("contains_key_message", check_contains_key_message)
+        ("clickbait_phrases", check_clickbait_phrases),
+        ("contains_key_message", check_contains_key_message),
+        ("synonymous_members", check_synonymous_members),
+        ("product_name_consistency", check_product_name_consistency)
     ]
 
     # 4. Remaining LLM checks
     llm_checks_group2 = [
-        ("disconnected_sentences", check_disconnected_sentences),
-        ("synonymous_members", check_synonymous_members),
-        ("specialized_terms", check_specialized_terms),
         ("offensive_phrases", check_offensive_phrases),
+        ("disconnected_sentences", check_disconnected_sentences),
+        ("specialized_terms", check_specialized_terms)
     ]
 
     # Perform morphological checks