Update app.py

app.py

@@ -20,7 +20,9 @@ from transformers import AutoTokenizer, AutoModel
 from utils import best_text_choice
 
 import httpx
+import aiohttp
 import asyncio
+import ssl
 
 tokenizer = AutoTokenizer.from_pretrained("Sergeyzh/rubert-tiny-turbo")
 model = AutoModel.from_pretrained("Sergeyzh/rubert-tiny-turbo")
@@ -194,22 +196,30 @@ def clean_message(message):
 #def generate_message_mistral_generate(prompt, max_retries=5):
 async def generate_message_mistral_generate(prompt):
     try:
-        async with httpx.AsyncClient() as client:
-            response = await client.post(
+        # Создаём SSL-контекст для отключения проверки сертификатов
+        # В продакшн-среде НЕ РЕКОМЕНДУЕТСЯ использовать verify=False
+        ssl_context = ssl.create_default_context()
+        ssl_context.check_hostname = False
+        ssl_context.verify_mode = ssl.CERT_NONE
+
+        async with aiohttp.ClientSession() as session:
+            async with session.post(
                 "https://gigachat-preview.devices.sberbank.ru/api/v1/",
                 headers={
                     "Authorization": f"Bearer {gc_key}",
                     "Content-Type": "application/json"
                 },
                 json={"messages": [{"role": "system", "content": prompt}]},
-                verify=False  # Отключение проверки SSL-сертификатов
-            )
-            response.raise_for_status()
-            res = response.json()
-            cleaned_message = clean_message(res.get("content", "").strip())
-            return cleaned_message
+                ssl=ssl_context  # Используем ssl=ssl_context для отключения проверки
+            ) as response:
+                response.raise_for_status()
+                res = await response.json()
+                cleaned_message = clean_message(res.get("content", "").strip())
+                return cleaned_message
+    except aiohttp.ClientError as e:
+        return f"HTTP ошибка при обращении к GigaChat-Pro: {e}"
     except Exception as e:
-        return f"Ошибка при обращении к GigaChat-Pro: {e}"
+        return f"Неизвестная ошибка при обращении к GigaChat-Pro: {e}"
     
 #    retries = 0
 #    while retries < max_retries:
@@ -244,22 +254,28 @@ async def generate_message_mistral_generate(prompt):
 #def generate_message_mistral_check(prompt, max_retries=5):
 async def generate_message_mistral_check(prompt):
     try:
-        async with httpx.AsyncClient() as client:
-            response = await client.post(
+        ssl_context = ssl.create_default_context()
+        ssl_context.check_hostname = False
+        ssl_context.verify_mode = ssl.CERT_NONE
+
+        async with aiohttp.ClientSession() as session:
+            async with session.post(
                 "https://gigachat-preview.devices.sberbank.ru/api/v1/",
                 headers={
                     "Authorization": f"Bearer {gc_key}",
                     "Content-Type": "application/json"
                 },
                 json={"messages": [{"role": "system", "content": prompt}]},
-                verify=False  # Отключение проверки SSL-сертификатов
-            )
-            response.raise_for_status()
-            res2 = response.json()
-            cleaned_message = clean_message(res2.get("content", "").strip())
-            return cleaned_message
+                ssl=ssl_context
+            ) as response:
+                response.raise_for_status()
+                res2 = await response.json()
+                cleaned_message = clean_message(res2.get("content", "").strip())
+                return cleaned_message
+    except aiohttp.ClientError as e:
+        return f"HTTP ошибка при обращении к GigaChat-Pro: {e}"
     except Exception as e:
-        return f"Ошибка при обращении к GigaChat-Pro: {e}"
+        return f"Неизвестная ошибка при обращении к GigaChat-Pro: {e}"
 #    retries = 0
 #    while retries < max_retries:
 #        try: