Update app.py

app.py

@@ -2,58 +2,30 @@ import gradio as gr
 import random
 import json
 import os
-from huggingface_hub import HfApi, hf_hub_download, upload_file, create_repo
-from huggingface_hub.utils import RepositoryNotFoundError
 
-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")
+DATA_DIR = "/data"
+IMAGES_DIR = "/images"
+DATA_FILE = os.path.join(DATA_DIR, "saved_data.json")
 
 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:
+    if os.path.exists(DATA_FILE):
+        with open(DATA_FILE, 'r') as f:
             return json.load(f)
-    except:
-        return {
-            "scene_tags": [], "position_tags": [], "outfit_tags": [],
-            "camera_tags": [], "concept_tags": [], "lora_tags": [],
-            "characters": {}
-        }
+    return {
+        "scene_tags": [], "position_tags": [], "outfit_tags": [],
+        "camera_tags": [], "concept_tags": [], "lora_tags": [],
+        "characters": {}
+    }
 
 def save_data(data):
-    ensure_repo_exists()
+    os.makedirs(DATA_DIR, exist_ok=True)
     with open(DATA_FILE, 'w') as f:
         json.dump(data, f)
-    api = HfApi()
-    api.upload_file(
-        path_or_fileobj=DATA_FILE,
-        path_in_repo=DATA_FILE,
-        repo_id=REPO_ID,
-        repo_type="space"
-    )
 
 def save_character_image(name, image):
-    ensure_repo_exists()
-    if not os.path.exists(IMAGES_DIR):
-        os.makedirs(IMAGES_DIR)
+    os.makedirs(IMAGES_DIR, exist_ok=True)
     image_path = os.path.join(IMAGES_DIR, f"{name}.png")
     image.save(image_path)
-    api = HfApi()
-    api.upload_file(
-        path_or_fileobj=image_path,
-        path_in_repo=f"{IMAGES_DIR}/{name}.png",
-        repo_id=REPO_ID,
-        repo_type="space"
-    )
     return image_path
 
 def generate_prompt(scene_tags, num_people, position_tags, selected_characters, outfit_tags, camera_tags, concept_tags, lora_tags, tag_counts, data):
@@ -65,20 +37,19 @@ def generate_prompt(scene_tags, num_people, position_tags, selected_characters,
         "concept": concept_tags.split(','),
     }
     
-    for category in all_tags:
-        all_tags[category] = [tag.strip() for tag in all_tags[category] if tag.strip()]
+    all_tags = {k: [tag.strip() for tag in v if tag.strip()] for k, v in all_tags.items()}
     
-    character_prompts = []
-    for char_name in selected_characters:
-        if char_name in data["characters"]:
-            char_traits = data["characters"][char_name]["traits"]
-            char_prompt = f"{char_name}, " + ", ".join(random.sample(char_traits, min(tag_counts["character"], len(char_traits))))
-            character_prompts.append(char_prompt)
+    character_prompts = [
+        f"{char_name}, " + ", ".join(random.sample(data["characters"][char_name]["traits"], 
+                                    min(tag_counts["character"], len(data["characters"][char_name]["traits"]))))
+        for char_name in selected_characters if char_name in data["characters"]
+    ]
     
-    selected_tags = []
-    for category, tags in all_tags.items():
-        if tags:
-            selected_tags.extend(f"{tag}:{random.uniform(0.8, 1.2):.2f}" for tag in random.sample(tags, min(tag_counts[category], len(tags))))
+    selected_tags = [
+        f"{tag}:{random.uniform(0.8, 1.2):.2f}" 
+        for category, tags in all_tags.items() 
+        for tag in random.sample(tags, min(tag_counts[category], len(tags)))
+    ]
     
     if num_people.strip():
         selected_tags.append(f"{num_people} people:1.1")
@@ -92,12 +63,10 @@ def generate_prompt(scene_tags, num_people, position_tags, selected_characters,
     
     fixed_tags = "source_anime, score_9, score_8_up, score_7_up, masterpiece, best quality, very aesthetic, absurdres, anime artwork, anime style, vibrant, studio anime, highly detailed"
     
-    full_prompt = f"{main_prompt}, {fixed_tags} {lora_prompt}".strip()
-    return full_prompt
+    return f"{main_prompt}, {fixed_tags} {lora_prompt}".strip()
 
 def update_data(data, key, value):
-    if isinstance(data[key], list):
-        data[key] = list(set(data[key] + [v.strip() for v in value.split(',') if v.strip()]))
+    data[key] = list(set(data[key] + [v.strip() for v in value.split(',') if v.strip()]))
     save_data(data)
     return data