teste recuperar imagens

app.py

@@ -6,6 +6,43 @@ import numpy as np
 import cv2
 import base64
 import io
+import os
+import uuid
+import boto3
+from supabase import create_client, Client
+
+# Variaveis de ambiente
+
+# --- AWS S3 ---
+AWS_ACCESS_KEY_ID = os.environ.get("AWS_ACCESS_KEY_ID")
+AWS_SECRET_ACCESS_KEY = os.environ.get("AWS_SECRET_ACCESS_KEY")
+AWS_S3_BUCKET_NAME = os.environ.get("AWS_S3_BUCKET_NAME")
+AWS_S3_REGION = os.environ.get("AWS_S3_REGION")
+
+# --- Supabase ---
+SUPABASE_URL = os.environ.get("SUPABASE_URL")
+SUPABASE_KEY = os.environ.get("SUPABASE_KEY")
+
+# Supabase
+try:
+    supabase: Client = create_client(SUPABASE_URL, SUPABASE_KEY)
+    print("Cliente Supabase inicializado com sucesso.")
+except Exception as e:
+    print(f"ERRO ao inicializar Supabase: {e}")
+    raise
+
+# S3 (boto3)
+try:
+    s3_client = boto3.client(
+        's3',
+        aws_access_key_id=AWS_ACCESS_KEY_ID,
+        aws_secret_access_key=AWS_SECRET_ACCESS_KEY,
+        region_name=AWS_S3_REGION
+    )
+    print("Cliente S3 (boto3) inicializado com sucesso.")
+except Exception as e:
+    print(f"ERRO ao inicializar S3: {e}")
+    raise
 
 app = FastAPI(title="Detector de Corrosão Branca")
 
@@ -90,6 +127,49 @@ def process_image_bytes(img_bytes: bytes):
         "corrosion_image": corrosion_b64,
     }
 
+def image_link_gen(amostra_id: int) -> dict | None:
+    """
+    Busca uma amostra pelo ID no Supabase e gera URLs pré-assinadas para suas imagens.
+    Retorna um dicionário com os links ou None se a amostra não for encontrada.
+    """
+    print(f"\nBuscando amostra com ID: {amostra_id} no Supabase...")
+        
+    # 1. Busca os dados da amostra no banco
+    response = supabase.from_("amostra").select("imagem_original, imagem_resultado").eq("id", amostra_id).single().execute()
+        
+    amostra = response.data
+    if not amostra:
+        print(f"Amostra com ID {amostra_id} não encontrada.")
+        return None
+        
+    print("Amostra encontrada. Gerando links temporários...")
+        
+    # Pega as chaves S3 do resultado
+    s3_key_original = amostra.get("imagem_original")
+    s3_key_resultado = amostra.get("imagem_resultado")
+        
+    links = {}
+        
+    # 2. Gera URL pré-assinada para a imagem original
+    if s3_key_original:
+        url_original = s3_client.generate_presigned_url(
+            'get_object',
+            Params={'Bucket': AWS_S3_BUCKET_NAME, 'Key': s3_key_original},
+            ExpiresIn=3600  # Link válido por 1 hora (3600 segundos)
+        )
+        links['original'] = url_original
+            
+    # 3. Gera URL pré-assinada para a imagem de resultado (se existir)
+    if s3_key_resultado:
+        url_resultado = s3_client.generate_presigned_url(
+            'get_object',
+            Params={'Bucket': AWS_S3_BUCKET_NAME, 'Key': s3_key_resultado},
+            ExpiresIn=3600
+        )
+        links['resultado'] = url_resultado
+            
+    return links
+
 @app.post("/analyze")
 async def analyze(file: UploadFile = File(...)):
     content = await file.read()
@@ -99,6 +179,31 @@ async def analyze(file: UploadFile = File(...)):
         raise HTTPException(status_code=400, detail=str(e))
     return JSONResponse(result)
 
+@app.get("/links/{amostra_id}")
+async def get_link(amostra_id: int)
+    try:
+        
+        links = image_link_gen(amostra_id)
+        
+        if links is None:
+            raise HTTPException(
+                status_code=404, 
+                detail=f"Amostra com ID {amostra_id} não encontrada."
+            )
+        
+        return links
+
+    except Exception as e:
+
+        print(f"Erro inesperado ao processar ID {amostra_id}: {e}")
+        
+        raise HTTPException(
+            status_code=500, 
+            detail=f"Erro interno no servidor ao tentar buscar os links. {e}"
+        )
+    
+
 @app.get("/")
 def read_root():
-    return {"status": "ok"}
+    return {"status": "ok"}
+