Update app.py

app.py

@@ -5,9 +5,6 @@ import numpy as np
 import time
 import requests
 from io import BytesIO
-import json
-import os
-from datetime import datetime
 
 def washing_machine(image):
     # Convert to PIL Image if it's a numpy array
@@ -28,8 +25,7 @@ def washing_machine(image):
         yield "Spin cycle complete!", rotated
     
     # Final clean image
-    timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
-    yield "Your image is clean and background-free!", (result, timestamp)
+    yield "Your image is clean and background-free!", result
 
 def load_image_from_url(url):
     try:
@@ -39,7 +35,7 @@ def load_image_from_url(url):
     except Exception as e:
         raise gr.Error(f"Failed to load image from URL: {str(e)}")
 
-# Custom CSS and JS for the app
+# Custom CSS for washing machine theme
 css = """
 #wash-btn {background-color: #00a8ff; color: white; border: none; padding: 10px 20px; text-align: center; text-decoration: none; display: inline-block; font-size: 16px; margin: 4px 2px; cursor: pointer; border-radius: 12px;}
 #wash-btn:hover {background-color: #0095e6;}
@@ -48,29 +44,9 @@ css = """
 #output-image {border: 5px dashed #00a8ff; border-radius: 10px;}
 #input-image {border: 5px dashed #ccc; border-radius: 10px;}
 .tab {padding: 20px;}
-.history-item {margin-bottom: 15px; padding: 10px; border: 1px solid #ddd; border-radius: 5px;}
-.history-img {max-width: 200px; max-height: 200px;}
 """
 
-js = """
-function saveToHistory(imageData, timestamp) {
-    let history = JSON.parse(localStorage.getItem('bgWashHistory') || '[]');
-    history.unshift({image: imageData, timestamp: timestamp});
-    localStorage.setItem('bgWashHistory', JSON.stringify(history));
-    return history;
-}
-
-function loadHistory() {
-    return JSON.parse(localStorage.getItem('bgWashHistory') || '[]');
-}
-
-function clearHistory() {
-    localStorage.removeItem('bgWashHistory');
-    return [];
-}
-"""
-
-with gr.Blocks(css=css, js=js) as demo:
+with gr.Blocks(css=css) as demo:
     gr.Markdown("<h1 class='title'>🚀 Background Washing Machine 🧼</h1>")
     gr.Markdown("<p class='description'>Upload your image and we'll wash the background right off!</p>")
     
@@ -84,7 +60,6 @@ with gr.Blocks(css=css, js=js) as demo:
                 with gr.Column():
                     file_status = gr.Textbox(label="Washing Status")
                     file_output = gr.Image(label="Clean Image (no background)")
-                    file_timestamp = gr.Textbox(visible=False)
         
         with gr.TabItem("From URL"):
             with gr.Row():
@@ -97,19 +72,6 @@ with gr.Blocks(css=css, js=js) as demo:
                 with gr.Column():
                     url_status = gr.Textbox(label="Washing Status")
                     url_output = gr.Image(label="Clean Image (no background)")
-                    url_timestamp = gr.Textbox(visible=False)
-        
-        with gr.TabItem("History"):
-            with gr.Row():
-                with gr.Column():
-                    history_html = gr.HTML("<div id='history-container'></div>")
-                    clear_btn = gr.Button("Clear History", elem_id="wash-btn")
-                with gr.Column():
-                    history_output = gr.Image(label="Selected Image")
-            
-            # Hidden components for JS communication
-            history_data = gr.JSON(visible=False)
-            update_history = gr.Textbox(visible=False)
     
     gr.Markdown("### How it works:")
     gr.Markdown("1. Upload your image 🖼️ or paste a URL\n2. Click the washing button 🧼\n3. Wait while we wash away the background 🫧\n4. Get your clean, background-free image! ✅")
@@ -118,20 +80,16 @@ with gr.Blocks(css=css, js=js) as demo:
     file_btn.click(
         washing_machine,
         inputs=file_input,
-        outputs=[file_status, file_output, file_timestamp]
-    ).then(
-        None,
-        [file_output, file_timestamp],
-        None,
-        _js="(img, ts) => saveToHistory(img, ts)"
-    ).then(
-        None,
-        None,
-        update_history,
-        _js="() => loadHistory()"
+        outputs=[file_status, file_output]
     )
     
     # URL tab functionality
+    def fetch_image(url):
+        if not url:
+            raise gr.Error("Please enter a valid URL")
+        img = load_image_from_url(url)
+        return img, gr.Button(visible=True)
+    
     url_fetch.click(
         fetch_image,
         inputs=url_input,
@@ -141,68 +99,7 @@ with gr.Blocks(css=css, js=js) as demo:
     url_btn.click(
         washing_machine,
         inputs=url_preview,
-        outputs=[url_status, url_output, url_timestamp]
-    ).then(
-        None,
-        [url_output, url_timestamp],
-        None,
-        _js="(img, ts) => saveToHistory(img, ts)"
-    ).then(
-        None,
-        None,
-        update_history,
-        _js="() => loadHistory()"
-    )
-    
-    # History tab functionality
-    clear_btn.click(
-        None,
-        None,
-        [history_data, update_history],
-        _js="() => [clearHistory(), '']"
-    )
-    
-    update_history.change(
-        None,
-        update_history,
-        None,
-        _js="""
-        (history) => {
-            const container = document.getElementById('history-container');
-            if (!history || history.length === 0) {
-                container.innerHTML = '<p>No history yet. Process some images first!</p>';
-                return;
-            }
-            
-            let html = '<h3>Your Washed Images:</h3>';
-            history.forEach((item, index) => {
-                html += `
-                <div class="history-item">
-                    <p>${item.timestamp}</p>
-                    <img class="history-img" src="${item.image}" onclick="displaySelectedImage(${index})" />
-                </div>
-                `;
-            });
-            container.innerHTML = html;
-        }
-        
-        function displaySelectedImage(index) {
-            const history = JSON.parse(localStorage.getItem('bgWashHistory') || '[]');
-            if (history[index]) {
-                const imgElement = document.querySelector('#history-container img.selected');
-                if (imgElement) imgElement.classList.remove('selected');
-                document.querySelectorAll('#history-container img')[index].classList.add('selected');
-                window.selectedImage = history[index].image;
-            }
-        }
-        """
-    )
-    
-    demo.load(
-        None,
-        None,
-        update_history,
-        _js="() => loadHistory()"
+        outputs=[url_status, url_output]
     )
 
 demo.launch()