Update app.py

app.py

@@ -12,6 +12,11 @@ from qwenimage.pipeline_qwenimage_edit_plus import QwenImageEditPlusPipeline
 from qwenimage.transformer_qwenimage import QwenImageTransformer2DModel
 from qwenimage.qwen_fa3_processor import QwenDoubleStreamAttnProcessorFA3
 import requests
+import logging
+
+# Set up logging to suppress print statements in UI
+logging.basicConfig(level=logging.INFO, filename='qwen_image_editor.log', filemode='a')
+logger = logging.getLogger(__name__)
 
 # --- Translation Function ---
 @spaces.GPU
@@ -29,10 +34,10 @@ def translate_albanian_to_english(text, language="en"):
             )
             response.raise_for_status()
             translated = response.json().get("translate", "")
-            print(f"Translation response: {translated}")
+            logger.info(f"Translation response: {translated}")
             return translated
         except Exception as e:
-            print(f"Translation error (attempt {attempt + 1}): {e}")
+            logger.error(f"Translation error (attempt {attempt + 1}): {e}")
             if attempt == 1:
                 raise gr.Error("Translation failed. Please try again.")
     raise gr.Error("Translation failed. Please try again.")
@@ -85,11 +90,7 @@ QUALITY_PROMPT = ", high quality, detailed, vibrant, professional lighting"
 
 # --- Main Inference Function ---
 @spaces.GPU(duration=40)
-def infer(
-    image,
-    prompt,
-    progress=gr.Progress(track_tqdm=True),
-):
+def infer(image, prompt):
     """
     Generates an image using the local Qwen-Image diffusers pipeline.
     """
@@ -118,14 +119,14 @@ def infer(
             elif hasattr(image, "name"):
                 pil_image = Image.open(image.name).convert("RGB")
         except Exception as e:
-            print(f"Error loading image: {e}")
+            logger.error(f"Error loading image: {e}")
             raise gr.Error("Failed to load input image.")
 
     if height == 256 and width == 256:
         height, width = None, None
-    print(f"Calling pipeline with prompt: '{prompt_final}'")
-    print(f"Negative Prompt: '{negative_prompt}'")
-    print(f"Seed: {seed}, Steps: {num_inference_steps}, Guidance: {true_guidance_scale}, Size: {width}x{height}")
+    logger.info(f"Calling pipeline with prompt: '{prompt_final}'")
+    logger.info(f"Negative Prompt: '{negative_prompt}'")
+    logger.info(f"Seed: {seed}, Steps: {num_inference_steps}, Guidance: {true_guidance_scale}, Size: {width}x{height}")
 
     # Generate the image
     output = pipe(
@@ -281,25 +282,6 @@ def create_demo():
         div[aria-label="Input Image"] .gr-button {
             display: none !important; /* Force hide all toolbar elements */
         }
-        .image-container.processing {
-            background: #000000 !important;
-        }
-        .image-container.processing::before {
-            content: "Processing...";
-            color: #FFFFFF !important;
-            opacity: 0.7;
-            font-size: 1rem;
-            font-weight: 600;
-            position: absolute;
-            top: 50%;
-            left: 50%;
-            transform: translate(-50%, -50%);
-            text-align: center;
-            width: auto;
-            max-width: 150px;
-            white-space: nowrap;
-            display: inline-block;
-        }
         input, textarea {
             background: #000000 !important;
             color: #FFFFFF !important;
@@ -355,8 +337,8 @@ def create_demo():
         button[aria-label="Download"]:hover {
             box-shadow: 0 0 12px rgba(255, 255, 255, 0.5) !important;
         }
-        .progress-text, .gr-progress {
-            display: none !important;
+        .progress-text, .gr-progress, .gr-prose, .gr-log {
+            display: none !important; /* Hide all progress bars and logs */
         }
         footer, .gr-button-secondary {
             display: none !important;
@@ -389,11 +371,6 @@ def create_demo():
                 box-shadow: 0 0 8px rgba(255, 255, 255, 0.3) !important;
                 border: 0.5px solid #FFFFFF !important; /* Thinner white border */
             }
-            .image-container.processing::before {
-                font-size: 0.9rem;
-                max-width: 90%;
-                white-space: normal;
-            }
         }
         </style>
         <script>
@@ -406,11 +383,6 @@ def create_demo():
                 console.log(`Container ${container.id || container.className}: width=${width}px, box-shadow=${style.boxShadow}, background=${style.background}, border=${style.border} (Viewport: ${window.innerWidth}px)`);
                 container.setAttribute('data-width', `${width}px`);
             });
-            const processingText = document.querySelectorAll('.image-container.processing::before');
-            processingText.forEach(text => {
-                const style = window.getComputedStyle(text);
-                console.log(`Processing text: width=${style.width}, font-size=${style.fontSize}`);
-            });
             const editButton = document.querySelector('.gr-button-primary');
             if (editButton) {
                 const style = window.getComputedStyle(editButton);
@@ -457,23 +429,12 @@ def create_demo():
                 toolbar.style.display = 'none';
                 console.log(`Forced hide toolbar: ${toolbar.className}`);
             });
-            // Toggle processing state
-            const runButton = document.querySelector('.gr-button-primary');
-            const resultImage = document.querySelector('.image-container[aria-label="Result Image"]');
-            if (runButton && resultImage) {
-                runButton.addEventListener('click', () => {
-                    resultImage.classList.add('processing');
-                    console.log('Processing state added to result_image');
-                });
-                const observer = new MutationObserver(() => {
-                    if (resultImage.querySelector('img')) {
-                        resultImage.classList.remove('processing');
-                        console.log('Processing state removed from result_image');
-                        observer.disconnect();
-                    }
-                });
-                observer.observe(resultImage, { childList: true, subtree: true });
-            }
+            // Hide any technical output containers
+            const outputContainers = document.querySelectorAll('.gr-prose, .gr-log, .progress-text, .gr-progress');
+            outputContainers.forEach(container => {
+                container.style.display = 'none';
+                console.log(`Forced hide output container: ${container.className}`);
+            });
         });
         </script>
         """)
@@ -520,6 +481,6 @@ def create_demo():
     return demo
 
 if __name__ == "__main__":
-    print(f"Gradio version: {gr.__version__}")
+    logger.info(f"Gradio version: {gr.__version__}")
     demo = create_demo()
     demo.queue().launch(share=True)