|
|
import os |
|
|
ADMIN_PASSWORD = os.getenv("ADMIN_PASSWORD", "") |
|
|
""" |
|
|
🐾 BATUTO_encicloPromt🐾 |
|
|
Gestor de prompts con interfaz Gradio para Hugging Face Spaces |
|
|
- Clasificación automática: Personajes / Fotografía profesional |
|
|
- Modo admin protegido por contraseña (ADMIN_PASSWORD en Secrets) |
|
|
- Modo cliente: solo visual, copiar y descargar categorías |
|
|
""" |
|
|
import os |
|
|
import json |
|
|
from datetime import datetime |
|
|
from io import StringIO |
|
|
import tempfile |
|
|
import gradio as gr |
|
|
admin_pass = gr.Textbox(label="Contraseña", type="password") |
|
|
|
|
|
DATA_FILE = "prompts.json" |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
class GestorPrompts: |
|
|
def __init__(self): |
|
|
self.personajes = {} |
|
|
self.fotografia = {} |
|
|
|
|
|
self.keywords_personajes = [ |
|
|
"character", "personaje", "humanoid", "guardian", "portrait", "retrato", |
|
|
"warrior", "creature", "avatar", "hero", "villain", "body", "full body", |
|
|
"pose", "anatomy", "armor", "criatura" |
|
|
] |
|
|
self.keywords_fotografia = [ |
|
|
"photography", "foto", "bokeh", "macro", "studio", "dslr", |
|
|
"aperture", "shutter", "lens", "professional", "portrait photography", |
|
|
"product", "still life", "food", "fashion", "editorial" |
|
|
] |
|
|
|
|
|
def cargar(self, ruta=DATA_FILE): |
|
|
if os.path.exists(ruta): |
|
|
with open(ruta, "r", encoding="utf-8") as f: |
|
|
data = json.load(f) |
|
|
self.personajes = data.get("personajes", {}) |
|
|
self.fotografia = data.get("fotografia", {}) |
|
|
return self |
|
|
|
|
|
def guardar(self, ruta=DATA_FILE): |
|
|
data = {"personajes": self.personajes, "fotografia": self.fotografia} |
|
|
with open(ruta, "w", encoding="utf-8") as f: |
|
|
json.dump(data, f, ensure_ascii=False, indent=2) |
|
|
|
|
|
def clasificar(self, texto: str) -> str: |
|
|
t = texto.lower() |
|
|
if any(k in t for k in self.keywords_personajes): |
|
|
return "personajes" |
|
|
if any(k in t for k in self.keywords_fotografia): |
|
|
return "fotografia" |
|
|
return "fotografia" |
|
|
|
|
|
def agregar(self, nombre: str, texto: str, tags=None): |
|
|
categoria = self.clasificar(texto) |
|
|
entrada = { |
|
|
"texto": texto.strip(), |
|
|
"fecha": datetime.now().isoformat(timespec="seconds"), |
|
|
"tags": tags or [] |
|
|
} |
|
|
if categoria == "personajes": |
|
|
self.personajes[nombre] = entrada |
|
|
else: |
|
|
self.fotografia[nombre] = entrada |
|
|
return categoria |
|
|
|
|
|
def agregar_lista(self, lista_textos): |
|
|
resultados = [] |
|
|
for i, texto in enumerate(lista_textos, start=1): |
|
|
if not texto.strip(): |
|
|
continue |
|
|
nombre = f"Prompt_{i}" |
|
|
cat = self.agregar(nombre, texto) |
|
|
resultados.append((nombre, cat)) |
|
|
return resultados |
|
|
|
|
|
def obtener_categoria(self, categoria: str): |
|
|
return self.personajes if categoria == "personajes" else self.fotografia |
|
|
|
|
|
def eliminar(self, categoria: str, nombre: str) -> bool: |
|
|
coleccion = self.obtener_categoria(categoria) |
|
|
if nombre in coleccion: |
|
|
del coleccion[nombre] |
|
|
return True |
|
|
return False |
|
|
|
|
|
def actualizar(self, categoria: str, nombre: str, nuevo_texto: str = None, nuevos_tags=None) -> bool: |
|
|
coleccion = self.obtener_categoria(categoria) |
|
|
if nombre not in coleccion: |
|
|
return False |
|
|
if nuevo_texto is not None: |
|
|
coleccion[nombre]["texto"] = nuevo_texto.strip() |
|
|
if nuevos_tags is not None: |
|
|
coleccion[nombre]["tags"] = nuevos_tags |
|
|
return True |
|
|
|
|
|
def buscar(self, query: str, categoria: str = "todas"): |
|
|
q = query.strip().lower() |
|
|
res = [] |
|
|
cats = ["personajes", "fotografia"] if categoria == "todas" else [categoria] |
|
|
for c in cats: |
|
|
col = self.personajes if c == "personajes" else self.fotografia |
|
|
for nombre, data in col.items(): |
|
|
if q in nombre.lower() or q in data["texto"].lower() or any(q in t.lower() for t in data.get("tags", [])): |
|
|
res.append((c, nombre, data["texto"])) |
|
|
return res |
|
|
|
|
|
def exportar_categoria_json_bytes(self, categoria: str) -> bytes: |
|
|
data = self.obtener_categoria(categoria) |
|
|
si = StringIO() |
|
|
json.dump(data, si, ensure_ascii=False, indent=2) |
|
|
return si.getvalue().encode("utf-8") |
|
|
|
|
|
def concatenar_prompts_categoria(self, categoria: str) -> str: |
|
|
data = self.obtener_categoria(categoria) |
|
|
bloques = [f"# {nombre}\n{info['texto']}" for nombre, info in data.items()] |
|
|
return "\n\n---\n\n".join(bloques) |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
ADMIN_PASSWORD = os.getenv("ADMIN_PASSWORD", "") |
|
|
store = GestorPrompts().cargar() |
|
|
|
|
|
def es_admin(pass_ingresado): |
|
|
return bool(ADMIN_PASSWORD) and pass_ingresado == ADMIN_PASSWORD |
|
|
|
|
|
def do_login(pass_ingresado): |
|
|
ok = es_admin(pass_ingresado) |
|
|
msg = "✅ Acceso admin concedido." if ok else "❌ Contraseña incorrecta." |
|
|
return ok, gr.update(visible=ok), gr.update(visible=ok), gr.update(value=msg), gr.update(value="") |
|
|
|
|
|
def refrescar_listas(): |
|
|
def to_table(dic): |
|
|
return [[nombre, data["texto"]] for nombre, data in dic.items()] |
|
|
tabla_personajes = to_table(store.personajes) |
|
|
tabla_foto = to_table(store.fotografia) |
|
|
concat_personajes = store.concatenar_prompts_categoria("personajes") |
|
|
concat_foto = store.concatenar_prompts_categoria("fotografia") |
|
|
nombres_personajes = list(store.personajes.keys()) |
|
|
nombres_foto = list(store.fotografia.keys()) |
|
|
return (tabla_personajes, tabla_foto, concat_personajes, concat_foto, |
|
|
gr.update(choices=nombres_personajes), gr.update(choices=nombres_foto)) |
|
|
|
|
|
def admin_agregar(nombre, texto, lista_masiva): |
|
|
if (not nombre or not texto) and not lista_masiva.strip(): |
|
|
return "⚠️ Provee un prompt o una lista.", *refrescar_listas() |
|
|
msg = [] |
|
|
if nombre and texto: |
|
|
cat = store.agregar(nombre, texto) |
|
|
msg.append(f"Agregado '{nombre}' en {cat}.") |
|
|
if lista_masiva.strip(): |
|
|
textos = [t.strip() for t in lista_masiva.split("\n") if t.strip()] |
|
|
res = store.agregar_lista(textos) |
|
|
if res: |
|
|
msg.append(f"Lista agregada ({len(res)} prompts).") |
|
|
store.guardar() |
|
|
return " | ".join(msg) if msg else "Sin cambios.", *refrescar_listas() |
|
|
|
|
|
def admin_actualizar(categoria, nombre, nuevo_texto): |
|
|
if not nombre: |
|
|
return "⚠️ Selecciona un prompt.", *refrescar_listas() |
|
|
ok = store.actualizar(categoria, nombre, nuevo_texto=nuevo_texto) |
|
|
store.guardar() |
|
|
msg = "✏️ Actualizado." if ok else "⚠️ No existe." |
|
|
return msg, *refrescar_listas() |
|
|
|
|
|
def admin_eliminar(categoria, nombre): |
|
|
if not nombre: |
|
|
return "⚠️ Selecciona un prompt.", *refrescar_listas() |
|
|
ok = store.eliminar(categoria, nombre) |
|
|
store.guardar() |
|
|
msg = "🗑️ Eliminado." if ok else "⚠️ No existe." |
|
|
return msg, *refrescar_listas() |
|
|
|
|
|
def buscar(query, categoria): |
|
|
resultados = store.buscar(query, categoria) |
|
|
if not resultados: |
|
|
return [["-", "-", "Sin resultados"]] |
|
|
return [[c, n, t] for c, n, t in resultados] |
|
|
|
|
|
def preparar_descarga(categoria): |
|
|
data = store.exportar_categoria_json_bytes(categoria) |
|
|
tmp = tempfile.NamedTemporaryFile(delete=False, suffix=f"_{categoria}.json") |
|
|
tmp.write(data) |
|
|
tmp.flush() |
|
|
return tmp.name |
|
|
|
|
|
with gr.Blocks(theme="soft", fill_height=True) as demo: |
|
|
gr.Markdown("# 🐾 BATUTO_encicloPromt🐾") |
|
|
gr.Markdown("Explora, copia y descarga categorías. Solo el admin puede agregar o editar.") |
|
|
|
|
|
admin_state = gr.State(False) |
|
|
|
|
|
with gr.Row(): |
|
|
with gr.Column(): |
|
|
gr.Markdown("### 🔍 Búsqueda") |
|
|
in_query = gr.Textbox(label="Buscar texto o tag") |
|
|
sel_cat = gr.Dropdown(choices=["todas", "personajes", "fotografia"], value="todas", label="Categoría") |
|
|
btn_buscar = gr.Button("Buscar") |
|
|
tabla_busqueda = gr.Dataframe(headers=["Categoría", "Nombre", "Texto"], interactive=False, wrap=True) |
|
|
btn_buscar.click( |
|
|
fn=buscar, |
|
|
inputs=[in_query, sel_cat], |
|
|
outputs=[tabla_busqueda] |
|
|
) |
|
|
if __name__ == "__main__": |
|
|
demo.launch() |
