Create app.py

app.py

@@ -0,0 +1,178 @@
+import gradio as gr
+import requests
+import random
+import time
+import html
+import os
+
+# --- CONFIGURATION ---
+# Pulls credentials secretly from Hugging Face Space Settings
+API_KEY = os.environ.get("GELBOORU_API_KEY")
+USER_ID = os.environ.get("GELBOORU_USER_ID")
+
+HEADERS = {"User-Agent": "GelbooruTagFetcher/8.0 (Python/HuggingFace)"}
+ENDPOINT = "https://gelbooru.com/index.php"
+
+# --- DATA LISTS (RAW VALUES) ---
+FRANCHISES_RAW =[
+    "kancolle", "fire_emblem", "azur_lane", "honkai", "umamusume", 
+    "arknights", "girls_und_panzer", "girls'_frontline", "zenless_zone_zero", 
+    "granblue_fantasy", "kemono_friends", "nikke", "project_moon", 
+    "wuthering_waves", "league_of_legends", "princess_connect!", 
+    "chainsaw_man", "guilty_gear", "reverse:1999", "mega_man", "dq", "tales"
+]
+
+HAIR_COLORS_RAW =[
+    "Any", 
+    "white_hair",, "red_hair", "purple_hair", "pink_hair", 
+    "orange_hair", "grey_hair", "green_hair", "brown_hair", "blue_hair", 
+    "blonde_hair", "black_hair", "aqua_hair"
+]
+
+# --- UI FORMATTERS ---
+FRANCHISE_CHOICES =[(f.replace("_", " ").title(), f) for f in FRANCHISES_RAW]
+HAIR_CHOICES =[(h.replace("_", " ").title(), h) for h in HAIR_COLORS_RAW]
+
+# --- BACKEND FUNCTIONS ---
+
+def get_tag_count(tag_name):
+    params = {
+        "page": "dapi", "s": "tag", "q": "index", 
+        "json": 1, "name": tag_name
+    }
+    if API_KEY and USER_ID:
+        params["api_key"] = API_KEY
+        params["user_id"] = USER_ID
+
+    try:
+        response = requests.get(ENDPOINT, params=params, headers=HEADERS)
+        response.raise_for_status()
+        data = response.json()
+        tags = data.get("tag",[]) if isinstance(data, dict) else data
+        return int(tags[0].get("count", 0)) if tags else 0
+    except Exception:
+        return 0
+
+def find_character_tag(selected_franchises, selected_hair):
+    if not selected_franchises:
+        return "⚠️ Error: Select a franchise", "Please select at least one franchise checkbox."
+
+    chosen_franchise = random.choice(selected_franchises)
+    tag_modifier = f"({chosen_franchise}" 
+    
+    search_tags = f"score:>10 1girl solo -animated -*cosplay* *{tag_modifier}* sort:random"
+    if selected_hair != "Any":
+        search_tags += f" {selected_hair}"
+
+    log_buffer = f"🎲 Randomly selected franchise: {chosen_franchise}\n"
+    log_buffer += f"🔎 Searching: {search_tags}\n"
+
+    params = {
+        "page": "dapi", "s": "post", "q": "index", 
+        "json": 1, "limit": 50, "tags": search_tags
+    }
+    if API_KEY and USER_ID:
+        params["api_key"] = API_KEY
+        params["user_id"] = USER_ID
+
+    try:
+        response = requests.get(ENDPOINT, params=params, headers=HEADERS)
+        response.raise_for_status()
+        
+        data = response.json()
+        posts = data.get("post",[]) if isinstance(data, dict) else data
+
+        if not posts:
+            log_buffer += "❌ No results found. Try selecting 'Any' hair color."
+            return "No Match Found", log_buffer
+
+        for post in posts:
+            raw_tags = html.unescape(post.get("tags", ""))
+            tag_list = raw_tags.split()
+
+            franchise_tags = [t for t in tag_list if tag_modifier in t]
+
+            if len(franchise_tags) == 1:
+                candidate = franchise_tags[0]
+                
+                count = get_tag_count(candidate)
+                time.sleep(0.2) 
+                
+                if count >= 10:
+                    log_buffer += f"✅ Match found on Post ID {post.get('id')}\n"
+                    log_buffer += f"🏷️ Tag '{candidate}' has {count} posts."
+                    return candidate.replace('_', ' '), log_buffer 
+                else:
+                    log_buffer += f"⚠️ Candidate '{candidate}' only has {count} posts. Skipping.\n"
+
+        log_buffer += "❌ Checked 50 posts but none matched criteria."
+        return "No Match Found", log_buffer
+
+    except Exception as e:
+        return "API Error", f"An error occurred: {str(e)}"
+
+# --- UI HELPER FUNCTIONS ---
+def select_all(): return FRANCHISES_RAW
+def select_none(): return[]
+def invert_selection(current_selection): return[item for item in FRANCHISES_RAW if item not in current_selection]
+
+def copy_notification(x):
+    if x and "Error" not in x and "No Match" not in x:
+        gr.Info(f"Copied '{x}' to clipboard!")
+    return x
+
+# --- GRADIO LAYOUT ---
+JS_COPY_LOGIC = """
+(x) => {
+    if (x && x !== "Result will appear here..." && !x.includes("Error")) {
+        navigator.clipboard.writeText(x);
+    }
+    return x;
+}
+"""
+
+with gr.Blocks(theme=gr.themes.Soft()) as app:
+    gr.Markdown("## 🏷️ Gelbooru Character Tag Finder")
+    gr.Markdown("Pick your franchises and a hair color. The script will return **one specific character tag**.")
+
+    with gr.Row():
+        with gr.Column(scale=1):
+            franchise_input = gr.CheckboxGroup(
+                choices=FRANCHISE_CHOICES, value=FRANCHISES_RAW, label="1. Select Franchises"
+            )
+            
+            with gr.Row():
+                btn_all = gr.Button("Select All", size="sm")
+                btn_none = gr.Button("Clear All", size="sm")
+                btn_invert = gr.Button("Invert Selection", size="sm")
+
+            hair_input = gr.Radio(
+                choices=HAIR_CHOICES, value="Any", label="2. Select Hair Color"
+            )
+            submit_btn = gr.Button("Find Character Tag", variant="primary")
+
+        with gr.Column(scale=1):
+            gr.Markdown("### 👇 Click the button below to copy the tag")
+            
+            result_btn = gr.Button(
+                value="Result will appear here...", variant="secondary", size="lg", interactive=True
+            )
+            
+            log_output = gr.TextArea(
+                label="Process Logs", lines=10, max_lines=15, interactive=False
+            )
+
+    btn_all.click(fn=select_all, outputs=franchise_input)
+    btn_none.click(fn=select_none, outputs=franchise_input)
+    btn_invert.click(fn=invert_selection, inputs=franchise_input, outputs=franchise_input)
+    
+    submit_btn.click(
+        fn=find_character_tag, inputs=[franchise_input, hair_input], outputs=[result_btn, log_output]
+    )
+    
+    result_btn.click(
+        fn=copy_notification, inputs=result_btn, outputs=result_btn, js=JS_COPY_LOGIC
+    )
+
+if __name__ == "__main__":
+    app.launch()