server.py

import io
import os
import asyncio
import time
from aiohttp import web
from PIL import Image
import aiohttp_cors

countint = int(0)

# Ruta del directorio estático
static_dir = os.path.join(os.path.dirname(__file__), 'static')
local_host = '0.0.0.0' # Cambiar por la dirección IP de tu máquina
# Crear el directorio 'converted' dentro del directorio estático si no existe
converted_dir = os.path.join(static_dir, 'converted')
if not os.path.exists(converted_dir):
    os.makedirs(converted_dir)
# Función para eliminar el archivo después de un minuto
async def delete_file_after_delay(file_path, delay):
    await asyncio.sleep(delay)
    if os.path.exists(file_path):
        os.remove(file_path)
        print(f"Deleted file {file_path}")

async def handle_index(request):
    return web.FileResponse(os.path.join(static_dir, 'index.html'))


# Manejador para la API
# Se espera que la imagen sea enviada por POST
# http://[host]/convert
async def handle_api(request):
    getTime = int(time.time())
    print(f"Time: {getTime}")
    # Obtener la imagen y el formato enviados por POST
    data = await request.post()
    image = data['image'].file.read()
    format = data['format']
    quality = 90

    # Check the size of the uploaded image
    max_size = 10 * 1024 * 1024  # 10MB
    if len(image) > max_size:
        # Reduce the quality if the image size exceeds the limit
        quality = 50
    # Crear un objeto Image
    img = Image.open(io.BytesIO(image))
    
    # Convertir la imagen al formato deseado
    buffer = io.BytesIO()
    if format == 'meme':
        format = 'jpeg'
        quality = 1
        resolution = (150, 150)
        img = img.resize(resolution)

    if format == 'jpeg':
        img.convert('RGB')
    img.save(buffer, format=format.upper(), quality=quality, optimize=True, )
    buffer.seek(0)
    
    # Guardar la imagen convertida en el directorio estático
    output_filename = f'converted_image{getTime}.{format}'
    output_path = os.path.join(static_dir,'converted', output_filename)
    print(f"Saving image to {output_path}")
    with open(output_path, 'wb') as f:
        f.write(buffer.getvalue())
        countint += 1
    
    # Programar la eliminación del archivo después de un minuto
    asyncio.create_task(delete_file_after_delay(output_path, 60))
    
    # Devolver la imagen convertida
    response_data = {'filename': output_filename}
    return web.json_response(response_data)

# Crear la aplicación y configurar las rutas
app = web.Application(client_max_size=1024**2*10)  # Set max size to 10MB
app.add_routes([
    web.get('/', handle_index),
    web.post('/convert', handle_api),
    web.static('/static', static_dir)  # Servir archivos estáticos
])

# Configurar CORS
cors = aiohttp_cors.setup(app, defaults={
    "*": aiohttp_cors.ResourceOptions(
        allow_credentials=True,
        expose_headers="*",
        allow_headers="*",
    )
})

async def cleanup_generated_files(app):
    for filename in os.listdir(static_dir+ '/converted'):
        file_path = os.path.join(static_dir,'converted', filename)
        if os.path.isfile(file_path):
            os.remove(file_path)
            print(f"Deleted file {file_path}")

app.on_cleanup.append(cleanup_generated_files)
app.on_shutdown.append(cleanup_generated_files)

# Añadir CORS a las rutas
for route in list(app.router.routes()):
    cors.add(route)

if __name__ == '__main__':
    host = os.getenv('HOST', local_host)
    port = int(os.getenv('PORT', 8080))
    print(f"Starting server on {host}:{port}")
    web.run_app(app, host=host, port=port)