Update app.py

app.py

@@ -1,7 +1,6 @@
 from asyncio import create_subprocess_shell, gather, sleep
 from logging import ERROR, INFO, basicConfig, getLogger
 from pathlib import Path
-from random import choice
 from shutil import rmtree
 from subprocess import CalledProcessError, PIPE
 from typing import Any, List
@@ -12,6 +11,7 @@ from fastapi import FastAPI, HTTPException
 from fastapi.responses import PlainTextResponse
 from httpx import AsyncClient, HTTPStatusError, RequestError
 from pydantic import BaseModel, HttpUrl
+from ytelegraph import TelegraphAPI
 from uvicorn import run as uvicorn_run
 
 need_logging = False
@@ -135,25 +135,44 @@ async def upload_image(file_path: Path | str, file_type: str = 'png') -> str | N
 async def optimize_and_upload(images_urls: list[str] | str, convert: bool = False) -> list[str]:
     images_urls = [images_urls] if isinstance(images_urls, str) else images_urls
     print(f'принятые ссылки в обработку ({len(images_urls)}): {images_urls}')
+
     images_paths = await download_pngs(images_urls)
     if not convert:
         await optimize_pngs(images_paths)
+
     new_images_urls = []
     images_paths = images_paths if not convert else await convert_to_jpegs(images_paths)
-    for image_path in images_paths:
+
+    for old_url, image_path in zip(images_urls, images_paths):
         new_url = await upload_image(image_path, 'png' if not convert else 'jpeg')
         if new_url:
             new_images_urls.append(new_url)
-        print(f'загружено изображение {image_path} в {new_url}')
+            print(f'загружено изображение {image_path} в {new_url}')
+        else:
+            new_images_urls.append(old_url)
+            print(f'не удалось загрузить изображение {image_path}, оставим старую ссылку: {old_url}')
+
         try:
             image_path.unlink()
         except Exception as e:
             print(f'не удалось удалить файл {image_path}: {e}')
+
     print(f'новые ссылки: ({len(new_images_urls)}): {new_images_urls}')
+
     try:
         rmtree(images_paths[0].parent)
     except Exception as e:
         print(f'не удалось удалить файл {images_paths[0].parent}: {e}')
+
+    if convert:
+        try:
+            ph = TelegraphAPI()
+            content = '\n\n'.join([f'![]({url})' for url in new_images_urls])
+            ph_link = ph.create_page_md('Dall-E 3 картинки', content)
+            return ph_link
+        except Exception as e:
+            print(f'не удалось создать страницу в телеграфе: {e}')
+
     return new_images_urls