better structures

app.py

@@ -1,29 +1,16 @@
 import os
-import gradio as gr
-from huggingface_hub import InferenceClient
-from PIL import Image
-from io import BytesIO
-
-import os
+import sys
 import traceback
+from io import BytesIO
+from PIL import Image
 import gradio as gr
+from huggingface_hub import InferenceClient
 
-def main():
-    try:
-        # seu código normal: init, load model, pipeline, interface
-        ...
-        demo = gr.Interface(...)  # ou Blocks
-        demo.launch(enable_queue=True, debug=True)
-    except Exception as e:
-        print("Error Int:", file=sys.stderr)
-        traceback.print_exc()
-        sys.exit(1)
-
-if __name__ == "__main__":
-    main()
 
+# ============================
+#   TOKEN / CLIENTE
+# ============================
 
-# Checa token
 hf_token = os.environ.get("HF_TOKEN")
 token_missing = hf_token is None
 
@@ -32,18 +19,24 @@ client = None if token_missing else InferenceClient(
     api_key=hf_token,
 )
 
+
+# ============================
+#   FUNÇÕES PRINCIPAIS
+# ============================
+
 def aspect_ratio_preview(img, max_dim=256):
     img_copy = img.copy()
     img_copy.thumbnail((max_dim, max_dim))
     return img_copy
 
+
 def generate_images(img1, img2, img3, prompt="", n_outputs=1, aspect_ratio="4:3"):
     if token_missing:
         return []
 
     images = [img for img in [img1, img2, img3] if img is not None]
     if not images:
-        return [Image.new("RGB", (256,256), color="gray")]
+        return [Image.new("RGB", (256, 256), "gray")]
 
     n_outputs = min(int(n_outputs), 3)
     outputs = []
@@ -58,104 +51,87 @@ def generate_images(img1, img2, img3, prompt="", n_outputs=1, aspect_ratio="4:3"
                 model="Qwen/Qwen-Image-Edit"
             )
         except Exception:
-            image = Image.new("RGB", (256,256), color="gray")
+            traceback.print_exc()
+            image = Image.new("RGB", (256, 256), "gray")
+
         outputs.append(aspect_ratio_preview(image))
 
     return outputs
 
+
+# ============================
+#   INTERFACE GRADIO
+# ============================
+
 with gr.Blocks(title="Qwen Image Multi-Edit By Chalenger") as demo:
 
     if token_missing:
-        gr.Markdown(
-            "⚠️ **HF_TOKEN not set!** Go to Settings → Secrets in your HF Space and add it to use the generator."
-        )
+        gr.Markdown("⚠️ **HF_TOKEN not set!** Adicione o token em *Settings → Secrets*.")
+
+    gr.Markdown("## Multi-Image Editor for Qwen Image Edit — Max 3 Images")
 
-    gr.Markdown("## Multi-Image Editor for Qwen Image Edit -- Max 3 Images")
-    
     with gr.Row():
         img1 = gr.Image(label="Image Upload 1", type="pil")
         img2 = gr.Image(label="Image Upload 2", type="pil")
         img3 = gr.Image(label="Image Upload 3", type="pil")
-    
-    prompt = gr.Textbox(label="Prompt (optional)", placeholder="Description / Descrição", lines=2)
-    
+
+    prompt = gr.Textbox(
+        label="Prompt (opcional)",
+        placeholder="Description / Descrição",
+        lines=2
+    )
+
     nout = gr.Dropdown(
         label="Outputs / Saída",
-        choices=[1,2,3],
+        choices=[1, 2, 3],
         value=1
     )
 
     aspect_ratio_state = gr.State("4:3")
+
     with gr.Row():
         btn_4_3 = gr.Button("📺 4:3 CRT")
         btn_16_9 = gr.Button("🖥️ 16:9 LED")
         btn_9_16 = gr.Button("📱 9:16 Phone")
         btn_1_1 = gr.Button("⬛ 1:1 Square")
 
-    btn_4_3.click(lambda: "4:3", outputs=aspect_ratio_state)
-    btn_16_9.click(lambda: "16:9", outputs=aspect_ratio_state)
-    btn_9_16.click(lambda: "9:16", outputs=aspect_ratio_state)
-    btn_1_1.click(lambda: "1:1", outputs=aspect_ratio_state)
+    btn_4_3.click(lambda: "4:3", None, aspect_ratio_state)
+    btn_16_9.click(lambda: "16:9", None, aspect_ratio_state)
+    btn_9_16.click(lambda: "9:16", None, aspect_ratio_state)
+    btn_1_1.click(lambda: "1:1", None, aspect_ratio_state)
 
-    # Botões Gradio reais
     with gr.Row():
         play_btn = gr.Button("▶️ Start Engine")
         stop_btn = gr.Button("⏹️ Kill Tasks")
 
-    gallery = gr.Gallery(label="Outputs / Saídas", columns=2, height="500")
-
-    # HTML invisível para LEDs
-    js_leds = gr.HTML(
-        """
-        <script>
-        const playBtnEl = document.querySelector('button:contains("▶️ Gerar")');
-        const stopBtnEl = document.querySelector('button:contains("⏹️ Parar")');
-
-        // Cria LED para Gerar
-        const ledPlay = document.createElement('span');
-        ledPlay.style.cssText = 'width:12px; height:12px; background:#400; display:inline-block; margin-right:8px; border-radius:2px;';
-        playBtnEl.prepend(ledPlay);
-
-        // Cria LED para Parar
-        const ledStop = document.createElement('span');
-        ledStop.style.cssText = 'width:12px; height:12px; background:#400; display:inline-block; margin-right:8px; border-radius:2px;';
-        stopBtnEl.prepend(ledStop);
-
-        let blinkInterval = null;
-
-        playBtnEl.addEventListener('click', () => {
-            if(blinkInterval) clearInterval(blinkInterval);
-            blinkInterval = setInterval(() => {
-                ledPlay.style.background = ledPlay.style.background === 'red' ? '#800' : 'red';
-            }, 500);
-            ledStop.style.background = '#400';
-        });
-
-        stopBtnEl.addEventListener('click', () => {
-            if(blinkInterval) clearInterval(blinkInterval);
-            ledStop.style.background = 'red';
-            ledPlay.style.background = '#400';
-        });
-
-        if (%s){
-            playBtnEl.disabled = true;
-            stopBtnEl.disabled = true;
-            playBtnEl.title = 'HF_TOKEN not set! Add it in Secrets to enable.';
-            stopBtnEl.title = 'HF_TOKEN not set! Add it in Secrets to enable.';
-        }
-        </script>
-        """ % ("true" if token_missing else "false"),
-        visible=False
+    gallery = gr.Gallery(
+        label="Outputs / Saídas",
+        columns=2,
+        height="500"
     )
 
-    # Conecta botão Gradio à função
     play_btn.click(
-        generate_images,
+        fn=generate_images,
         inputs=[img1, img2, img3, prompt, nout, aspect_ratio_state],
-        outputs=gallery
+        outputs=gallery,
+        queue=True
     )
 
-    stop_btn.click(lambda: None, inputs=None, outputs=gallery)
+    stop_btn.click(lambda: None, None, gallery)
+
+
+# ============================
+#   LAUNCH CORRETO
+# ============================
 
 if __name__ == "__main__":
-    demo.launch(server_name="0.0.0.0", server_port=7860)
+    try:
+        demo.launch(
+            server_name="0.0.0.0",
+            server_port=7860,
+            enable_queue=True,
+            debug=True
+        )
+    except Exception:
+        traceback.print_exc()
+        sys.exit(1)