Update backup.app.py

backup.app.py

@@ -32,16 +32,12 @@ device = torch.device(
 torch_device = device
 torch_dtype = torch.float16
 
-
-
 # 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.
@@ -63,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
@@ -74,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):
@@ -95,12 +88,13 @@ def save_all_images(images):
     zip_filename = create_zip(images)  # Create a zip file from the list of image files
     print(' Zip file created:' + zip_filename)
 
-    # gr.Button(link="/file=" + zip_filename)
+    gr.Button(link="/file=" + zip_filename)
 
     # remove?
     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>'
-
+    gr.HTML(download_link)
+    
     # redirect_button = gr.Button("Clear", variant='secondary')
     # redirect_button.click(None, None,None, _js="window.location.assign('https://google.com');")
     
@@ -113,10 +107,10 @@ def save_all_button_click():
     if zip_filename:
         print(zip_filename)
         gr.Button(link=zip_filename)
-        # gr.File(value=zip_filename)
+        gr.File(value=zip_filename)
     if download_link:
         print(download_link)
-        #gr.HTML(download_link)
+        gr.HTML(download_link)
         gr.Button(link=download_link)
 
 # Function to handle "Clear All" button click
@@ -309,9 +303,12 @@ with gr.Blocks(css=css) as demo:
 
         # Attach click event handlers to the buttons
         save_all_button.click(save_all_button_click)
-        clear_all_button.click(clear_all_button_click)
 
-        
+        with gr.Column():
+            file_obj = gr.File(label="Input File")
+            input= file_obj
+                    
+        clear_all_button.click(clear_all_button_click)
 
 demo.queue()
-demo.launch()
+demo.launch()