Upload app.py

app.py

@@ -175,6 +175,25 @@ html, body, [class*="css"] {
     background: #161b2e !important;
 }
 
+/* ── Copy button — Claude style ── */
+[data-testid="stChatMessage"] .stButton > button {
+    background: transparent !important;
+    border: 1px solid #2a2a2a !important;
+    color: #555 !important;
+    border-radius: 6px !important;
+    font-size: 0.75rem !important;
+    padding: 2px 6px !important;
+    width: auto !important;
+    min-height: 0 !important;
+    height: 24px !important;
+    transition: all 0.15s !important;
+}
+[data-testid="stChatMessage"] .stButton > button:hover {
+    border-color: #3b82f6 !important;
+    color: #ececec !important;
+    background: #1a1f2e !important;
+}
+
 /* ── Spinner ── */
 .stSpinner > div { border-top-color: #3b82f6 !important; }
 
@@ -1889,14 +1908,28 @@ for i, msg in enumerate(st.session_state.messages):
     avatar = "🧑‍🎓" if msg["role"] == "user" else "📐"
     with st.chat_message(msg["role"], avatar=avatar):
         st.markdown(msg["content"])
-        # Copy button on AI responses
+        # Claude-style copy button — pure HTML/JS
         if msg["role"] == "assistant":
-            if st.button("📋 Copy", key=f"copy_{i}", help="Copy response"):
-                st.write(
-                    f'<script>navigator.clipboard.writeText({repr(msg["content"])})</script>',
-                    unsafe_allow_html=True
-                )
-                st.toast("✅ Copied!", icon="📋")
+            safe_text = (msg["content"]
+                         .replace("&", "&amp;")
+                         .replace("<", "&lt;")
+                         .replace(">", "&gt;")
+                         .replace('"', "&quot;")
+                         .replace("'", "&#39;"))
+            btn_id = f"copybtn_{i}"
+            copy_html = (
+                f'<button id="{btn_id}" '
+                f'onclick="navigator.clipboard.writeText(this.getAttribute(\'data-text\'))'
+                f'.then(()=>{{this.textContent=\'✅ Copied!\';setTimeout(()=>{{this.textContent=\'📋 Copy\';}},2000)}})'
+                f'.catch(()=>{{this.textContent=\'❌ Failed\'}})" '
+                f'data-text="{safe_text}" '
+                f'style="background:transparent;border:1px solid #2a2a2a;color:#666;'
+                f'border-radius:6px;padding:2px 10px;font-size:0.75rem;'
+                f'cursor:pointer;margin-top:4px;" '
+                f'onmouseover="this.style.borderColor=\'#3b82f6\';this.style.color=\'#ececec\'" '
+                f'onmouseout="this.style.borderColor=\'#2a2a2a\';this.style.color=\'#666\'">📋 Copy</button>'
+            )
+            st.markdown(copy_html, unsafe_allow_html=True)
 
 # ════════════════════════════════════════════════════════════════════
 # INPUT — use st.chat_input (cleaner than text_input + button)