Update app.py

app.py

@@ -2,13 +2,22 @@ import gradio as gr
 import random
 import json
 import os
-from huggingface_hub import HfApi, hf_hub_download, upload_file
+from huggingface_hub import HfApi, hf_hub_download, upload_file, create_repo
+from huggingface_hub.utils import RepositoryNotFoundError
 
-REPO_ID = "throaway2854/promptGen"  # Replace with your actual Space name
+REPO_ID = "throaway2854/promptGen"
 DATA_FILE = "saved_data.json"
 IMAGES_DIR = "character_images"
 
+def ensure_repo_exists():
+    try:
+        api = HfApi()
+        api.repo_info(repo_id=REPO_ID, repo_type="space")
+    except RepositoryNotFoundError:
+        create_repo(REPO_ID, repo_type="space", space_sdk="gradio")
+
 def load_data():
+    ensure_repo_exists()
     try:
         file_path = hf_hub_download(repo_id=REPO_ID, filename=DATA_FILE)
         with open(file_path, 'r') as f:
@@ -21,6 +30,7 @@ def load_data():
         }
 
 def save_data(data):
+    ensure_repo_exists()
     with open(DATA_FILE, 'w') as f:
         json.dump(data, f)
     api = HfApi()
@@ -32,6 +42,7 @@ def save_data(data):
     )
 
 def save_character_image(name, image):
+    ensure_repo_exists()
     if not os.path.exists(IMAGES_DIR):
         os.makedirs(IMAGES_DIR)
     image_path = os.path.join(IMAGES_DIR, f"{name}.png")
@@ -137,10 +148,8 @@ def create_ui():
                     with gr.Column():
                         output = gr.Textbox(label="Generated Prompt", lines=5)
                         
-                        with gr.Row():
-                            char_images = [char_data["image"] for char_data in data["characters"].values() if char_data["image"]]
-                            char_names = [char_name for char_name, char_data in data["characters"].items() if char_data["image"]]
-                            gr.Gallery(value=char_images, label="Character Images", show_label=True, elem_id="char_gallery", columns=2, rows=2, height="auto")
+                        char_images = [char_data["image"] for char_data in data["characters"].values() if char_data["image"]]
+                        gr.Gallery(value=char_images, label="Character Images", show_label=True, elem_id="char_gallery", columns=2, rows=2, height="auto")
             
             with gr.TabItem("Character Creation"):
                 with gr.Row():