Update app.py

app.py

@@ -73,28 +73,6 @@ def run_lora(prompt, cfg_scale, steps, randomize_seed, seed, width, height, lora
         seed = random.randint(0, MAX_SEED)
     generator = torch.Generator(device="cuda").manual_seed(seed)
 
-    # Moderação de texto
-    moderation_client = client_gradio("duchaba/Friendly_Text_Moderation")
-    result = moderation_client.predict(
-        msg=f"{prompt}",
-        safer=0.02,
-        api_name="/fetch_toxicity_level"
-    )
-
-    if float(json.loads(result[1])['sexual_minors']) > 0.03:
-        print('🔴 Conteúdo não permitido')
-        supabase.table("requests").insert({
-            "prompt": prompt,
-            "cfg_scale": cfg_scale,
-            "steps": steps,
-            "randomized_seed": randomize_seed,
-            "seed": seed,
-            "lora_scale_1": lora_scale_1,
-            "lora_scale_2": lora_scale_2,
-            "moderated": 'true'
-        }).execute()
-        raise gr.Error("🚫 Requisição não autorizada!")
-
     # Aplica os adaptadores LoRA corretamente
     pipe.set_adapters([selected_lora], adapter_weights=[1.0])
 
@@ -139,9 +117,10 @@ def run_lora(prompt, cfg_scale, steps, randomize_seed, seed, width, height, lora
 # Interface Gradio
 gr_theme = os.getenv("THEME")
 with gr.Blocks(theme=gr_theme) as app:
-    gr.Markdown("#Collos Image Generator")
+    gr.Markdown("# Androflux Image Generator")
+    
     with gr.Row():
-        with gr.Column(scale=2):
+        with gr.Column(scale=3):
             prompt = gr.TextArea(label="Prompt", placeholder="Digite um prompt (máx 77 caracteres)", lines=3)
             generate_button = gr.Button("Gerar")
             cfg_scale = gr.Slider(label="CFG Scale", minimum=1, maximum=20, step=0.5, value=3.5)
@@ -150,16 +129,23 @@ with gr.Blocks(theme=gr_theme) as app:
             height = gr.Slider(label="Height", minimum=256, maximum=1024, step=64, value=768)  
             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 (AndroFlux)", minimum=0, maximum=1, step=0.01, value=0.7)
+            lora_scale_2 = gr.Slider(label="LoRA Scale (VitorCollos)", minimum=0, maximum=1, step=0.01, value=1)
+
             selected_lora = gr.Dropdown(label="Selecionar LoRA", choices=["AndroFlux", "VitorCollos"], value="AndroFlux")
-        with gr.Column(scale=2):
+
+        with gr.Column(scale=1):
             result = gr.Image(label="Generated Image")
             gr.Markdown("Gere imagens usando Androflux LoRA e um prompt de texto.")
 
+    # 🔥 Agora os sliders são usados corretamente
     generate_button.click(
         run_lora,
-        inputs=[prompt, cfg_scale, steps, randomize_seed, seed, width, height, lora_scale_1, lora_scale_2, selected_lora],  # Correção aqui
+        inputs=[prompt, cfg_scale, steps, randomize_seed, seed, width, height, lora_scale_1, lora_scale_2, selected_lora],
         outputs=[result, seed],
     )
 
 app.queue()
-app.launch(share=True)  # `share=True` cria um link público
+app.launch(share=True)