init

app.py

@@ -1,3 +1,4 @@
+
 import os
 import io
 import base64
@@ -128,11 +129,26 @@ body, .gradio-container {
     border: 1px solid #ffeaa7;
     color: #856404;
 }
+.status-processing {
+    background: #cce5ff;
+    border: 1px solid #99ccff;
+    color: #004085;
+}
 .status-error {
     background: #f8d7da;
     border: 1px solid #f5c6cb;
     color: #721c24;
 }
+.queue-info {
+    background: #e8f4fd;
+    border: 1px solid #bee5eb;
+    padding: 12px;
+    border-radius: 8px;
+    margin: 10px 0;
+    text-align: center;
+    font-size: 14px;
+    color: #0c5460;
+}
 """
 
 def image_to_base64(image):
@@ -168,18 +184,18 @@ def try_connect_backend():
         test_client = Client("SnapwearAI/Pattern-Transfer-Backend", hf_token=HF_TOKEN)
         client = test_client
         backend_connected = True
-        return "🟢 Model is ready!", True
+        return "🟢 Backend is ready! You can now generate pattern transfers.", True
     except Exception as e:
         client = None
         backend_connected = False
         error_str = str(e).lower()
         if "timeout" in error_str or "read operation timed out" in error_str:
-            return "🟡 Model is starting up (this takes 5-6 minutes on first load). Please wait and try again.", False
+            return "🟡 Backend is starting up (this takes 5-6 minutes on first load). Please wait and try again.", False
         else:
             return f"🔴 Backend error: {str(e)}", False
 
 def call_backend_with_retry(print_image, product_image, max_retries=3):
-    """Call the backend with proper error handling."""
+    """Call the backend with proper error handling and queue awareness."""
     global client, backend_connected
     
     # Validate inputs
@@ -210,15 +226,26 @@ def call_backend_with_retry(print_image, product_image, max_retries=3):
             
             logger.info("Images converted to base64")
             
-            # Make the backend call
+            # Make the backend call with progress tracking
             start_time = time.time()
-            result = client.predict(
-                print_b64,
-                product_b64,
-                guidance_scale,
-                num_steps,
-                api_name="/predict"
-            )
+            
+            # Add queue position info if available
+            try:
+                result = client.predict(
+                    print_b64,
+                    product_b64,
+                    guidance_scale,
+                    num_steps,
+                    api_name="/predict"
+                )
+            except Exception as prediction_error:
+                # Handle queue-related messages in error
+                error_str = str(prediction_error).lower()
+                if "queue" in error_str or "position" in error_str:
+                    # Extract queue info if present
+                    return None, f"📋 Request queued. {str(prediction_error)}"
+                else:
+                    raise prediction_error
             
             processing_time = time.time() - start_time
             logger.info(f"Backend call completed in {processing_time:.2f}s")
@@ -231,6 +258,9 @@ def call_backend_with_retry(print_image, product_image, max_retries=3):
                     result_image = base64_to_image(result_b64)
                     if result_image:
                         logger.info("Successfully received and decoded result image")
+                        # Add processing time to status if not already present
+                        if "Generated in" not in status:
+                            status = f"{status} (Total time: {processing_time:.1f}s)"
                         return result_image, status
                     else:
                         return None, "❌ Failed to decode result image"
@@ -245,29 +275,31 @@ def call_backend_with_retry(print_image, product_image, max_retries=3):
                 # Backend might be starting up again
                 backend_connected = False
                 client = None
-                return None, "🟡 Model timed out. It may be starting up again. Please wait 5-6 minutes and try the 'Check Status' button."
+                return None, "🟡 Backend timed out. It may be starting up or busy with other requests. Please try again in a few moments."
+            elif "queue" in error_str or "busy" in error_str:
+                return None, f"📋 Server is busy processing other requests. Please wait and try again. {str(e)}"
             
             logger.error(f"Backend call attempt {attempt + 1} failed: {e}")
             if attempt == max_retries - 1:
                 return None, f"❌ Backend error: {str(e)}"
-            time.sleep(2)  # Wait before retry
+            time.sleep(3)  # Wait before retry
     
     return None, "❌ All attempts failed"
 
 # ─────────  Main UI ─────────
-with gr.Blocks(css=css, title="Pattern & Color Transfer Studio") as demo:
+with gr.Blocks(css=css, title="AI Style Transfer Studio - Pattern & Color Transfer") as demo:
     
     # ──────── Hero Section ────────
     gr.HTML("""
         <div class="hero-section">
             <h1 style="font-size:48px;margin:0;background:linear-gradient(45deg,#fff,#f0f8ff);-webkit-background-clip:text;-webkit-text-fill-color:transparent;">
