guardiancc commited on
Commit
2d74807
·
verified ·
1 Parent(s): 7e6f61d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -12
app.py CHANGED
@@ -11,8 +11,6 @@ import signal
11
  from typing import Dict, Any, List, Optional
12
  import base64
13
  import io
14
- import aiohttp_swagger
15
- from aiohttp_swagger import setup_swagger
16
  from PIL import Image
17
 
18
  import pillow_avif
@@ -78,10 +76,6 @@ async def websocket_handler(request: web.Request) -> web.WebSocketResponse:
78
  logger.exception("Full traceback:")
79
  return ws
80
 
81
- @aiohttp_swagger.docs(
82
- summary="Upload de imagem",
83
- description="Recebe uma imagem e retorna informações sobre ela",
84
- )
85
  async def handle_upload(request: web.Request) -> web.Response:
86
  """Recebe uma imagem e retorna informações sobre ela."""
87
  engine = request.app['engine']
@@ -89,10 +83,12 @@ async def handle_upload(request: web.Request) -> web.Response:
89
  res = await engine.load_image(data)
90
  return web.json_response(res)
91
 
92
- @aiohttp_swagger.docs(
93
- summary="Modificar imagem",
94
- description="Recebe uma imagem, aplica transformações e retorna no formato webp",
95
- )
 
 
96
  async def handle_modify(request: web.Request) -> web.Response:
97
  """Recebe uma imagem e retorna informações sobre ela."""
98
  engine = request.app['engine']
@@ -116,8 +112,7 @@ async def initialize_app() -> web.Application:
116
  # Configure routes
117
  app.router.add_post("/upload", handle_upload)
118
  app.router.add_post("/modify", handle_modify)
119
-
120
- setup_swagger(app, swagger_url='/', api_version='1.0', title='FacePoke API')
121
 
122
  logger.info("Application routes configured")
123
 
 
11
  from typing import Dict, Any, List, Optional
12
  import base64
13
  import io
 
 
14
  from PIL import Image
15
 
16
  import pillow_avif
 
76
  logger.exception("Full traceback:")
77
  return ws
78
 
 
 
 
 
79
  async def handle_upload(request: web.Request) -> web.Response:
80
  """Recebe uma imagem e retorna informações sobre ela."""
81
  engine = request.app['engine']
 
83
  res = await engine.load_image(data)
84
  return web.json_response(res)
85
 
86
+
87
+ async def handle_hello(request: web.Request) -> web.Response:
88
+ return web.json_response({
89
+ "message": "hello folks!"
90
+ })
91
+
92
  async def handle_modify(request: web.Request) -> web.Response:
93
  """Recebe uma imagem e retorna informações sobre ela."""
94
  engine = request.app['engine']
 
112
  # Configure routes
113
  app.router.add_post("/upload", handle_upload)
114
  app.router.add_post("/modify", handle_modify)
115
+ app.router.add_get("/", handle_hello)
 
116
 
117
  logger.info("Application routes configured")
118