backup

app.py

@@ -0,0 +1,73 @@
+import gradio as gr
+from datasets import load_dataset
+from PIL import Image
+import io
+import time
+
+# Load and prepare the dataset
+dataset = load_dataset(
+    "taesiri/PhotoshopRequest-DailyDump", split="train", streaming=True
+)
+
+BUFFER_SIZE = 1
+sample_iterator = None
+sample_count = 0
+
+
+def reshuffle_dataset():
+    global sample_iterator, sample_count
+    seed = int(time.time())  # Convert time to an integer
+    shuffled_dataset = dataset.shuffle(seed=seed, buffer_size=BUFFER_SIZE)
+    sample_iterator = iter(shuffled_dataset)
+    sample_count = 0
+
+
+reshuffle_dataset()  # Initial shuffle
+
+
+def get_next_sample():
+    global sample_count
+
+    if sample_count >= BUFFER_SIZE:
+        reshuffle_dataset()
+
+    sample = next(sample_iterator)
+    sample_count += 1
+
+    post_id = sample["post_id"]
+    title = sample["title"]
+    reddit_url = f"https://www.reddit.com/r/PhotoshopRequest/comments/{post_id}"
+    markdown_text = f"# {title}\n\n[View post on r/PhotoshopRequest]({reddit_url})"
+
+    return (
+        markdown_text,
+        sample["source_image"],
+        sample["edited_image"],
+    )
+
+
+with gr.Blocks() as demo:
+    gr.Markdown("# PhotoshopRequest Dataset Sampler")
+
+    gr.Markdown(
+        """
+    This is a preview of the PhotoshopRequest dataset. Each sample represents a Photoshop editing request post.
+    Click the 'Sample New Item' button to retrieve a random sample from the dataset.
+    </hr>
+    """
+    )
+
+    post_info = gr.Markdown()
+
+    with gr.Row():
+        source_image = gr.Image(label="Source Image")
+        edited_image = gr.Image(label="Edited Image")
+
+    sample_button = gr.Button("Sample New Item")
+
+    sample_button.click(
+        get_next_sample, outputs=[post_info, source_image, edited_image]
+    )
+
+if __name__ == "__main__":
+    demo.launch()