-                🎨 Snapwear Texture Transfer Studio
+                🎨 AI Style Transfer Studio
             </h1>
             <h2 style="font-size:24px;margin:10px 0;opacity:0.9;">
-                Transfer Any Pattern onto Any Product Instantly
+                Transform Any Pattern onto Any Product Instantly
             </h2>
             <p style="font-size:18px;margin:15px 0;opacity:0.8;">
-                • Instant results • Perfect for designers, brands & creators
+                Professional-grade AI pattern transfer • Instant results • Perfect for designers, brands & creators
             </p>
             <div class="social-links">
                 <a href="https://snapwear.io" target="_blank">🌐 Official Website</a>
@@ -282,24 +314,33 @@ with gr.Blocks(css=css, title="Pattern & Color Transfer Studio") as demo:
         with gr.Column():
             # Initial status message
             if backend_connected:
-                initial_status = '<div class="status-banner status-ready">🟢 Model is ready! You can generate pattern transfers.</div>'
+                initial_status = '<div class="status-banner status-ready">🟢 Backend is ready! You can generate pattern transfers.</div>'
             else:
-                initial_status = '<div class="status-banner status-starting">🟡 Model may be starting up. Click "Check Status" to verify.</div>'
+                initial_status = '<div class="status-banner status-starting">🟡 Backend may be starting up. Click "Check Status" to verify.</div>'
             
             status_display = gr.HTML(value=initial_status)
             
             # Status check button
-            check_status_btn = gr.Button("🔄 Check Status", size="sm")
+            check_status_btn = gr.Button("🔄 Check Backend Status", size="sm")
+    
+    # ──────── Queue Info Section ────────
+    gr.HTML("""
+        <div class="queue-info">
+            <strong>ℹ️ Multiple Users:</strong> If others are using the system, your request will be queued. 
+            Processing time is 30-60 seconds per request. You'll see queue position updates during processing.
+        </div>
+    """)
     
     # ──────── Info Box ────────
     gr.HTML("""
         <div style="background:#e8f4fd;border:1px solid #bee5eb;border-radius:12px;padding:20px;margin:20px 0;">
-            <h3 style="color:#0c5460;margin:0 0 10px 0;">ℹ️ First Time Here?</h3>
-            <p style="color:#0c5460;margin:0;">
-                Our AI model takes 5-6 minutes to start up on first visit after being idle. 
-                Use the "Check Status" button above to see when the system is ready. 
-                Once ready, pattern transfers complete in 30-60 seconds!
-            </p>
+            <h3 style="color:#0c5460;margin:0 0 10px 0;">ℹ️ How It Works</h3>
+            <div style="color:#0c5460;margin:0;">
+                <p><strong>First Time:</strong> Backend takes 5-6 minutes to start up after being idle.</p>
+                <p><strong>Multiple Users:</strong> Requests are processed one at a time to ensure quality. You'll be queued if others are using the system.</p>
+                <p><strong>Processing Time:</strong> 30-60 seconds per request once processing begins.</p>
+                <p><strong>Queue Updates:</strong> You'll see your position and estimated wait time.</p>
+            </div>
         </div>
     """)
 
@@ -307,16 +348,20 @@ with gr.Blocks(css=css, title="Pattern & Color Transfer Studio") as demo:
     gr.HTML("""
         <div style="display:grid;grid-template-columns:repeat(auto-fit,minmax(250px,1fr));gap:20px;margin:30px 0;">
             <div class="feature-box">
-                <h3>🚀 Instant Transfer</h3>
-                <p>Apply any pattern to any product in 30-60 seconds with professional results</p>
+                <h3>🚀 Smart Queuing</h3>
+                <p>Automatic request queuing ensures stable results even with multiple simultaneous users</p>
             </div>
             <div class="feature-box">
                 <h3>🎯 Perfect Mapping</h3>
-                <p>Preserve product shape, lighting, and texture for realistic results</p>
+                <p>AI preserves product shape, lighting, and texture for realistic results</p>
+            </div>
+            <div class="feature-box">
+                <h3>💎 Professional Quality</h3>
+                <p>Exhibition-grade 4K/8K outputs ready for commercial use</p>
             </div>
             <div class="feature-box">
-                <h3>🎨 Endless Possibilities</h3>
-                <p>Transfer prints, patterns, textures, and colors across any product type</p>
+                <h3>🎨 Reliable Processing</h3>
+                <p>Single-threaded processing prevents errors and ensures consistent quality</p>
             </div>
         </div>
     """)
