Updates

README.md

@@ -10,4 +10,26 @@ pinned: false
 license: apache-2.0
 ---
 
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+# Image FFT Playground
+
+Gradio app to play with FFT of images.
+
+Huggingface Spaces Demo: <https://huggingface.co/spaces/acozma/image_fft_playground>
+
+## Usage
+
+- Upload your own input image, or use one of the example images
+- Optionally draw on the input image to see how the FFT changes
+- Mask out areas of the FFT magnitude and phase to see how the image changes
+- Apply IFFT to the modified FFT to see the result
+
+## Example Screenshots
+
+![Generate FFT](image.png)
+
+![Generate IFFT](image-1.png)
+
+## Known Bugs
+
+- Gradio Image canvas sometimes bugs out and unintentionally resizes/crops the images on the FFT Magnitude and Phase canvases. As a result, taking the IFFT will produce an error. To fix this, just refresh the page and try again.
+  - In the process of re-writing the app using Streamlit, since Gradio image canvas is not very stable and the API is not very flexible.

app.py

@@ -5,6 +5,7 @@ import gradio as gr
 import numpy as np
 from PIL import Image, ImageChops, ImageOps
 
+samples_dir = "./samples"
 
 class ImageInfo:
     def __init__(
@@ -290,7 +291,7 @@ def update_image_input(state, selection):
             None,
         )
 
-    image_path = os.path.join("./images", selection)
+    image_path = os.path.join(samples_dir, selection)
     print(f"image_path: {image_path}")
     if not os.path.exists(image_path):
         raise gr.Error(f"Image not found: {image_path}")
@@ -353,7 +354,7 @@ with gr.Blocks(css=css) as demo:
                 mask_opacity=1.0,
                 elem_classes=["inp_img"],
             )
-            files = os.listdir("./images")
+            files = os.listdir(samples_dir)
             files = sorted(files)
             inp_samples = gr.Dropdown(
                 choices=files,

generate_samples.py

@@ -6,25 +6,25 @@ from scipy.ndimage import gaussian_filter
 
 
 def create_blank_image(size=512):
-    img = np.ones((size, size)) * 255
+    img = np.zeros((size, size))
     return img.astype(np.uint8)
 
 
 def add_horizontal_lines(image, spacing=10):
     for i in range(0, image.shape[0], spacing):
-        image[i, :] = 0
+        image[i, :] = 255
     return image
 
 
 def add_vertical_lines(image, spacing=10):
     for i in range(0, image.shape[1], spacing):
-        image[:, i] = 0
+        image[:, i] = 255
     return image
 
 
 def add_diagonal_lines(image, spacing=10):
     for i in range(0, image.shape[0], spacing):
-        np.fill_diagonal(image[i:, i:], 0)
+        np.fill_diagonal(image[i:, i:], 255)
     return image
 
 
@@ -35,18 +35,18 @@ def add_circle(image, radius=50):
         -center_y : image.shape[0] - center_y, -center_x : image.shape[1] - center_x
     ]
     mask = x * x + y * y <= radius * radius
-    image[mask] = 0
+    image[mask] = 255
     return image
 
 
 def add_checkerboard(image, square_size=16):
     for i in range(0, image.shape[0], square_size * 2):
         for j in range(0, image.shape[1], square_size * 2):
-            image[i : i + square_size, j : j + square_size] = 0
+            image[i : i + square_size, j : j + square_size] = 255
             image[
                 i + square_size : i + square_size * 2,
                 j + square_size : j + square_size * 2,
-            ] = 0
+            ] = 255
     return image
 
 

requirements.txt

@@ -2,3 +2,5 @@ gradio==3.50.2
 numpy==1.25.1
 Pillow==10.0.0
 scipy==1.11.1
+streamlit==1.28.0
+streamlit-drawable-canvas==0.9.3