Update app.py

app.py

@@ -89,6 +89,266 @@ def process(image, progress=gr.Progress()):
     # Return the processed image and the file for download
     return output_array, gr.update(visible=True, value=filepath, interactive=True)
 
+# Gradio interface setup
+title = "Background Removal Tool"
+description = """
+<style>
+    .custom-container {
+        text-align: center;
+        max-width: 800px;
+        margin: 0 auto;
+        padding: 20px;
+        background: rgba(10, 10, 30, 0.6);
+        border-radius: 15px;
+        box-shadow: 0 0 20px rgba(0, 255, 255, 0.2);
+    }
+    .custom-logo {
+        display: flex;
+        align-items: center;
+        justify-content: center;
+        margin-bottom: 2rem;
+        transform: scale(0.8);
+    }
+    .logo-mark {
+        position: relative;
+        width: 60px;
+        height: 60px;
+        margin-right: 15px;
+        animation: logo-spin 12s linear infinite;
+    }
+    .logo-hexagon {
+        position: absolute;
+        width: 100%;
+        height: 100%;
+        background: linear-gradient(135deg, #ff00de, #00ffff);
+        clip-path: polygon(50% 0%, 100% 25%, 100% 75%, 50% 100%, 0% 75%, 0% 25%);
+        animation: pulse-glow 4s infinite alternate;
+    }
+    .logo-inner {
+        position: absolute;
+        top: 50%;
+        left: 50%;
+        transform: translate(-50%, -50%);
+        color: #00ffff;
+        font-family: 'Orbitron', sans-serif;
+        font-weight: 700;
+        font-size: 28px;
+        text-shadow: 0 0 10px rgba(0, 255, 255, 0.5);
+    }
+    .logo-text {
+        font-family: 'Orbitron', sans-serif;
+        font-size: 36px;
+        font-weight: 700;
+        background: linear-gradient(45deg, #ff00de, #00ffff);
+        -webkit-background-clip: text;
+        background-clip: text;
+        -webkit-text-fill-color: transparent;
+        text-shadow: 0 0 10px rgba(255, 0, 222, 0.5);
+        animation: text-glow 2s infinite alternate;
+    }
+    .title-text {
+        color: #00ffff;
+        font-family: 'Orbitron', sans-serif;
+        font-size: 2.5em;
+        margin: 20px 0;
+        text-shadow: 0 0 10px rgba(0, 255, 255, 0.5);
+        animation: title-pulse 2s infinite alternate;
+    }
+    .subtitle-text {
+        color: #a0a0a0;
+        font-family: 'Roboto Mono', monospace;
+        font-size: 1.2em;
+        margin-top: 10px;
+        line-height: 1.5;
+    }
+    @keyframes logo-spin {
+        0% { transform: rotateY(0deg); }
+        100% { transform: rotateY(360deg); }
+    }
+    @keyframes pulse-glow {
+        0% { filter: brightness(1) drop-shadow(0 0 5px #00ffff); }
+        100% { filter: brightness(1.2) drop-shadow(0 0 15px #00ffff); }
+    }
+    @keyframes text-glow {
+        0% { text-shadow: 0 0 5px rgba(255, 0, 222, 0.5); }
+        100% { text-shadow: 0 0 15px rgba(255, 0, 222, 0.8), 0 0 25px rgba(0, 255, 255, 0.5); }
+    }
+    @keyframes title-pulse {
+        0% { text-shadow: 0 0 5px rgba(0, 255, 255, 0.5); }
+        100% { text-shadow: 0 0 15px rgba(0, 255, 255, 0.8), 0 0 25px rgba(0, 255, 255, 0.5); }
+    }
+</style>
+<div class="custom-container">
+    <div class="custom-logo">
+        <div class="logo-mark">
+            <div class="logo-hexagon"></div>
+            <div class="logo-inner">A</div>
+        </div>
+        <div class="logo-text">AicO Tools</div>
+    </div>
+    <h1 class="title-text">AI Background Removal</h1>
+    <p class="subtitle-text">
+        Remove backgrounds instantly using advanced AI technology
+    </p>
+</div>
+"""
+
+# Create the Gradio interface
+with gr.Blocks(css="""
+    /* Import fonts */
+    @import url('https://fonts.googleapis.com/css2?family=Orbitron:wght@400;500;700&family=Roboto+Mono:wght@300;400;700&display=swap');
+    
+    /* Variables */
+    :root {
+        --neon-cyan: #00ffff;
+        --neon-pink: #ff00de;
+        --neon-yellow: #ffdd00;
+        --dark-background: #0a0a1e;
+        --deep-blue: #121238;
+    }
+
+    /* Global styles */
+    body {
+        font-family: 'Roboto Mono', monospace;
+        background: linear-gradient(135deg, var(--dark-background) 0%, var(--deep-blue) 100%);
+        color: #ffffff;
+        min-height: 100vh;
+    }
+
+    /* Container styling */
+    .container {
+        background: rgba(0, 0, 0, 0.6);
+        backdrop-filter: blur(10px);
+        border-radius: 15px;
+        padding: 20px;
+        box-shadow: 0 0 20px rgba(0, 255, 255, 0.2);
+        margin: 20px;
+    }
+
+    /* Input/Output areas */
+    .input-image, .output-image {
+        background: rgba(18, 18, 56, 0.7) !important;
+        border: 2px solid var(--neon-cyan) !important;
+        border-radius: 12px !important;
+        transition: all 0.3s ease !important;
+        overflow: hidden !important;
+    }
+
+    .input-image:hover, .output-image:hover {
+        box-shadow: 0 0 20px rgba(0, 255, 255, 0.4) !important;
+        transform: translateY(-2px) !important;
+    }
+
+    /* Download button styling */
+    .download-container [data-testid="file"] button {
+        background: linear-gradient(45deg, var(--neon-cyan), var(--neon-pink)) !important;
+        color: white !important;
+        border: none !important;
+        padding: 12px 28px !important;
+        font-family: 'Orbitron', sans-serif !important;
+        font-size: 16px !important;
+        font-weight: 600 !important;
+        text-transform: uppercase !important;
+        letter-spacing: 1px !important;
+        border-radius: 8px !important;
+        cursor: pointer !important;
+        transition: all 0.3s ease !important;
+        animation: button-glow 2s infinite alternate !important;
+    }
+
+    .download-container [data-testid="file"] button:hover {
+        transform: translateY(-2px) !important;
+        box-shadow: 0 5px 15px rgba(0, 255, 255, 0.4) !important;
+    }
+
+    /* Labels */
+    label {
+        color: var(--neon-cyan) !important;
+        font-family: 'Orbitron', sans-serif !important;
+        font-size: 1.1em !important;
+        text-shadow: 0 0 5px rgba(0, 255, 255, 0.5) !important;
+        margin-bottom: 8px !important;
+    }
+
+    /* Progress bar */
+    .progress-bar {
+        background: linear-gradient(90deg, var(--neon-cyan), var(--neon-pink)) !important;
+        height: 4px !important;
+        border-radius: 2px !important;
+        box-shadow: 0 0 10px rgba(0, 255, 255, 0.5) !important;
+    }
+
+    /* Additional Animations */
+    @keyframes button-glow {
+        0% { box-shadow: 0 0 5px rgba(0, 255, 255, 0.5); }
+        100% { box-shadow: 0 0 15px rgba(0, 255, 255, 0.8), 0 0 25px rgba(255, 0, 222, 0.5); }
+    }
+""") as demo:
+    gr.Markdown(description)
+    
+    with gr.Row(equal_height=True):
+        with gr.Column(scale=1):
+            input_image = gr.Image(
+                type="numpy",
+                label="Upload Your Image",
+                elem_id="input-image",
+                elem_classes="input-image"
+            )
+        
+        with gr.Column(scale=1):
+            output_image = gr.Image(
+                type="numpy",
+                label="Result",
+                elem_id="output-image",
+                elem_classes="output-image"
+            )
+            
+            with gr.Row(elem_classes="download-container"):
+                download_file = gr.File(
+                    label="",
+                    file_count="single",
+                    interactive=True,
+                    visible=False
+                )
+    
+    # Process automatically when image is uploaded
+    input_image.change(
+        fn=process,
+        inputs=input_image,
+        outputs=[output_image, download_file],
+        show_progress="full"
+    )
+
+if __name__ == "__main__":
+    demo.launch(share=False)    ma = torch.max(result)
+    mi = torch.min(result)
+    result = (result - mi) / (ma - mi)
+    
+    # Convert the result to an image
+    result_array = (result * 255).cpu().data.numpy().astype(np.uint8)
+    pil_mask = Image.fromarray(np.squeeze(result_array))
+    
+    # Add the mask as alpha channel to the original image
+    new_im = orig_image.copy()
+    new_im.putalpha(pil_mask)
+    
+    progress(0.8, desc="Preparing download...")
+    # Generate a unique filename
+    unique_id = str(uuid.uuid4())[:8]
+    filename = f"background_removed_{unique_id}.png"
+    filepath = os.path.join(OUTPUT_DIR, filename)
+    
+    # Save the processed image
+    new_im.save(filepath, format='PNG')
+    
+    # Convert to numpy array for display
+    output_array = np.array(new_im.convert("RGBA"))
+    
+    progress(1.0, desc="Done!")
+    
+    # Return the processed image and the file for download
+    return output_array, gr.update(visible=True, value=filepath, interactive=True)
+
 # Gradio interface setup
 title = "Background Removal Tool"
 description = """