Update app.py

app.py

@@ -10,11 +10,10 @@ import random
 import torch
 import time
 import shutil  # Added for zip functionality
+import zipfile
 from PIL import Image
 from io import BytesIO
 from diffusers import DiffusionPipeline, LCMScheduler, AutoencoderTiny
-import zipfile
-import glob
 
 try:
     import intel_extension_for_pytorch as ipex
@@ -33,54 +32,12 @@ device = torch.device(
 torch_device = device
 torch_dtype = torch.float16
 
-
-# Function to create a zip file from a list of files
-def create_zip(files):
-    timestamp = datetime.datetime.now().strftime("%Y%m%d%H%M%S")
-    zip_filename = f"images_{timestamp}.zip"
-    print('Creating file ' + zip_filename)
-    with zipfile.ZipFile(zip_filename, 'w') as zipf:
-        for file in files:
-            zipf.write(file, os.path.basename(file))
-            print('added:' + file)
-    return zip_filename
-    
-# Function to save all images as a zip file and provide a base64 download link
-def save_all_images(images):
-    if len(images) == 0:
-        return None, None
-    zip_filename = create_zip(images)  # Create a zip file from the list of image files
-    print(' Zip file created:' + zip_filename)
-
-    zip_base64 = encode_file_to_base64(zip_filename)  # Encode the zip file to base64
-    download_link = f'<a href="data:application/zip;base64,{zip_base64}" download="{zip_filename}">Download All</a>'
-    
-    return zip_filename, download_link
-        
-# Function to handle "Save All" button click
-def save_all_button_click():
-    images = [file for file in os.listdir() if file.lower().endswith((".png", ".jpg", ".jpeg"))]
-    zip_filename, download_link = save_all_images(images)   
-    if zip_filename:
-        print(zip_filename)
-        gr.Button(link=zip_filename)
-    if download_link:
-        print(download_link)
-        #gr.HTML(download_link)
-        gr.Button(link=download_link)
-
-# Function to handle "Clear All" button click
-def clear_all_button_click():
-    clear_all_images()
-
 # Function to encode a file to base64
 def encode_file_to_base64(file_path):
     with open(file_path, "rb") as file:
         encoded = base64.b64encode(file.read()).decode()
     return encoded
 
-
-    
 def create_zip_of_files(files):
     """
     Create a zip file from a list of files.
@@ -102,8 +59,6 @@ def get_zip_download_link(zip_file):
     href = f'<a href="data:application/zip;base64,{b64}" download="{zip_file}">Download All</a>'
     return href
 
-    
-
 # Function to clear all image files
 def clear_all_images():
     base_dir = os.getcwd()  # Get the current base directory
@@ -113,8 +68,7 @@ def clear_all_images():
     for file in img_files:
         os.remove(file)
         print('removed:' + file)
-
-        
+  
 # add file save and download and clear:
 # Function to create a zip file from a list of files
 def create_zip(files):
@@ -324,11 +278,9 @@ with gr.Blocks(css=css) as demo:
             
             ```py
             from diffusers import DiffusionPipeline, LCMScheduler
-
             pipe = DiffusionPipeline.from_pretrained("Lykon/dreamshaper-7").to("cuda") 
             pipe.scheduler = LCMScheduler.from_config(pipe.scheduler.config)
             pipe.load_lora_weights("latent-consistency/lcm-lora-sdv1-5") #yes, it's a normal LoRA
-
             results = pipe(
                 prompt="ImageEditor",
                 num_inference_steps=4,
@@ -352,7 +304,5 @@ with gr.Blocks(css=css) as demo:
         save_all_button.click(save_all_button_click)
         clear_all_button.click(clear_all_button_click)
 
-        
-
 demo.queue()
 demo.launch()