Update app.py

app.py

@@ -11,7 +11,8 @@ import signal
 from typing import Dict, Any, List, Optional
 import base64
 import io
-
+import aiohttp_swagger
+from aiohttp_swagger import setup_swagger
 from PIL import Image
 
 import pillow_avif
@@ -77,6 +78,10 @@ async def websocket_handler(request: web.Request) -> web.WebSocketResponse:
         logger.exception("Full traceback:")
     return ws
 
+@aiohttp_swagger.docs(
+    summary="Upload de imagem",
+    description="Recebe uma imagem e retorna informações sobre ela",
+)
 async def handle_upload(request: web.Request) -> web.Response:
     """Recebe uma imagem e retorna informações sobre ela."""
     engine = request.app['engine']
@@ -84,6 +89,10 @@ async def handle_upload(request: web.Request) -> web.Response:
     res = await engine.load_image(data)
     return web.json_response(res)
 
+@aiohttp_swagger.docs(
+    summary="Modificar imagem",
+    description="Recebe uma imagem, aplica transformações e retorna no formato webp",
+)
 async def handle_modify(request: web.Request) -> web.Response:
     """Recebe uma imagem e retorna informações sobre ela."""
     engine = request.app['engine']
@@ -107,6 +116,8 @@ async def initialize_app() -> web.Application:
         # Configure routes
         app.router.add_post("/upload", handle_upload)
         app.router.add_post("/modify", handle_modify)
+        
+        setup_swagger(app, swagger_url='/', api_version='1.0', title='FacePoke API')
 
         logger.info("Application routes configured")