import os, json, math, random, time

import numpy as np

import gradio as gr

import plotly.graph_objects as go

import requests

DEFAULT_BASE = os.getenv("FEATHER_API_BASE", "http://localhost:8000")

DEFAULT_KEY = os.getenv("FEATHER_API_KEY", "feather-9ad2c644da0d76a253b9326bd4d15d16")

DEFAULT_NS = os.getenv("FEATHER_NS", "demo")

DIM = int(os.getenv("FEATHER_DIM", "768"))

CSS = """

def _h(key): return {"X-API-Key": key, "Content-Type": "application/json"}

def _get(base, key, path):

except Exception as e: return 0, {"error": str(e)}

def _post(base, key, path, body):

except Exception as e: return 0, {"error": str(e)}

def _put(base, key, path, body):

except Exception as e: return 0, {"error": str(e)}

def _delete(base, key, path):

except Exception as e: return 0, {"error": str(e)}

def _rand_vec(seed=None, dim=DIM):

return (v / n if n > 0 else v).tolist()

def _list_records(base, key, ns, k=50, modality="text"):

return out

def get_stats(base, key):

return status_txt, ns_count, dim_txt, str(health.get("namespaces_loaded", "?")), version

def overview(base, key):

return md, pie, imp_fig

def browse_records(base, key, ns, modality, k):

return "\n".join(lines)

def add_record(base, key, ns, rec_id, modality, content, source, entity, importance, tags, use_random):

return f"❌ {status}: {json.dumps(resp)}"

def delete_record(base, key, ns, rec_id):

return f"❌ {status}: {json.dumps(resp)}"

def update_importance(base, key, ns, rec_id, new_imp):

return f"❌ {status}: {json.dumps(resp)}"

def get_record(base, key, ns, rec_id):

return f"❌ {status}: {json.dumps(resp)}"

def search(base, key, ns, modality, k, query_seed):

return "\n".join(cards), fig, obs_html

def clear_search():

return "", None, "", 3 # clear cards, chart, obs, reset seed

_MOD_COLORS = {"text": "#7c3aed", "visual": "#34d399", "image": "#34d399",

"audio": "#fbbf24", "video": "#60a5fa"}

_ENT_PALETTE = ["#a78bfa","#34d399","#60a5fa","#f97316","#fbbf24","#f472b6","#e879f9"]

def build_graph(base, key, ns, sample_k, modality):

return fig, obs

def raw_api(base, key, method, path, body_str):

return f"{tag} **HTTP {status}**\n\n```json\n{json.dumps(resp, indent=2)}\n```"

with gr.Blocks(title="Feather DB") as app:

# ── Header ───────────────────────────────────────────────────────────────

with gr.Row(elem_id="feather-header"):

gr.HTML('<span id="feather-logo">🪶 Feather DB</span>'

'<span class="header-sep">/</span>')

with gr.Row(elem_classes=["config-inputs"]):

cfg_base = gr.Textbox(value=DEFAULT_BASE, label="Base URL",

placeholder="http://...", scale=3, min_width=180)

cfg_key = gr.Textbox(value=DEFAULT_KEY, label="API Key",

type="password", scale=3, min_width=160)

cfg_ns = gr.Textbox(value=DEFAULT_NS, label="Namespace",

placeholder="demo", scale=1, min_width=80)

# ── Stats bar ─────────────────────────────────────────────────────────────

with gr.Row(elem_id="stats-bar"):

stats_html = gr.HTML("""

