Update app.py

app.py

@@ -1,149 +1,155 @@
-import gradio as gr
-import json
 import os
-
-PASSWORD_FILE = "password.txt"
-PROMPTS_FILE = "prompts.json"
-
-# Cargar prompts desde un archivo JSON si existe
-def cargar_prompts_desde_archivo():
-    if os.path.exists(PROMPTS_FILE):
-        with open(PROMPTS_FILE, 'r') as f:
-            return json.load(f)
-    else:
-        # Prompts por defecto
-        return {
-            "secretaria": [
-                {
-                    "name": "Cinemático oficina",
-                    "prompt": "Ultra photorealistic full body portrait of a confident blonde businesswoman dressed in an elegant secretary outfit, fitted white blouse, dark pencil skirt, silky black thigh-high stockings, and stilettos. Standing in a modern glass office with city skyline behind. Dramatic cinematic lighting, HDR, depth of field, highly detailed skin texture, professional business atmosphere, 8k, volumetric light --ar 9:16 --v 6 --q 2 --style raw",
-                },
-                {
-                    "name": "Editorial moda",
-                    "prompt": "Hyper realistic fashion photography of a blonde woman in a secretary-inspired outfit, high-waisted pencil skirt, white silk blouse slightly unbuttoned, sheer thigh-high stockings with lace tops, black high heels. Studio fashion editorial setting, Vogue magazine style, full body shot, high detail fabric textures, HDRI lighting, cinematic DOF, ray tracing reflections, lifelike colors, soft shadows, 16k render --ar 9:16 --v 6 --q 2",
-                },
-                {
-                    "name": "Oficina futurista",
-                    "prompt": "Photorealistic 8k full body portrait of a blonde woman in a sleek secretary-style outfit, fitted blouse, glossy pencil skirt, sheer thigh-high stockings with slight sheen, futuristic office environment with glass walls and neon lights. Global illumination, volumetric fog, reflective floor, cinematic composition, hyper detailed skin pores, realistic light refractions, PBR textures, HDR photograph look --ar 9:16 --v 6 --q 2 --s 800",
-                },
-            ]
-        }
-
-# Guardar prompts en un archivo JSON
-def guardar_prompts_en_archivo(prompts_lib):
-    with open(PROMPTS_FILE, 'w') as f:
-        json.dump(prompts_lib, f, indent=4)
-
-prompts_lib = cargar_prompts_desde_archivo()
-
-favorites = []
-
-def guardar_contraseña(pw):
-    if len(pw) != 4 or not pw.isdigit():
-        return "La contraseña debe tener exactamente 4 dígitos"
-    with open(PASSWORD_FILE, "w") as f:
-        f.write(pw)
-    return "Contraseña guardada correctamente. Ahora úsala para entrar."
-
-def cargar_contraseña():
-    if not os.path.exists(PASSWORD_FILE):
+import gradio as gr
+from hf_hub_cto import Repo
+from hf_hub_cto.config import (
+    HF_TOKEN_FILE_PATH,
+    MAX_FILE_SIZE,
+    MAX_REPO_SIZE,
+    REPO_TYPES,
+)
+from hf_hub_cto.repo_info import RepoInfo
+from hf_hub_cto.repo_utils import (
+    clone_repo,
+    delete_repo,
+    get_repo_info_df,
+    get_repo_files,
+    is_valid_file_size,
+    repo_file_upload,
+    repo_rename_file,
+    sync_repo,
+)
+
+# Initialize the huggingface hub repository
+repo = Repo(
+    repo_id=os.getenv("REPO_ID"),
+    repo_type=os.getenv("REPO_TYPE"),
+    token=os.getenv("HF_TOKEN"),
+    local_dir=os.getenv("LOCAL_DIR"),
+)
+
+def get_session_info(session_id: str):
+    user_info = gr.LoginButton.get_user_info(session_id)
+    if not user_info:
         return None