@@ -339,16 +384,16 @@ with gr.Blocks(css=css, title="Pattern & Color Transfer Studio") as demo:
                 type="pil",
                 height=400,
             )
-            gr.HTML('<p style="text-align:center;color:#666;font-size:14px;">Upload any color, print, or design you want to transfer</p>')
+            gr.HTML('<p style="text-align:center;color:#666;font-size:14px;">Upload any pattern, print, texture, or design you want to transfer</p>')
             
             # Print examples
             if os.path.exists("Assets/print"):
-                print_examples = [os.path.join("Assets/print", f) for f in os.listdir("Assets/print")][:10]
+                print_examples = [os.path.join("Assets/print", f) for f in os.listdir("Assets/print")][:6]
                 if print_examples:
                     gr.Examples(
                         label="✨ Example Patterns",
                         inputs=print_image,
-                        examples_per_page=10,
+                        examples_per_page=6,
                         examples=print_examples,
                     )
 
@@ -363,12 +408,12 @@ with gr.Blocks(css=css, title="Pattern & Color Transfer Studio") as demo:
             
             # Product examples
             if os.path.exists("Assets/product"):
-                product_examples = [os.path.join("Assets/product", f) for f in os.listdir("Assets/product")][:12]
+                product_examples = [os.path.join("Assets/product", f) for f in os.listdir("Assets/product")][:6]
                 if product_examples:
                     gr.Examples(
                         label="📦 Example Products",
                         inputs=product_image,
-                        examples_per_page=12,
+                        examples_per_page=6,
                         examples=product_examples,
                     )
 
@@ -380,7 +425,7 @@ with gr.Blocks(css=css, title="Pattern & Color Transfer Studio") as demo:
                 height=400
             )
 
-            # Status display
+            # Status display with queue info
             status_text = gr.Text(
                 label="Generation Status",
                 interactive=False,
@@ -394,6 +439,13 @@ with gr.Blocks(css=css, title="Pattern & Color Transfer Studio") as demo:
                 size="lg",
                 variant="primary"
             )
