Merge pull request #39 from NillionNetwork/chore/fixes

chore: various fixes

Author: psofiterol

src/app/app/chat/[id]/page.tsx

@@ -14,7 +14,7 @@ interface ChatPageProps {
 }
 
 export default function ChatPage({ params }: ChatPageProps) {
-  const { selectedPersona, userSecretKeySeed } = useApp();
+  const { selectedPersona, userSecretKeySeed, setUserSecretKeySeed } = useApp();
 
   // biome-ignore lint/suspicious/noExplicitAny: TODO: add type
   const [messages, setMessages] = useState<any[]>([]);
@@ -25,6 +25,15 @@ export default function ChatPage({ params }: ChatPageProps) {
   const { user } = useAuth();
   const { decryptWithStatus, hasSecretKey } = useEncryption();
 
+  // Function to handle updating passphrase
+  const handleUpdatePassphrase = () => {
+    // Clear the current passphrase from both context and session storage
+    setUserSecretKeySeed("");
+    sessionStorage.removeItem("userSecretKeySeed");
+    // Open the modal
+    setShowSecretKeyModal(true);
+  };
+
   // Track when passphrase has been loaded from session storage
   useEffect(() => {
     // Check if passphrase exists in session storage
@@ -141,7 +150,7 @@ export default function ChatPage({ params }: ChatPageProps) {
                     </p>
                   </div>
                   <button
-                    onClick={() => setShowSecretKeyModal(true)}
+                    onClick={handleUpdatePassphrase}
                     className="ml-4 px-4 py-2 bg-[#FFC971] text-black text-sm font-medium rounded-full hover:bg-[#FFD584] transition-colors focus:outline-none focus:ring-2 focus:ring-offset-2 focus:ring-[#FFC971]"
                   >
                     Update Passphrase

src/components/AttestationModal.tsx

@@ -114,12 +114,12 @@ const AttestationModal = () => {
                 disabled={isLoading}
                 size="sm"
                 className="flex items-center gap-2 ml-auto"
-                data-umami-event="Generate Attestation Clicked"
+                data-umami-event="Fetch Attestation Clicked"
               >
                 <RefreshCw
                   className={`w-4 h-4 ${isLoading ? "animate-spin" : ""}`}
                 />
-                Generate
+                Fetch
               </Button>
             </div>
           </div>

src/components/chat/StreamingChatArea.tsx

@@ -497,14 +497,6 @@ const StreamingChatArea: React.FC<StreamingChatAreaProps> = ({
         },
         selectedPersona,
       );
-
-      window.umami.track("Message Sent", {
-        persona: getPersonaById(selectedPersona)?.name,
-        ...(userMessageAttachments &&
-          userMessageAttachments.length > 0 && {
-            attachments: userMessageAttachments,
-          }),
-      });
     } catch (error) {
       console.error("Error:", error);
       setMessages((prev) => {
@@ -516,6 +508,14 @@ const StreamingChatArea: React.FC<StreamingChatAreaProps> = ({
         return updated;
       });
     }
+
+    window.umami.track("Message Sent", {
+      persona: getPersonaById(selectedPersona)?.name,
+      ...(userMessageAttachments &&
+        userMessageAttachments.length > 0 && {
+          attachments: userMessageAttachments,
+        }),
+    });
   };
 
   return (