Update routers/textclas.py

routers/textclas.py

@@ -1,44 +1,43 @@
-from fastapi import APIRouter, Query, HTTPException
-from keybert import KeyBERT
-
-# Inicializa o modelo KeyBERT com DistilBERT (mais rápido que BERT completo)
-kw_model = KeyBERT(model='distilbert-base-nli-mean-tokens')
+from fastapi import APIRouter, HTTPException, Query
+from fastapi.responses import JSONResponse
+from PIL import Image
+import requests
+import base64
+from io import BytesIO
 
+# Criação do roteador
 router = APIRouter()
 
-@router.get("/keywords/extract/")
-def extract_keywords(
-    text: str = Query(..., description="Texto para extração de palavras-chave"),
-    num_keywords: int = Query(5, description="Número de palavras-chave a serem retornadas", ge=1, le=20)
+@router.get("/image/base64/")
+def get_image_as_base64(
+    image_url: str = Query(..., description="URL da imagem para ser processada")
 ):
     """
-    Extrai palavras-chave relevantes de um texto.
+    Recebe uma URL de imagem, converte para base64 e retorna no navegador.
     """
     try:
-        # Extrai palavras-chave
-        keywords = kw_model.extract_keywords(
-            text,
-            keyphrase_ngram_range=(1, 2),
-            stop_words='english',
-            top_n=num_keywords
-        )
+        # Baixar a imagem da URL
+        response = requests.get(image_url)
+        if response.status_code != 200:
+            raise HTTPException(
+                status_code=400, detail="Não foi possível baixar a imagem da URL fornecida."
+            )
+        
+        # Abrir a imagem com Pillow
+        image = Image.open(BytesIO(response.content))
         
-        # Formata o retorno
-        keyword_list = [kw[0] for kw in keywords]
+        # Converter a imagem para base64
+        buffered = BytesIO()
+        image.save(buffered, format=image.format if image.format else "PNG")
+        img_base64 = base64.b64encode(buffered.getvalue()).decode("utf-8")
         
-        return {
-            "text": text,
-            "num_keywords": num_keywords,
-            "keywords": keyword_list
-        }
+        # Retornar a imagem em base64
+        return JSONResponse(content={
+            "image_url": image_url,
+            "image_base64": f"data:image/{image.format.lower()};base64,{img_base64}"
+        })
     
-    except ValueError as ve:
-        raise HTTPException(
-            status_code=400,
-            detail=f"Invalid input: {str(ve)}"
-        )
+    except requests.exceptions.RequestException as e:
+        raise HTTPException(status_code=400, detail=f"Erro ao acessar a URL: {str(e)}")
     except Exception as e:
-        raise HTTPException(
-            status_code=500,
-            detail=f"An error occurred during keyword extraction: {str(e)}"
-        )
+        raise HTTPException(status_code=500, detail=f"Erro interno do servidor: {str(e)}")