Update app.py

app.py

@@ -8,27 +8,9 @@ import spaces
 import torch
 from diffusers import DiffusionPipeline
 from PIL import Image
-from huggingface_hub import HfFolder, Repository
-from pathlib import Path
 
-# Set up Hugging Face repository for permanent storage
-REPO_ID = "YOUR_USERNAME/YOUR_SPACE_NAME"  # Replace with your actual repo ID
-REPO_TYPE = "space"
-HF_TOKEN = os.getenv("HF_TOKEN")  # Set this in your Space's secrets
-PERSISTENT_DIR = os.path.join(os.getcwd(), "persistent")
-
-# Initialize repository
-if HF_TOKEN:
-    HfFolder.save_token(HF_TOKEN)
-    repo = Repository(
-        local_dir=PERSISTENT_DIR,
-        clone_from=REPO_ID,
-        repo_type=REPO_TYPE,
-        use_auth_token=HF_TOKEN,
-    )
-
-# Create persistent storage directory
-SAVE_DIR = os.path.join(PERSISTENT_DIR, "generated_images")
+# Create permanent storage directory
+SAVE_DIR = "saved_images"  # Gradio will handle the persistence
 if not os.path.exists(SAVE_DIR):
     os.makedirs(SAVE_DIR, exist_ok=True)
 
@@ -43,13 +25,6 @@ pipeline = pipeline.to(device)
 MAX_SEED = np.iinfo(np.int32).max
 MAX_IMAGE_SIZE = 1024
 
-def git_push():
-    if HF_TOKEN:
-        repo.git_pull()
-        repo.git_add()
-        repo.git_commit("Update generated images")
-        repo.git_push()
-
 def save_generated_image(image, prompt):
     # Generate unique filename with timestamp
     timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
@@ -65,19 +40,12 @@ def save_generated_image(image, prompt):
     with open(metadata_file, "a", encoding="utf-8") as f:
         f.write(f"{filename}|{prompt}|{timestamp}\n")
     
-    # Push changes to Hugging Face
-    git_push()
-    
     return filepath
 
 def load_generated_images():
     if not os.path.exists(SAVE_DIR):
         return []
     
-    # Pull latest changes
-    if HF_TOKEN:
-        repo.git_pull()
-    
     # Load all images from the directory
     image_files = [os.path.join(SAVE_DIR, f) for f in os.listdir(SAVE_DIR) 
                   if f.endswith(('.png', '.jpg', '.jpeg', '.webp'))]
@@ -122,8 +90,8 @@ def inference(
         joint_attention_kwargs={"scale": lora_scale},
     ).images[0]
     
-    # Save the generated image with the prompt
-    save_generated_image(image, prompt)
+    # Save the generated image
+    filepath = save_generated_image(image, prompt)
     
     # Return the image, seed, and updated gallery
     return image, seed, load_generated_images()
@@ -143,7 +111,7 @@ footer {
 }
 """
 
-with gr.Blocks(theme="Yntec/HaleyCH_Theme_Orange", css=css) as demo:
+with gr.Blocks(theme="Yntec/HaleyCH_Theme_Orange", css=css, analytics_enabled=False) as demo:
     gr.HTML('<div class="title"> Claude Monet STUDIO </div>')
     gr.HTML('<div class="title">😄Image to Video Explore: <a href="https://huggingface.co/spaces/ginigen/theater" target="_blank">https://huggingface.co/spaces/ginigen/theater</a></div>')
 
@@ -218,24 +186,37 @@ with gr.Blocks(theme="Yntec/HaleyCH_Theme_Orange", css=css) as demo:
                 )
 
         with gr.Tab("Gallery"):
+            gallery_header = gr.Markdown("### Generated Images Gallery")
             generated_gallery = gr.Gallery(
                 label="Generated Images",
                 columns=6,
                 show_label=False,
                 value=load_generated_images(),
-                elem_id="generated_gallery"
+                elem_id="generated_gallery",
+                height="auto"
             )
+            refresh_btn = gr.Button("🔄 Refresh Gallery")
 
     # Add sample gallery section at the bottom
     gr.Markdown("### Claude Monet Style Examples")
     predefined_gallery = gr.Gallery(
         label="Sample Images", 
-        columns=3, 
+        columns=3,
         rows=2,
         show_label=False, 
         value=load_predefined_images()
     )
 
+    # Event handlers
+    def refresh_gallery():
+        return load_generated_images()
+
+    refresh_btn.click(
+        fn=refresh_gallery,
+        inputs=None,
+        outputs=generated_gallery,
+    )
+
     gr.on(
         triggers=[run_button.click, prompt.submit],
         fn=inference,