Update app.py

app.py

@@ -79,24 +79,28 @@ def files_to_prompt(files, current: str) -> str:
 # ── Chat ──────────────────────────────────────────────────────────────────────
 
 def _bot_stream(history: list, system_prompt: str, max_tokens: int):
-    """Generator: yields updated history list with streaming assistant response."""
+    """Generator: yields updated history list with streaming assistant response.
+    History format: [[user_msg, bot_msg], ...] — Gradio 6 tuple style.
+    """
     if not history:
         return
 
     if not HF_TOKEN:
-        history[-1]["content"] = "⚠️ No HF_TOKEN found. Add it in Settings → Secrets."
+        history[-1][1] = "⚠️ No HF_TOKEN found. Add it in Settings → Secrets."
         yield history
         return
 
-    sys = system_prompt.strip() or "You are a helpful assistant."
-    msgs = [{"role": "system", "content": sys}]
-    for m in history[:-1]:                  # all except the empty assistant placeholder
-        msgs.append({"role": m["role"], "content": m["content"]})
-    # last entry already added as assistant placeholder — skip it
-    last_user = history[-2]["content"] if len(history) >= 2 else ""
-    msgs.append({"role": "user", "content": last_user})
+    sys_prompt = system_prompt.strip() or "You are a helpful assistant."
+    msgs = [{"role": "system", "content": sys_prompt}]
+    # Build message list from all complete turns + current user message
+    for user_msg, bot_msg in history[:-1]:
+        msgs.append({"role": "user",      "content": user_msg or ""})
+        msgs.append({"role": "assistant", "content": bot_msg  or ""})
+    # Add the pending user message (bot slot is still None)
+    msgs.append({"role": "user", "content": history[-1][0] or ""})
 
     client = InferenceClient(provider="novita", api_key=HF_TOKEN)
+    history[-1][1] = ""
     try:
         for chunk in client.chat.completions.create(
             model=MODEL_ID,
@@ -105,19 +109,16 @@ def _bot_stream(history: list, system_prompt: str, max_tokens: int):
             stream=True,
         ):
             delta = chunk.choices[0].delta.content or ""
-            history[-1]["content"] += delta
+            history[-1][1] += delta
             yield history
     except Exception as e:
-        history[-1]["content"] += f"\n\n⚠️ Error: {e}"
+        history[-1][1] += f"\n\n⚠️ Error: {e}"
         yield history
 
 
 def _user_submit(msg: str, history: list):
     """Add user message + empty assistant placeholder to history."""
-    history = history + [
-        {"role": "user",      "content": msg},
-        {"role": "assistant", "content": ""},
-    ]
+    history = history + [[msg, None]]
     return "", history
 
 # ── Activate Agent ────────────────────────────────────────────────────────────
@@ -548,8 +549,7 @@ button.secondary:hover {
 .message-bubble-border { border-color: var(--border2) !important; }
 
 /* ── Accordion ── */
-.accordion > .label-wrap {
-    background: var(--card2) !important;
+.accordion > .label-wrap {Acard2) !important;
     border-color: var(--border) !important;
     color: var(--text) !important;
     font-family: var(--font-head) !important;
@@ -590,7 +590,7 @@ footer { display: none !important; }
 
 # ── Build UI ──────────────────────────────────────────────────────────────────
 
-with gr.Blocks(title="AgentForge ⚡", head=JS_HEAD, theme=gr.themes.Base()) as demo:
+with gr.Blocks(title="AgentForge ⚡") as demo:
 
     # ── Global state ──────────────────────────────────────────────────────────
     system_prompt_state = gr.State("")
@@ -680,16 +680,8 @@ with gr.Blocks(title="AgentForge ⚡", head=JS_HEAD, theme=gr.themes.Base()) as
 
             chatbot = gr.Chatbot(
                 height=430,
-                type="messages",
                 show_label=False,
-                placeholder=(
-                    "<div style='text-align:center;color:#444;padding:40px;font-family:\"Chakra Petch\",sans-serif;'>"
-                    "<div style='font-size:2.5rem;margin-bottom:12px;'>⚡</div>"
-                    "<div style='font-weight:700;font-size:.9rem;color:#666;margin-bottom:6px;'>"
-                    "Forge your agent, then start chatting.</div>"
-                    "<div style='font-size:.78rem;color:#444;'>"
-                    "Drop a .md or .skill file → activate → chat.</div></div>"
-                ),
+                bubble_full_width=False,
             )
 
             with gr.Row():
@@ -781,10 +773,7 @@ from community-forged prompts.
     def _submit(msg, hist):
         if not msg.strip():
             return gr.update(), hist
-        return "", hist + [
-            {"role": "user",      "content": msg},
-            {"role": "assistant", "content": ""},
-        ]
+        return "", hist + [[msg, None]]
 
     msg_box.submit(
         fn=_submit,
@@ -824,4 +813,7 @@ from community-forged prompts.
 
 
 if __name__ == "__main__":
-    demo.launch(css=CSS)
+    demo.launch(
+        css=CSS,
+        head=JS_HEAD,
+    )