Upload folder using huggingface_hub

.gitattributes

@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+neurobait-logo.jpg filter=lfs diff=lfs merge=lfs -text

app.py

@@ -15,7 +15,7 @@ from threading import Lock, Thread
 import spaces
 import torch
 
-from ui import CSS, I18N, THEME, build_demo, message_text
+from ui import CSS, I18N, JS, THEME, build_demo, message_text
 
 
 BASE_MODEL = os.environ.get("BASE_MODEL", "unsloth/gemma-3-12b-it")
@@ -196,4 +196,11 @@ demo = build_demo(respond)
 
 
 if __name__ == "__main__":
-    demo.launch(show_error=True, css=CSS, theme=THEME, i18n=I18N)
+    demo.launch(
+        show_error=True,
+        css=CSS,
+        theme=THEME,
+        js=JS,
+        i18n=I18N,
+        allowed_paths=["."],
+    )

ui.py

@@ -26,11 +26,18 @@ MOODS = ["Calm", "Tired", "Anxious", "Focused"]
 
 THEME = gr.themes.Soft(primary_hue="teal", neutral_hue="slate")
 
-TITLE = "🧠 NeuroBait"
+TITLE = ""
 
 DESCRIPTION = (
     "A warm space and a gentle boost for your everyday. "
-    "No red pen · no urgency · no streaks."
+    "No red pen · no urgency · no streaks. "
+    "Not diagnostic · needs ADHD experts."
+)
+
+HEADER_HTML = (
+    '<div class="nb-header">'
+    '<img src="/file=neurobait-logo.jpg" alt="NeuroBait logo" class="nb-logo" />'
+    "</div>"
 )
 
 # Force the Gradio shell footer to English so it does not clash with the app's
@@ -93,6 +100,18 @@ body, .gradio-container {
 
 h1 { letter-spacing: -0.5px; font-weight: 800; }
 
+.nb-header {
+  display: flex;
+  justify-content: center;
+  padding: 1.5rem 0 0.75rem;
+}
+.nb-logo {
+  display: block;
+  width: min(360px, 76vw);
+  height: auto;
+  border-radius: 0;
+}
+
 /* user bubble gets the brand accent, assistant stays flat/quiet */
 .message.user, [data-testid="user"] {
   background: var(--brand-grad) !important; color: #fff !important; border: none !important;
@@ -178,6 +197,38 @@ button.primary, .submit-button, [variant="primary"] {
 }
 """
 
+JS = """
+() => {
+  const replacements = new Map([
+    ["Gunakan melalui API", "Use via API"],
+    ["Dibuat dengan Gradio", "Built with Gradio"],
+    ["Pengaturan", "Settings"],
+  ]);
+  const normalizeFooter = () => {
+    document.querySelectorAll("a, button, span").forEach((node) => {
+      const text = node.textContent?.trim();
+      if (text && replacements.has(text)) {
+        node.textContent = replacements.get(text);
+      }
+      const label = node.getAttribute?.("aria-label");
+      if (label && replacements.has(label)) {
+        node.setAttribute("aria-label", replacements.get(label));
+      }
+      const title = node.getAttribute?.("title");
+      if (title && replacements.has(title)) {
+        node.setAttribute("title", replacements.get(title));
+      }
+    });
+  };
+  normalizeFooter();
+  new MutationObserver(normalizeFooter).observe(document.body, {
+    childList: true,
+    subtree: true,
+    characterData: true,
+  });
+}
+"""
+
 
 def message_text(content) -> str:
     if isinstance(content, str):
@@ -210,16 +261,19 @@ def build_demo(respond_fn):
         placeholder=PLACEHOLDER,
         avatar_images=(None, None),
     )
-    return gr.ChatInterface(
-        fn=respond_fn,
-        chatbot=chatbot,
-        additional_inputs=[mood],
-        additional_inputs_accordion=gr.Accordion("🌳 How are you feeling?", open=True),
-        title=TITLE,
-        description=DESCRIPTION,
-        examples=EXAMPLES,
-        cache_examples=False,
-        save_history=True,
-        fill_height=True,
-        autoscroll=True,
-    )
+    with gr.Blocks(fill_height=True) as demo:
+        gr.HTML(HEADER_HTML, elem_id="neurobait-header")
+        gr.ChatInterface(
+            fn=respond_fn,
+            chatbot=chatbot,
+            additional_inputs=[mood],
+            additional_inputs_accordion=gr.Accordion("🌳 How are you feeling?", open=True),
+            title=TITLE,
+            description=DESCRIPTION,
+            examples=EXAMPLES,
+            cache_examples=False,
+            save_history=True,
+            fill_height=True,
+            autoscroll=True,
+        )
+    return demo