Update app.py

app.py

@@ -28,6 +28,10 @@ CHECKPOINT_GEN = HF_FILENAME
 MODEL = None
 MODEL_LOAD_ERROR = None
 DOWNLOADED_CHECKPOINT_PATH = None # Store the path after download
+image_paths = ["1.png", "2.png", "3.png", "4.png", "5.png", "6.png", "7.png"]
+images = [Image.open(path) for path in image_paths]
+N = len(images)
+
 
 print(f"Attempting to download/load model '{HF_FILENAME}' from repo '{HF_REPO_ID}' onto device '{DEVICE}'...")
 
@@ -168,11 +172,23 @@ def download_and_clear(url):
         raise gr.Error(f"An unexpected error occurred preparing the image: {e}")
 
 
+def show_image(index):
+    index = index % N
+    return images[index], index
+
+def next_img(index):
+    return show_image(index + 1)
+
+def prev_img(index):
+    return show_image(index - 1)
+
+
 # --- Gradio Blocks Interface Definition ---
 print("Setting up Gradio Blocks interface...")
 
 with gr.Blocks() as demo:
     saved_image_path = gr.State()
+    index_state = gr.State(0)
     gr.Markdown(f"""
         # Watermark Remover