Update routers/profanity.py

routers/profanity.py

@@ -3,96 +3,44 @@ from fastapi.responses import StreamingResponse
 from PIL import Image
 import requests
 from io import BytesIO
+import base64
 
 router = APIRouter()
 
-# Charset para representar até 62 cores únicas
-palette_charset = "0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz"
-
-# Utilidades
-def get_image_from_url(url: str) -> Image.Image:
-    response = requests.get(url)
-    if response.status_code != 200:
-        raise HTTPException(status_code=400, detail="Unable to download image.")
-    return Image.open(BytesIO(response.content)).convert("RGB")
-
-def color_to_char(index):
-    return palette_charset[index]
-
-def char_to_index(char):
-    return palette_charset.index(char)
-
-def hex_to_color(hex_code):
-    return tuple(int(hex_code[i:i+2], 16) for i in (1, 3, 5))
-
-# ENCODE
-@router.get("/image/encode/")
-def encode_custom(url: str = Query(...), width: int = 2, height: int = 2):
+def image_to_custom_string(img: Image.Image) -> str:
+    img = img.convert("RGB").resize((32, 32))  # Reduz pra facilitar
+    pixels = list(img.getdata())
+
+    # Representa cada pixel como R,G,B separado por vírgula e ponto-e-vírgula por pixel
+    pixel_data = ";".join([f"{r},{g},{b}" for r, g, b in pixels])
+    return pixel_data
+
+def custom_string_to_image(data: str, size=(32, 32)) -> Image.Image:
+    pixels = [
+        tuple(map(int, pixel.split(",")))
+        for pixel in data.strip().split(";") if pixel
+    ]
+    img = Image.new("RGB", size)
+    img.putdata(pixels)
+    return img
+
+@router.get("/encode")
+def encode_image(url: str = Query(..., description="URL da imagem")):
     try:
-        img = get_image_from_url(url)
-        img = img.resize((width, height))
-        pixels = list(img.getdata())
-
-        # Montar paleta e mapa de índices
-        palette = []
-        index_map = []
-        for pixel in pixels:
-            if pixel not in palette:
-                if len(palette) >= len(palette_charset):
-                    raise HTTPException(status_code=400, detail="Too many unique colors for encoding.")
-                palette.append(pixel)
-            index = palette.index(pixel)
-            index_map.append(color_to_char(index))
-
-        # Codificar string compactada
-        palette_str = ",".join([f"#{r:02X}{g:02X}{b:02X}" for r, g, b in palette])
-        data_str = "".join(index_map)
-        code = f"W{width}H{height}|{palette_str}|{data_str}"
-        compact_code = f"W{width}H{height}|{''.join([color_to_char(i) for i in range(len(palette))])}|{data_str}"
-        return {
-            "code": code,             # Humano-legível
-            "compact_code": compact_code  # Super comprimido
-        }
+        response = requests.get(url)
+        image = Image.open(BytesIO(response.content))
+        encoded_string = image_to_custom_string(image)
+        return {"encoded": encoded_string}
     except Exception as e:
-        raise HTTPException(status_code=500, detail=str(e))
+        raise HTTPException(status_code=400, detail=f"Erro ao processar imagem: {e}")
 
-# DECODE
-@router.get("/image/decode/")
-def decode_custom(code: str = Query(...)):
+@router.get("/decode")
+def decode_image(encoded: str = Query(..., description="String da imagem codificada")):
     try:
-        parts = code.split("|")
-        if len(parts) != 3:
-            raise HTTPException(status_code=400, detail="Invalid code format.")
-
-        size = parts[0]
-        palette_keys = parts[1]
-        pixel_indices = parts[2]
-
-        width = int(size.split("H")[0][1:])
-        height = int(size.split("H")[1])
-
-        # Montar paleta de cores (dummy, associar cores fictícias ou embutidas na versão full)
-        # Aqui está usando uma tabela temporária pra leitura de paleta real
-        # Mas se quiser usar as cores reais na string (como na versão anterior), é só trocar aqui
-        palette = []
-        if "," in palette_keys:
-            # Caso legacy (com hex)
-            palette = [hex_to_color(c) for c in palette_keys.split(",")]
-        else:
-            # Compacto: gerar paleta fictícia baseada nos índices (para testes visuais)
-            base_colors = [(255, 0, 0), (0, 0, 255), (0, 255, 0), (255, 255, 0), (255, 0, 255), (0, 255, 255)]
-            for i in range(len(palette_keys)):
-                palette.append(base_colors[i % len(base_colors)])
-
-        # Montar imagem
-        pixels = [palette[char_to_index(c)] for c in pixel_indices]
-        img = Image.new("RGB", (width, height))
-        img.putdata(pixels)
-
+        image = custom_string_to_image(encoded)
         buffer = BytesIO()
-        img.save(buffer, format="PNG")
+        image.save(buffer, format="PNG")
         buffer.seek(0)
         return StreamingResponse(buffer, media_type="image/png")
-
     except Exception as e:
-        raise HTTPException(status_code=500, detail=f"Decoding failed: {str(e)}")
+        raise HTTPException(status_code=400, detail=f"Erro ao decodificar imagem: {e}")