Update app.py

app.py

@@ -3,7 +3,6 @@ import requests
 from PIL import Image
 from io import BytesIO
 from tqdm import tqdm
-import os
 import time
 
 repo = "artificialguybr/TshirtDesignRedmond-V2"
@@ -43,26 +42,12 @@ def infer(color_prompt, phone_type_prompt, design_prompt):
         else:
             raise Exception(f"API Error: {response.status_code}")
 
-# Save the design
+# Function to save the design
 def save_design(image):
-    saved_designs_dir = "saved_designs"
-    os.makedirs(saved_designs_dir, exist_ok=True)
-    file_path = os.path.join(saved_designs_dir, f"design_{int(time.time())}.png")
+    file_path = "saved_design.png"
     image.save(file_path)
     return f"Design saved as {file_path}!"
 
-# Fetch saved designs for display on the Home page
-def fetch_saved_designs():
-    saved_designs_dir = "saved_designs"
-    if not os.path.exists(saved_designs_dir):
-        return []
-    
-    designs = []
-    for file_name in os.listdir(saved_designs_dir):
-        if file_name.endswith((".png", ".jpg", ".jpeg")):
-            designs.append(os.path.join(saved_designs_dir, file_name))
-    return designs
-
 # Custom CSS for animations
 custom_css = """
 body {
@@ -94,6 +79,16 @@ body::before {
 custom_js = """
 <script>
 document.addEventListener('DOMContentLoaded', function () {
+    // Add particles
+    const particlesContainer = document.createElement('div');
+    particlesContainer.classList.add('particles');
+    for (let i = 0; i < 5; i++) {
+        const particle = document.createElement('div');
+        particle.classList.add('particle');
+        particlesContainer.appendChild(particle);
+    }
+    document.body.appendChild(particlesContainer);
+
     // Text-to-speech functionality
     function speak(text) {
         const synth = window.speechSynthesis;
@@ -119,22 +114,8 @@ with gr.Blocks(css=custom_css) as interface:
     # Navigation Tabs
     with gr.Tabs():
         with gr.Tab("Home"):
-            gr.Markdown("Welcome to the **AI Phone Cover Designer**! Below are some of your saved designs.")
-    
-            saved_images_gallery = gr.Gallery(label="Saved Designs", show_label=True)
-    
-            refresh_button = gr.Button("Refresh Gallery")
-    
-            def refresh_gallery():
-                return fetch_saved_designs()
-    
-            refresh_button.click(
-                refresh_gallery,
-                inputs=[],
-                outputs=[saved_images_gallery]
-            )
-
-
+            gr.Markdown("Welcome to the **AI Phone Cover Designer**! Use the 'Design' tab to start creating custom designs.")
+        
         with gr.Tab("Design"):
             with gr.Row():
                 with gr.Column(scale=1):
@@ -161,7 +142,7 @@ with gr.Blocks(css=custom_css) as interface:
             )
 
         with gr.Tab("About"):
-            gr.Markdown("""            
+            gr.Markdown("""
             ## About AI Phone Cover Maker
             The **AI Phone Cover Maker** is a cutting-edge tool designed to help users create personalized phone cover designs quickly and easily. 
             Powered by AI, it uses advanced image generation techniques to craft unique, high-quality designs for any mobile device.