Update app.py

app.py

@@ -53,17 +53,6 @@ def load_generated_images():
     image_files.sort(key=lambda x: os.path.getctime(x), reverse=True)
     return image_files
 
-class ImageFlagging(gr.FlaggingCallback):
-    def setup(self, components, flagging_dir: str):
-        self.components = components
-        self.flagging_dir = SAVE_DIR
-        
-    def flag(self, flag_data, flag_option=None, flag_index=None, username=None) -> int:
-        """Save image and metadata permanently"""
-        image, prompt = flag_data
-        filepath = save_generated_image(image, prompt)
-        return 0
-
 def load_predefined_images():
     predefined_images = [
         "assets/cm1.webp",
@@ -123,8 +112,6 @@ footer {
 """
 
 with gr.Blocks(theme="Yntec/HaleyCH_Theme_Orange", css=css) as demo:
-    flagging_callback = ImageFlagging()
-
     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>')
 
@@ -217,7 +204,6 @@ with gr.Blocks(theme="Yntec/HaleyCH_Theme_Orange", css=css) as demo:
         value=load_predefined_images()
     )
 
-    # Enable flagging for permanent storage
     gr.on(
         triggers=[run_button.click, prompt.submit],
         fn=inference,
@@ -232,11 +218,6 @@ with gr.Blocks(theme="Yntec/HaleyCH_Theme_Orange", css=css) as demo:
             lora_scale,
         ],
         outputs=[result, seed, generated_gallery],
-    ).then(
-        fn=lambda x, y: None,  # Empty function instead of JavaScript
-        inputs=[result, prompt],
-        outputs=None,
-        flagging_callback=flagging_callback
     )
 
 demo.queue()