-    with open(PASSWORD_FILE, "r") as f:
-        return f.read().strip()
-
-def verificar_password(pw):
-    pw_guardada = cargar_contraseña()
-    if pw_guardada is None:
-        return "No hay contraseña registrada. Por favor crea una."
-    if pw == pw_guardada:
-        return "Acceso concedido"
-    else:
-        return "Contraseña incorrecta"
-
-def gestionar_password(pw, crear=False):
-    if crear:
-        return guardar_contraseña(pw)
-    else:
-        return verificar_password(pw)
-
-def agregar_prompt(categoria, nombre, texto):
-    categoria = categoria.strip().lower()
-    if not categoria:
-        categoria = "sin_categoria"
-    if categoria not in prompts_lib:
-        prompts_lib[categoria] = []
-    prompts_lib[categoria].append({
-        "name": nombre if nombre else f"Nuevo {len(prompts_lib[categoria])+1}",
-        "prompt": texto,
-    })
-    guardar_prompts_en_archivo(prompts_lib)
-    return f"Prompt añadido en '{categoria}'"
-
-def mostrar_prompts(categoria, filtro=""):
-    categoria = categoria.strip().lower()
-    prompts = prompts_lib.get(categoria, [])
-    rows = []
-    for p in prompts:
-        if filtro.lower() in p["prompt"].lower() or filtro.lower() in p["name"].lower():
-            rows.append([p["name"], p["prompt"]])
-    return rows
-
-def exportar_prompts(categoria):
-    categoria = categoria.strip().lower()
-    prompts = prompts_lib.get(categoria, [])
-    contenido = ""
-    for p in prompts:
-        contenido += f"Nombre: {p['name']}\nPrompt: {p['prompt']}\n\n"
-    return contenido
-
-with gr.Blocks(theme=gr.themes.Dark()) as demo:
-    gr.Markdown("# BAT-@talogo_pro")
-
-    with gr.Tab("Acceso"):
-        estado = gr.Textbox(interactive=False)
-        pass_input = gr.Textbox(label="Contraseña 4 dígitos", type="password", max_lines=1)
-        crear_btn = gr.Button("Crear contraseña (solo la primera vez)")
-        entrar_btn = gr.Button("Entrar")
-
-        crear_btn.click(fn=lambda pw: gestionar_password(pw, crear=True), inputs=pass_input, outputs=estado)
-        entrar_btn.click(fn=lambda pw: gestionar_password(pw, crear=False), inputs=pass_input, outputs=estado)
-
-    with gr.Tab("Buscar y visualizar"):
-        categoria_input = gr.Textbox(label="Categoría de outfit", placeholder="Ejemplo: secretaria")
-        filtro_input = gr.Textbox(label="Palabra clave (opcional)", placeholder="Ejemplo: moda, secretaria...")
-        buscar_btn = gr.Button("🔍 Buscar")
-        resultado = gr.Dataframe(headers=["Nombre", "Prompt"])
-        export_btn = gr.Button("Exportar TXT")
-        output_txt = gr.Textbox(label="Prompts exportados", interactive=False)
-
-        buscar_btn.click(
-            fn=mostrar_prompts,
-            inputs=[categoria_input, filtro_input],
-            outputs=resultado
+    return user_info
+
+
+def set_repo_info(repo_id: str, repo_type: str, token: str):
+    repo_id_info = RepoInfo(repo_id=repo_id, repo_type=repo_type)
+    return gr.update(value=repo_id), gr.update(value=repo_type), gr.update(value=token), repo_id_info
+
+
+def get_all_repos_df(session_id: str):
+    user_info = get_session_info(session_id)
+    if not user_info:
+        return gr.DataFrame(visible=False), gr.update(visible=True)
+
+    df = get_repo_info_df(user_info["name"], token=os.getenv("HF_TOKEN"))
+    df = df.sort_values(by="last_modified", ascending=False).reset_index(drop=True)
+    return gr.DataFrame(value=df, visible=True), gr.update(visible=False)
+
+
+def handle_repo_selection(
+    evt: gr.SelectData,
+    repo_df: gr.DataFrame,
+    session_id: str,
+):
+    selected_row = repo_df.loc[evt.index[0]]
+    repo_id = selected_row["repoId"]
+    repo_type = selected_row["type"]
+    return (
+        gr.Textbox(value=repo_id),
+        gr.Textbox(value=repo_type),
+        gr.Button(visible=True),
+        gr.Button(visible=True),
+        gr.Button(visible=True),
+    )
+
+
+def handle_create_repo(repo_id: str, repo_type: str):
+    try:
+        if repo_type == "Model":
+            Repo(repo_id, repo_type="model").create_repo()
+        elif repo_type == "Dataset":
+            Repo(repo_id, repo_type="dataset").create_repo()
+        else:
+            Repo(repo_id, repo_type="space").create_repo()
+        return "Repo creado exitosamente.", None
+    except Exception as e:
+        return None, str(e)
+
+
+def handle_delete_repo(repo_id: str, repo_type: str, session_id: str):
+    user_info = get_session_info(session_id)
+    if not user_info:
+        return "Error: usuario no autenticado."
+    try:
+        delete_repo(
+            repo_id,
+            repo_type,
+            user_info["name"],
+            os.getenv("HF_TOKEN"),
         )
-        export_btn.click(
-            fn=exportar_prompts,
-            inputs=categoria_input,
-            outputs=output_txt
-        )
-
-    with gr.Tab("Edición y agregado", visible=False) as edit_tab:
-        categoria_new = gr.Textbox(label="Categoría nueva o existente", placeholder="Ejemplo: secretaria")
-        nombre_new = gr.Textbox(label="Nombre del prompt", placeholder="Ejemplo: Estilo business moderno")
-        prompt_new = gr.Textbox(label="Prompt (puedes pegar desde portapapeles)", lines=6)
-        agregar_btn = gr.Button("Agregar a catálogo")
-        confirmacion = gr.Textbox(label="", interactive=False)
-
-        agregar_btn.click(
-            fn=agregar_prompt,
-            inputs=[categoria_new, nombre_new, prompt_new],
-            outputs=confirmacion
+        return "Repo eliminado exitosamente."
+    except Exception as e:
+        return f"Error al eliminar el repositorio: {str(e)}"
+
+
+def get_repo_contents(repo_id: str, repo_type: str):
+    files = get_repo_files(repo_id=repo_id, repo_type=repo_type, token=os.getenv("HF_TOKEN"))
+    return gr.DataFrame(value=files)
+
+
+def handle_clone_repo(repo_id: str, repo_type: str):
+    try:
+        clone_repo(repo_id, repo_type)
+        return "Repositorio clonado exitosamente."
+    except Exception as e:
+        return f"Error al clonar el repositorio: {str(e)}"
+
+
+def handle_sync_repo(repo_id: str, repo_type: str):
+    try:
+        sync_repo(repo_id, repo_type)
+        return "Repositorio sincronizado exitosamente."
+    except Exception as e:
+        return f"Error al sincronizar el repositorio: {str(e)}"
+
+
+def handle_file_upload(
+    files: list, repo_id: str, repo_type: str, upload_folder: str
+):
+    try:
+        for file in files:
+            if not is_valid_file_size(file.size, MAX_FILE_SIZE):
+                raise ValueError(
+                    f"El archivo {file.name} excede el límite de tamaño de {MAX_FILE_SIZE / (1024 ** 2)} MB."
+                )
+            repo_file_upload(
+                repo_id, repo_type, file.name, upload_folder, os.getenv("HF_TOKEN")
+            )
+        return "Archivos cargados exitosamente."
+    except Exception as e:
+        return f"Error al cargar archivos: {str(e)}"
+
+
+def handle_rename_file(repo_id: str, repo_type: str, old_path: str, new_path: str):
+    try:
+        repo_rename_file(
+            repo_id,
+            repo_type,
+            old_path,
+            new_path,
+            os.getenv("HF_TOKEN"),
         )
+        return "Archivo renombrado exitosamente."
+    except Exception as e:
+        return f"Error al renombrar el archivo: {str(e)}"
 
-    def mostrar_editor(mensaje):
-        return gr.update(visible=(mensaje == "Acceso concedido"))
+# --- INTERFAZ DE USUARIO CON GRADIO ---
 
-    estado.change(fn=mostrar_editor, inputs=estado, outputs=edit_tab)
+with gr.Blocks(theme=gr.themes.Base()) as demo:
+    # Resto del código de la interfaz de usuario...
+    pass
 
-# Lanzar el demo
 demo.launch()