Upload app.py

app.py

@@ -2405,12 +2405,35 @@ 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 — reliable st.code approach
         if msg["role"] == "assistant":
-            if st.button("📋 Copy", key=f"copy_{i}", help="Click to see copyable text"):
-                st.session_state[f"show_copy_{i}"] = not st.session_state.get(f"show_copy_{i}", False)
-            if st.session_state.get(f"show_copy_{i}"):
-                st.code(msg["content"], language=None)
+            escaped = (msg["content"]
+                .replace("\\", "\\\\")
+                .replace("`", "\\`")
+                .replace("$", "\\$")
+                .replace("\n", "\\n")
+                .replace("\r", "")
+                .replace('"', '\\"'))
+            st.markdown(f"""
+<button onclick="
+    var txt=`{escaped}`;
+    if(navigator.clipboard&&window.isSecureContext){{
+        navigator.clipboard.writeText(txt).then(()=>{{
+            this.innerText='✅ Copied';
+            setTimeout(()=>this.innerText='📋 Copy',1500);
+        }});
+    }}else{{
+        var ta=document.createElement('textarea');
+        ta.value=txt.replace(/\\\\n/g,'\\n');
+        ta.style.position='fixed';ta.style.opacity='0';
+        document.body.appendChild(ta);ta.focus();ta.select();
+        document.execCommand('copy');document.body.removeChild(ta);
+        this.innerText='✅ Copied';
+        setTimeout(()=>this.innerText='📋 Copy',1500);
+    }}
+" style="background:transparent;border:1px solid #1f2937;color:#4b5563;border-radius:6px;font-size:0.72rem;padding:2px 8px;cursor:pointer;margin-top:4px;transition:all 0.15s;"
+onmouseover="this.style.borderColor='#3b82f6';this.style.color='#ececec'"
+onmouseout="this.style.borderColor='#1f2937';this.style.color='#4b5563'">📋 Copy</button>
+""", unsafe_allow_html=True)
 
 # ════════════════════════════════════════════════════════════════════
 # INPUT — use st.chat_input (cleaner than text_input + button)
@@ -2486,10 +2509,39 @@ if problem and problem != st.session_state.last_submitted:
             answer = ask_ai_streaming(problem, sympy_result, st.session_state.messages)
             plot_graph(problem, sympy_result)
 
-    
+        # Copy button on live response — available immediately
+        escaped = (answer
+            .replace("\\", "\\\\")
+            .replace("`", "\\`")
+            .replace("$", "\\$")
+            .replace("\n", "\\n")
+            .replace("\r", "")
+            .replace('"', '\\"'))
+        st.markdown(f"""
+<button onclick="
+    var txt=`{escaped}`;
+    if(navigator.clipboard&&window.isSecureContext){{
+        navigator.clipboard.writeText(txt).then(()=>{{
+            this.innerText='✅ Copied';
+            setTimeout(()=>this.innerText='📋 Copy',1500);
+        }});
+    }}else{{
+        var ta=document.createElement('textarea');
+        ta.value=txt.replace(/\\\\n/g,'\\n');
+        ta.style.position='fixed';ta.style.opacity='0';
+        document.body.appendChild(ta);ta.focus();ta.select();
+        document.execCommand('copy');document.body.removeChild(ta);
+        this.innerText='✅ Copied';
+        setTimeout(()=>this.innerText='📋 Copy',1500);
+    }}
+" style="background:transparent;border:1px solid #1f2937;color:#4b5563;border-radius:6px;font-size:0.72rem;padding:2px 8px;cursor:pointer;margin-top:4px;transition:all 0.15s;"
+onmouseover="this.style.borderColor='#3b82f6';this.style.color='#ececec'"
+onmouseout="this.style.borderColor='#1f2937';this.style.color='#4b5563'">📋 Copy</button>
+""", unsafe_allow_html=True)
+
     # Save to history
     if not st.session_state.current_chat_id:
-        new_chat()  # create chat ID BEFORE appending messages
+        new_chat()  # create chat ID BEFORE appending — avoids wiping messages
     st.session_state.messages.append({"role": "user", "content": problem})
     st.session_state.messages.append({"role": "assistant", "content": answer})
     save_current_chat()