Update app.py

app.py

@@ -7,6 +7,7 @@ import random
 import os
 import json
 import io
+import uuid
 from gradio_client import Client as client_gradio
 from supabase import create_client, Client
 from datetime import datetime
@@ -60,12 +61,10 @@ def upload_image_to_supabase(image, filename):
     storage_path = f"images/{filename}"
 
     try:
-        # Agora passando os bytes corretamente para o upload
-        response = supabase.storage.from_("images").upload(storage_path, img_bytes.getvalue(), {"content-type": "image/png"})
-        
-        if response.get("error"):
-            raise Exception(f"Erro ao salvar no Supabase: {response['error']}")
+        # Upload da imagem
+        supabase.storage.from_("images").upload(storage_path, img_bytes.getvalue(), {"content-type": "image/png"})
 
+        # Retorna a URL pública
         base_url = f"{url}/storage/v1/object/public/images"
         return f"{base_url}/{filename}"
     except Exception as e:
@@ -105,17 +104,22 @@ def run_lora(prompt, cfg_scale, steps, randomize_seed, seed, width, height, lora
         print(f"❌ Erro ao fazer upload da imagem: {e}")
         image_url = None
 
-    # Salva os metadados no banco de dados Supabase apenas se `image_url` for válido
-    if image_url:
+    # Salva os metadados no banco de dados Supabase
+    try:
         supabase.table("images").insert({
+            "id": str(uuid.uuid4()),  # ID único
             "prompt": prompt,
             "cfg_scale": cfg_scale,
             "steps": steps,
             "seed": seed,
             "lora_scale_1": lora_scale_1,
             "lora_scale_2": lora_scale_2,
-            "image_url": image_url
+            "image_url": image_url,
+            "created_at": datetime.utcnow().isoformat()
         }).execute()
+        print("✅ Metadados salvos no Supabase")
+    except Exception as e:
+        print(f"❌ Erro ao salvar metadados no Supabase: {e}")
 
     return image, seed