images added

app.py

@@ -1,9 +1,11 @@
-
 import gradio as gr
 import requests
 import io
-from PIL import Image
 import os
+import json
+from PIL import Image
+
+# Load Hugging Face API Key from Space Secrets
 HF_API_KEY = os.getenv("HF_TOKEN")
 API_URL = "https://api-inference.huggingface.co/models/sameedhayat/corgy_dog_LoRA"
 headers = {"Authorization": f"Bearer {HF_API_KEY}"}
@@ -37,15 +39,21 @@ def generate_image(user_text):
         return f"Error generating image: {str(e)}"
 
 def fetch_showcase_images():
-    """Fetch images from the Hugging Face discussions page."""
-    url = "https://huggingface.co/sameedhayat/corgy_dog_LoRA/discussions"
-    response = requests.get(url)
+    """Fetch images from the local 'images' directory and use prompts from a JSON file."""
+    images_dir = "images"
+    json_file = os.path.join(images_dir, "prompts.json")
+    showcase_images = []
+    
+    if os.path.exists(json_file):
+        with open(json_file, "r") as f:
+            image_prompts = json.load(f)
+        
+        for filename, prompt in image_prompts.items():
+            image_path = os.path.join(images_dir, filename)
+            if os.path.exists(image_path):
+                showcase_images.append((image_path, prompt))
     
-    if response.status_code == 200:
-        # Extracting images from the discussion (Placeholder logic)
-        return ["https://picsum.photos/200/300"]  # Replace with actual parsing logic
-    else:
-        return []
+    return showcase_images
 
 def create_interface():
     showcase_images = fetch_showcase_images()
@@ -61,8 +69,9 @@ def create_interface():
         btn.click(generate_image, inputs=user_input, outputs=output_image)
         
         gr.Markdown("### 🎨 Showcase of AI-Generated Designs")
-        for img_url in showcase_images:
-            gr.Image(value=img_url, show_label=False)
+        for img_path, prompt in showcase_images:
+            gr.Markdown(f"**Prompt:** {prompt}")
+            gr.Image(value=img_path, show_label=False)
     
     demo.launch()
 

images/prompts.json

@@ -0,0 +1,11 @@
+{
+    "image1.jpeg": "Super Mario face",
+    "image2.jpg": "Ferrari back view",
+    "image3.jpeg": "Parrot sitting on a branch",
+    "image4.jpeg": "Letter A in floral style",
+    "image5.jpeg": "A beautiful butterfly",
+    "image6.jpeg": "Monkey in floral pattern style",
+    "image7.jpeg": "bicycle in silhouette",
+    "image8.jpeg": "BMW logo inside floral pattern",
+    "image9.jpeg": "Floral pattern"
+}