Update app.py

app.py

@@ -319,232 +319,5 @@ with gr.Blocks(css="""
         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 = """
-<div style="text-align: center; max-width: 800px; margin: 0 auto; padding: 20px;">
-    <div class="logo-container">
-        <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 style="color: #00ffff; font-size: 2.5em; margin: 20px 0; text-shadow: 0 0 10px rgba(0, 255, 255, 0.5);">
-        AI Background Removal
-    </h1>
-    <p style="color: #a0a0a0; font-size: 1.2em;">
-        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;
-    }
-
-    /* Logo styling */
-    .logo-container {
-        display: flex;
-        align-items: center;
-        margin-bottom: 2rem;
-        animation: float 3s ease-in-out infinite;
-    }
-
-    .logo-mark {
-        position: relative;
-        width: 40px;
-        height: 40px;
-        margin-right: 12px;
-        animation: spin 12s linear infinite;
-    }
-
-    .logo-hexagon {
-        position: absolute;
-        width: 100%;
-        height: 100%;
-        background: linear-gradient(135deg, var(--neon-pink), var(--neon-cyan));
-        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: var(--neon-cyan);
-        font-family: 'Orbitron', sans-serif;
-        font-weight: 700;
-        font-size: 20px;
-    }
-
-    .logo-text {
-        font-family: 'Orbitron', sans-serif;
-        font-size: 24px;
-        font-weight: 700;
-        color: var(--neon-pink);
-        text-shadow: 0 0 10px rgba(255, 0, 222, 0.5);
-    }
-
-    /* Container styling */
-    .container {
-        background: rgba(0, 0, 0, 0.6);
-        backdrop-filter: blur(10px);
-        border-radius: 10px;
-        padding: 20px;
-        box-shadow: 0 0 20px rgba(0, 255, 255, 0.2);
-    }
-
-    /* Input/Output areas */
-    .input-image, .output-image {
-        background: rgba(18, 18, 56, 0.7) !important;
-        border: 2px solid var(--neon-cyan) !important;
-        border-radius: 8px !important;
-        transition: all 0.3s ease !important;
-    }
-
-    .input-image:hover, .output-image:hover {
-        box-shadow: 0 0 15px rgba(0, 255, 255, 0.3) !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: pulse-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;
-    }
-
-    /* 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;
-    }
-
-    /* Animations */
-    @keyframes float {
-        0% { transform: translateY(0); }
-        50% { transform: translateY(-5px); }
-        100% { transform: translateY(0); }
-    }
-
-    @keyframes spin {
-        0% { transform: rotateY(0deg); }
-        100% { transform: rotateY(360deg); }
-    }
-
-    @keyframes pulse-glow {
-        0% { filter: brightness(1) drop-shadow(0 0 5px var(--neon-cyan)); }
-        100% { filter: brightness(1.2) drop-shadow(0 0 10px var(--neon-cyan)); }
-    }
-""") 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)