+            
+            # Queue status info
+            gr.HTML("""
+                <div style="font-size:12px;color:#666;text-align:center;margin-top:10px;">
+                    💡 If busy, you'll be automatically queued and see position updates
+                </div>
+            """)
 
     # ──────── Showcase Examples ────────
     gr.HTML("""
@@ -414,13 +466,12 @@ with gr.Blocks(css=css, title="Pattern & Color Transfer Studio") as demo:
                 [os.path.join("Assets/examples", "1_product.jpg"), os.path.join("Assets/examples", "1_print.jpg"), os.path.join("Assets/examples", "1_result.jpg")],
                 [os.path.join("Assets/examples", "2_product.jpg"), os.path.join("Assets/examples", "2_print.jpg"), os.path.join("Assets/examples", "2_result.jpg")],
                 [os.path.join("Assets/examples", "3_product.jpg"), os.path.join("Assets/examples", "3_print.jpg"), os.path.join("Assets/examples", "3_result.jpg")],
-                [os.path.join("Assets/examples", "4_product.jpg"), os.path.join("Assets/examples", "4_print.jpg"), os.path.join("Assets/examples", "4_result.jpg")],
             ]
             pattern_showcase = gr.Examples(
                 examples=showcase_examples,
                 inputs=[product_image, print_image, result_img],
                 label="Pattern Transfer Examples - Click any example to try it yourself!",
-                examples_per_page=4,
+                examples_per_page=3,
             )
     except:
         gr.HTML("<p style='text-align:center;color:#666;'>Pattern transfer examples will appear here once example files are added to Assets/examples/</p>")
@@ -430,20 +481,19 @@ with gr.Blocks(css=css, title="Pattern & Color Transfer Studio") as demo:
         gr.HTML('<h3 style="text-align:center;color:#764ba2;margin:20px 0;">🌈 Color Transfer Showcase</h3>')
     
     try:
-        if os.path.exists("Assets/examples/color"):
+        if os.path.exists("Assets/color_examples"):
             color_examples = [
-                [os.path.join("Assets/examples/color", "1_product.jpg"), os.path.join("Assets/examples/color", "1_print.jpg"), os.path.join("Assets/examples/color", "1_result.jpg")],
-                [os.path.join("Assets/examples/color", "2_product.jpg"), os.path.join("Assets/examples/color", "2_print.jpg"), os.path.join("Assets/examples/color", "2_result.jpg")],
-                [os.path.join("Assets/examples/color", "3_product.jpg"), os.path.join("Assets/examples/color", "3_print.jpg"), os.path.join("Assets/examples/color", "3_result.jpg")],
+                [os.path.join("Assets/color_examples", "1_product.jpg"), os.path.join("Assets/color_examples", "1_color.jpg"), os.path.join("Assets/color_examples", "1_result.jpg")],
+                [os.path.join("Assets/color_examples", "2_product.jpg"), os.path.join("Assets/color_examples", "2_color.jpg"), os.path.join("Assets/color_examples", "2_result.jpg")],
             ]
             color_showcase = gr.Examples(
                 examples=color_examples,
                 inputs=[product_image, print_image, result_img],
                 label="Color Transfer Examples - Perfect for recoloring products!",
-                examples_per_page=3,
+                examples_per_page=2,
             )
     except:
-        gr.HTML("<p style='text-align:center;color:#666;'>Color transfer examples will appear here once example files are added to Assets/examples/color/</p>")
+        gr.HTML("<p style='text-align:center;color:#666;'>Color transfer examples will appear here once example files are added to Assets/color_examples/</p>")
 
     # ──────── Use Cases ────────
     gr.HTML("""
@@ -470,6 +520,19 @@ with gr.Blocks(css=css, title="Pattern & Color Transfer Studio") as demo:
         </div>
     """)
 
+    # ──────── Technical Details ────────
+    gr.HTML("""
+        <div style="background:#f8fafc;border:1px solid #e2e8f0;padding:20px;border-radius:12px;margin:20px 0;">
+            <h3 style="color:#333;text-align:center;margin-bottom:15px;">🔧 Technical Details</h3>
+            <div style="font-size:14px;color:#666;">
+                <p><strong>Queue System:</strong> Requests are processed sequentially to prevent GPU memory conflicts and ensure stable results.</p>
+                <p><strong>Concurrency Protection:</strong> Single-worker processing eliminates the "index out of bounds" errors that occur with parallel processing.</p>
+                <p><strong>Timeout Protection:</strong> Requests timeout after 10 minutes to prevent infinite waiting.</p>
+                <p><strong>State Isolation:</strong> Each request gets its own random seed and generator to prevent interference.</p>
+            </div>
+        </div>
+    """)
+
     # ──────── Event Handlers ────────
     def update_status_display():
         """Check backend status and update display"""
@@ -491,13 +554,13 @@ with gr.Blocks(css=css, title="Pattern & Color Transfer Studio") as demo:
         outputs=[status_display]
     )
     
-    # Generate button click
+    # Generate button click with enhanced progress tracking
     generate_btn.click(
         fn=call_backend_with_retry,
         inputs=[print_image, product_image],
         outputs=[result_img, status_text],
         show_progress="full",
-        concurrency_limit=8,
+        concurrency_limit=1,  # Ensure only one generation at a time on frontend too
     )
 
     # ──────── Footer ────────
@@ -506,7 +569,7 @@ with gr.Blocks(css=css, title="Pattern & Color Transfer Studio") as demo:
             <h3 style="color:#333;">🚀 Powered by Snapwear AI</h3>
             <p style="color:#666;">
                 Transform your creative vision with professional-grade AI technology.<br/>
-                Join thousands of designers, brands, and creators already using our tools.
+                Now with smart queuing for reliable multi-user support.
             </p>
             <div class="social-links">
                 <a href="https://snapwear.io" target="_blank">🌐 Website</a>
@@ -522,7 +585,7 @@ with gr.Blocks(css=css, title="Pattern & Color Transfer Studio") as demo:
 if __name__ == "__main__":
     demo.queue(
         max_size=20,
-        default_concurrency_limit=8,
+        default_concurrency_limit=1,  # Single concurrent request to match backend
         api_open=False
     ).launch(
         server_name="0.0.0.0",