Update app.py

app.py

@@ -71,12 +71,6 @@ def clear_all_images():
     for file in img_files:
         os.remove(file)
 
-
-# Add "Save All" button with emoji
-save_all_button = gr.Button("💾 Save All", scale=1)
-# Add "Clear All" button with emoji
-clear_all_button = gr.Button("🗑️ Clear All", scale=1)
-
 # 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"))]
@@ -89,11 +83,6 @@ def save_all_button_click():
 def clear_all_button_click():
     clear_all_images()
 
-# Add buttons to the Streamlit app
-gr.button(save_all_button)
-gr.button(clear_all_button)
-
-
 
 
 
@@ -205,6 +194,15 @@ css = """
 """
 with gr.Blocks(css=css) as demo:
     
+    # Add "Save All" button with emoji
+    save_all_button = gr.Button("💾 Save All", scale=1)
+    # Add "Clear All" button with emoji
+    clear_all_button = gr.Button("🗑️ Clear All", scale=1)
+
+    # Add buttons to the Streamlit app
+    gr.Button(save_all_button)
+    gr.Button(clear_all_button)
+
     # Attach click event handlers to the buttons
     save_all_button.click(save_all_button_click)
     clear_all_button.click(clear_all_button_click)