Upload main.py

main.py

@@ -45,7 +45,7 @@ logger.add("bot_log.log", rotation="10 MB", level="INFO")
 async def upload_to_bucket(url, user_id, prefix="temp"):
     try:
         async with aiohttp.ClientSession() as session:
-            async with session.get(url) as resp:
+            async with session.get(url, timeout=30) as resp:
                 if resp.status == 200:
                     img_data = await resp.read()
                     file_name = f"{user_id}/{prefix}_{uuid.uuid4()}.png"
@@ -61,15 +61,21 @@ async def upload_to_bucket(url, user_id, prefix="temp"):
     return None, None
 
 
-async def download_file(url):
-    try:
-        async with aiohttp.ClientSession() as session:
-            async with session.get(url) as resp:
-                if resp.status == 200:
-                    data = await resp.read()
-                    return io.BytesIO(data)
-    except Exception as e:
-        logger.error(f"Ошибка скачивания файла для ВК: {e}")
+async def download_file(url, retries=5, delay=2):
+    """Пытается скачать файл из бакета несколько раз с задержкой"""
+    for i in range(retries):
+        try:
+            async with aiohttp.ClientSession() as session:
+                async with session.get(url, timeout=10) as resp:
+                    if resp.status == 200:
+                        data = await resp.read()
+                        if data and len(data) > 0:
+                            return io.BytesIO(data)
+            logger.warning(f"Попытка {i+1} скачивания файла не удалась, ждем {delay} сек...")
+            await asyncio.sleep(delay)
+        except Exception as e:
+            logger.error(f"Ошибка на попытке {i+1} скачивания файла для ВК: {e}")
+            await asyncio.sleep(delay)
     return None
 
 
@@ -269,16 +275,19 @@ async def free_trial_handler(message: Message):
         if res_url_bucket:
             try:
                 file_stream = await download_file(res_url_bucket)
-                photo_vk = await uploader.upload(file_stream, file_name="result.png")
-                await message.answer("Отлично! Твоя фотография готова ✨", attachment=photo_vk)
-
-                promo_text = (
-                    "Теперь ты можешь выбрать один из 10 готовых стилей для фото ✨\n"
-                    "или создать собственный образ в Творческом режиме.\n\n"
-                    "Выбирай, с чего начнём 👇"
-                )
-                await message.answer(promo_text, keyboard=get_main_menu(True))
-                supabase.table("users").update({"free_used": True}).eq("id", message.from_id).execute()
+                if file_stream:
+                    photo_vk = await uploader.upload(file_stream, file_name="result.png")
+                    await message.answer("Отлично! Твоя фотография готова ✨", attachment=photo_vk)
+
+                    promo_text = (
+                        "Теперь ты можешь выбрать один из 10 готовых стилей для фото ✨\n"
+                        "или создать собственный образ в Творческом режиме.\n\n"
+                        "Выбирай, с чего начнём 👇"
+                    )
+                    await message.answer(promo_text, keyboard=get_main_menu(True))
+                    supabase.table("users").update({"free_used": True}).eq("id", message.from_id).execute()
+                else:
+                    await message.answer("Ошибка: не удалось получить файл из хранилища. Попробуйте еще раз позже.")
             except Exception as e:
                 logger.error(f"Ошибка в free_trial VK upload: {e}")
                 await message.answer("Техническая ошибка при отправке фото.")
@@ -490,11 +499,14 @@ async def main_handler(message: Message):
             if res_url_bucket:
                 try:
                     file_stream = await download_file(res_url_bucket)
-                    photo_vk = await uploader.upload(file_stream, file_name="style.png")
-                    await message.answer("Отлично! Твое фото готово ✨", attachment=photo_vk)
-                    supabase.rpc("increment_balance", {"user_id": message.from_id, "amount": -1}).execute()
-                    supabase.table("users").update({"last_style": new_last_style, "status": "main"}).eq("id",
-                                                                                                        message.from_id).execute()
+                    if file_stream:
+                        photo_vk = await uploader.upload(file_stream, file_name="style.png")
+                        await message.answer("Отлично! Твое фото готово ✨", attachment=photo_vk)
+                        supabase.rpc("increment_balance", {"user_id": message.from_id, "amount": -1}).execute()
+                        supabase.table("users").update({"last_style": new_last_style, "status": "main"}).eq("id",
+                                                                                                            message.from_id).execute()
+                    else:
+                        await message.answer("Ошибка: не удалось получить файл из хранилища. Баланс сохранен.")
                 except Exception as e:
                     logger.error(f"VK Style Error: {e}")
                     await message.answer("ВК не принял фото. Баланс сохранен.")