""")

refresh_stats_btn = gr.Button("↻ Refresh", scale=0, size="sm", variant="secondary")

# ── Main layout ───────────────────────────────────────────────────────────

with gr.Row(elem_id="main-layout"):

# ── Sidebar ───────────────────────────────────────────────────────────

with gr.Column(elem_id="sidebar", scale=0, min_width=200):

gr.HTML('<div id="sidebar-section-label">Navigate</div>')

nav = gr.Radio(

choices=["Overview", "Records", "Search", "Graph", "Raw API"],

value="Overview",

label="",

elem_id="nav-radio",

)

# ── Content ───────────────────────────────────────────────────────────

with gr.Column(elem_id="content-area", scale=1):

with gr.Tabs(elem_id="main-tabs") as tabs:

# ── Overview ─────────────────────────────────────────────────

with gr.Tab("Overview", id=0):

with gr.Row():

ov_btn = gr.Button("Refresh Overview", variant="primary", scale=0)

ov_md = gr.Markdown()

with gr.Row():

ov_pie = gr.Plot(show_label=False)

ov_imp = gr.Plot(show_label=False)

ov_btn.click(overview, [cfg_base, cfg_key], [ov_md, ov_pie, ov_imp])

app.load(overview, [cfg_base, cfg_key], [ov_md, ov_pie, ov_imp])

# ── Records ───────────────────────────────────────────────────

with gr.Tab("Records", id=1):

with gr.Row():

with gr.Column(scale=3):

gr.Markdown("### Browse Records")

with gr.Row():

br_mod = gr.Dropdown(["text","visual","image","audio"],

value="text", label="Modality", scale=1)

br_k = gr.Slider(5, 100, value=20, step=5,

label="Limit", scale=2)

br_btn = gr.Button("Browse", variant="primary", scale=0)

br_out = gr.Markdown()

br_btn.click(browse_records,

[cfg_base,cfg_key,cfg_ns,br_mod,br_k], br_out)

with gr.Column(scale=1, min_width=220):

gr.Markdown("### Record Actions")

rec_id_inp = gr.Number(label="Record ID", precision=0, value=1)

with gr.Row():

get_btn = gr.Button("Get", variant="secondary")

del_btn = gr.Button("Delete", variant="stop")

new_imp = gr.Slider(0, 1, value=0.8, step=0.05, label="Importance")

upd_btn = gr.Button("Update importance", variant="primary")

rec_out = gr.Markdown()

get_btn.click(get_record, [cfg_base,cfg_key,cfg_ns,rec_id_inp], rec_out)

del_btn.click(delete_record, [cfg_base,cfg_key,cfg_ns,rec_id_inp], rec_out)

upd_btn.click(update_importance, [cfg_base,cfg_key,cfg_ns,rec_id_inp,new_imp], rec_out)

gr.Markdown("---\n### Add Record")

with gr.Row():

add_id = gr.Number(label="ID", value=int(time.time())%100000, precision=0)

add_mod = gr.Dropdown(["text","visual","image","audio"],

value="text", label="Modality")

add_imp = gr.Slider(0, 1, value=0.8, step=0.05, label="Importance")

add_content = gr.Textbox(label="Content", placeholder="Description of this record…")

with gr.Row():

add_src = gr.Textbox(value="dashboard", label="Source")

add_etype = gr.Textbox(value="record", label="Entity")

add_tags = gr.Textbox(value="", label="Tags (comma-separated)")

add_rnd = gr.Checkbox(value=True, label="Auto-generate vector (seed = ID)")

with gr.Row():

add_btn = gr.Button("Add Record", variant="primary")

add_save_btn = gr.Button("Save namespace to disk", variant="secondary")

add_out = gr.Markdown()

add_btn.click(add_record,

[cfg_base,cfg_key,cfg_ns,add_id,add_mod,

add_content,add_src,add_etype,add_imp,add_tags,add_rnd],

add_out)

add_save_btn.click(

lambda b, k, ns: f"✅ Saved" if _post(b, k, f"/v1/{ns}/save", {})[0]==200 else "❌ Save failed",

[cfg_base, cfg_key, cfg_ns], add_out

)

# ── Search ────────────────────────────────────────────────────

with gr.Tab("Search", id=2):

with gr.Row():

sr_mod = gr.Dropdown(["text","visual","image","audio"],

value="text", label="Modality", scale=1)

sr_k = gr.Slider(1, 30, value=8, step=1, label="Top K", scale=2)

sr_seed = gr.Slider(0, 999, value=3, step=1,