Update app.py

app.py

@@ -108,11 +108,11 @@ async def optimize_pngs(image_paths: list[str | Path] | str | Path) -> None:
     await gather(*tasks)
 
 
-async def upload_image_to_imgbb(file_path: Path) -> str | None:
+async def upload_image_to_imgbb(file_path: Path, file_type: str = 'png') -> str | None:
     url = f'https://api.imgbb.com/1/upload?key={choice(tokens)}'
     try:
         with file_path.open('rb') as file:
-            files = {'image': (file_path.name, file, 'image/png')}
+            files = {'image': (file_path.name, file, f'image/{file_type}')}
             data = {}
             async with AsyncClient() as client:
                 response = await client.post(url, files=files, data=data, timeout=30)
@@ -124,9 +124,9 @@ async def upload_image_to_imgbb(file_path: Path) -> str | None:
         return None
 
 
-async def upload_image(file_path: Path | str) -> str | None:
+async def upload_image(file_path: Path | str, file_type: str = 'png') -> str | None:
     file_path = Path(file_path)
-    return await upload_image_to_imgbb(file_path)
+    return await upload_image_to_imgbb(file_path, file_type)
 
 
 async def optimize_and_upload(images_urls: list[str] | str, convert: bool = False) -> list[str]:
@@ -138,7 +138,7 @@ async def optimize_and_upload(images_urls: list[str] | str, convert: bool = Fals
     new_images_urls = []
     images_paths = images_paths if not convert else await convert_to_jpegs(images_paths)
     for image_path in images_paths:
-        new_url = await upload_image(image_path)
+        new_url = await upload_image(image_path, 'png' if not convert else 'jpeg')
         if new_url:
             new_images_urls.append(new_url)
         logger.info(f'загружено изображение {image_path} в {new_url}')
@@ -178,7 +178,7 @@ async def optimize_images_endpoint(image_urls: ImageURLs):
 @app.post('/jpegs_by_urls/')
 async def optimize_images_endpoint(image_urls: ImageURLs):
     try:
-        optimized_urls = await optimize_and_upload([str(url) for url in image_urls.urls])
+        optimized_urls = await optimize_and_upload([str(url) for url in image_urls.urls], convert=True)
         return {"optimized_urls": optimized_urls}
     except Exception as e:
         raise HTTPException(status_code=500, detail=str(e))