Update app.py

app.py

@@ -4,6 +4,7 @@ import uuid
 from urllib.parse import quote
 from requests import get
 from bs4 import BeautifulSoup
+import hashlib
 
 import gradio as gr
 import numpy as np
@@ -12,7 +13,7 @@ import spaces
 import torch
 from diffusers import DiffusionPipeline
 
-DESCRIPTION = """# Playground v2.5"""
+DESCRIPTION = """# Playground v2.5Afri Ai Image"""
 if not torch.cuda.is_available():
     DESCRIPTION += "\n<p>Running on CPU 🥶 This demo may not work on CPU.</p>"
 
@@ -78,6 +79,8 @@ def translate_to_english(phrase, src_lang):
 
 
 @spaces.GPU(enable_queue=True)
+CACHE_DIR = "image_cache"
+
 def generate(
     phrase: str,
     input_lang: str,
@@ -88,7 +91,6 @@ def generate(
     height: int = 1024,
     guidance_scale: float = 3,
     randomize_seed: bool = False,
-    use_resolution_binning: bool = True,
     progress=gr.Progress(track_tqdm=True),
 ):
     pipe.to(device)
@@ -103,6 +105,14 @@ def generate(
     if not use_negative_prompt:
         negative_prompt = None  # type: ignore
 
+    # Hash the prompt and seed to create a unique cache key
+    cache_key = hashlib.sha256(f"{prompt}{seed}".encode()).hexdigest()
+
+    # Check if the image is already in the cache
+    if os.path.exists(os.path.join(CACHE_DIR, f"{cache_key}.png"))):
+        print("Image found in cache")
+        return [os.path.join(CACHE_DIR, f"{cache_key}.png")], seed
+
     images = pipe(
         prompt=prompt,
         negative_prompt=negative_prompt,
@@ -118,12 +128,19 @@ def generate(
 
     image_paths = [save_image(img) for img in images]
     print(image_paths)
+
+    # Save the first image to the cache directory
+    if not os.path.exists(CACHE_DIR):
+        os.makedirs(CACHE_DIR)
+    cache_path = os.path.join(CACHE_DIR, f"{cache_key}.png")
+    Image.open(image_paths[0]).save(cache_path)
+
     return image_paths, seed
 
 
 examples = [
-    ["neon holography crystal cat", "en"],
-    ["a cat eating a piece of cheese", "en"],
+    ["nyɔnu e nɔ sa tomati ɖo aximɛ", "fon"],
+    ["ọba ilẹ̀ benin kan", "yo"],
     ["an astronaut riding a horse in space", "en"],
     ["a cartoon of a boy playing with a tiger", "en"],
     ["a cute robot artist painting on an easel, concept art", "en"],