Update app.py

app.py

@@ -1,30 +1,32 @@
+import numpy as np
 import gradio as gr
-from pathlib import Path
 
-DATA_PATH = Path("./")  #  Path("/data")
-def get_storage():
-    files = [
-        {
-            "orig_name": file.name,
-            "name": file.resolve(),
-            "size": file.stat().st_size,
-            "data": None,
-            "is_file": True,
-        }
-        for file in DATA_PATH.glob("**/*")
-        if file.is_file()
-    ]
-    usage = sum([f['size'] for f in files])
-    return files, f"{usage/(1024.0 ** 3):.3f}GB"
+def sepia(input_img, strength):
+    sepia_filter = strength * np.array(
+        [[0.393, 0.769, 0.189], [0.349, 0.686, 0.168], [0.272, 0.534, 0.131]]
+    ) + (1-strength) * np.identity(3)
+    sepia_img = input_img.dot(sepia_filter.T)
+    sepia_img /= sepia_img.max()
+    return sepia_img
 
-with gr.Blocks() as app:
+callback = gr.CSVLogger()
+
+with gr.Blocks() as demo:
     with gr.Row():
         with gr.Column():
-            btn = gr.Button("Run")
-        with gr.Column():
-            files = gr.Files(label="Files")
-            storage = gr.Text(label="Total Usage")
-    btn.click(get_storage, inputs=None, outputs=[files, storage], postprocess=False)
+            img_input = gr.Image()
+            strength = gr.Slider(0, 1, 0.5)
+        img_output = gr.Image()
+    with gr.Row():
+        btn = gr.Button("Flag")
+        
+    # This needs to be called at some point prior to the first call to callback.flag()
+    callback.setup([img_input, strength, img_output], "flagged_data_points")
+
+    img_input.change(sepia, [img_input, strength], img_output)
+    strength.change(sepia, [img_input, strength], img_output)
+    
+    # We can choose which components to flag -- in this case, we'll flag all of them
+    btn.click(lambda *args: callback.flag(args), [img_input, strength, img_output], None, preprocess=False)
 
-# Files that you explicitly allow on allowed_paths 
-app.launch(allowed_paths=[str(DATA_PATH)])
+demo.launch()