Update app.py

app.py

@@ -35,10 +35,11 @@ lora_models = {
     },
     "Nanda": {
         "repo": "vcollos/Nanda",
-        "weights": "lora.safetensors",
+        "weights": "nanda",
         "trigger_word": "A photo of Nanda, RAW photo, (hyperrealistic portrait:1.3) of a [man/woman], (detailed eyes:1.2), (skin texture:1.4), (natural lighting:1.1), (soft shadows:1.1), (intricate hair details:1.3), (film grain:0.8), (8k:1.2), (depth of field:1.1), (sharp focus:1.1),"
     }
 }
+
 # Carrega os LoRAs
 for name, details in lora_models.items():
     try:
@@ -51,35 +52,40 @@ for name, details in lora_models.items():
 MAX_SEED = 2**32 - 1
 
 def upload_image_to_supabase(image, filename):
-    """ Faz upload da imagem gerada para o Supabase Storage e retorna a URL pública. """
+    """ Faz upload da imagem para o Supabase Storage e retorna a URL pública. """
     img_bytes = io.BytesIO()
     image.save(img_bytes, format="PNG")
-    img_bytes.seek(0)
+    img_bytes.seek(0)  # Move para o início do arquivo
 
     storage_path = f"images/{filename}"
-    response = supabase.storage.from_("images").upload(storage_path, img_bytes, {"content-type": "image/png"})
-
-    if response.get("error"):
-        raise Exception(f"Erro ao salvar no Supabase: {response['error']}")
 
-    base_url = f"{url}/storage/v1/object/public/images"
-    return f"{base_url}/{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']}")
+
+        base_url = f"{url}/storage/v1/object/public/images"
+        return f"{base_url}/{filename}"
+    except Exception as e:
+        print(f"❌ Erro no upload da imagem: {e}")
+        return None
 
 @spaces.GPU(duration=80)
-def run_lora(prompt, cfg_scale, steps, randomize_seed, seed, width, height, lora_scale_1, lora_scale_2, selected_lora, progress=gr.Progress(track_tqdm=True)):
-    # Define uma seed aleatória se necessário
+def run_lora(prompt, cfg_scale, steps, randomize_seed, seed, width, height, lora_scale_1, lora_scale_2, progress=gr.Progress(track_tqdm=True)):
     if randomize_seed:
         seed = random.randint(0, MAX_SEED)
     generator = torch.Generator(device="cuda").manual_seed(seed)
 
-    # Aplica os adaptadores LoRA corretamente
-    pipe.set_adapters([selected_lora], adapter_weights=[1.0])
+    # Aplica os dois LoRAs combinados
+    pipe.set_adapters(["vgn", "Nanda"], adapter_weights=[lora_scale_1, lora_scale_2])
 
-    # Adiciona o trigger word automaticamente se necessário
-    if selected_lora in lora_models and lora_models[selected_lora]["trigger_word"]:
-        prompt = f"{lora_models[selected_lora]['trigger_word']} {prompt}"
+    # Adiciona trigger words apenas se Nanda estiver ativado
+    if lora_scale_2 > 0:
+        prompt = f"{lora_models['Nanda']['trigger_word']} {prompt}"
 
-    # Gera a imagem com o modelo
+    # Gera a imagem
     image = pipe(
         prompt=prompt,
         num_inference_steps=steps,
@@ -92,7 +98,6 @@ def run_lora(prompt, cfg_scale, steps, randomize_seed, seed, width, height, lora
     # Define um nome único para a imagem
     filename = f"image_{seed}_{datetime.utcnow().strftime('%Y%m%d%H%M%S')}.png"
 
-    # Faz upload da imagem para o Supabase Storage
     try:
         image_url = upload_image_to_supabase(image, filename)
         print(f"✅ Imagem salva no Supabase: {image_url}")
@@ -100,16 +105,17 @@ 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
-    supabase.table("images").insert({
-        "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
-    }).execute()
+    # Salva os metadados no banco de dados Supabase apenas se `image_url` for válido
+    if image_url:
+        supabase.table("images").insert({
+            "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
+        }).execute()
 
     return image, seed
 
@@ -129,20 +135,18 @@ with gr.Blocks(theme=gr_theme) as app:
             randomize_seed = gr.Checkbox(False, label="Randomize seed")
             seed = gr.Slider(label="Seed", minimum=0, maximum=MAX_SEED, step=1, value=556215326)
 
-            # 🔥 Certificando que os sliders estão dentro do bloco correto
-            lora_scale_1 = gr.Slider(label="LoRA Scale (Vga)", minimum=0, maximum=1, step=0.01, value=0.5)
+            # Sliders para os pesos dos LoRAs
+            lora_scale_1 = gr.Slider(label="LoRA Scale (vgn)", minimum=0, maximum=1, step=0.01, value=0.5)
             lora_scale_2 = gr.Slider(label="LoRA Scale (Nanda)", minimum=0, maximum=1, step=0.01, value=1)
 
-            selected_lora = gr.Dropdown(label="Selecionar LoRA", choices=["vgn", "Nanda"], value="Nanda")
-
         with gr.Column(scale=2):
             result = gr.Image(label="Generated Image")
-            gr.Markdown("Gere imagens usando vgn LoRA e um prompt de texto.")
+            gr.Markdown("Gere imagens usando Collos LoRA e um prompt de texto.")
 
-    # 🔥 Agora os sliders são usados corretamente
+    # Botão para gerar imagem combinando os LoRAs
     generate_button.click(
         run_lora,
-        inputs=[prompt, cfg_scale, steps, randomize_seed, seed, width, height, lora_scale_1, lora_scale_2, selected_lora],
+        inputs=[prompt, cfg_scale, steps, randomize_seed, seed, width, height, lora_scale_1, lora_scale_2],
         outputs=[result, seed],
     )