From a44292654eb09f15b6a0549a5178ba8c7902c672 Mon Sep 17 00:00:00 2001
From: Omar Amin <65174210+omaramin-2000@users.noreply.github.com>
Date: Mon, 17 Aug 2026 21:39:32 +0300
Subject: [PATCH] Add volume persistence on mute/unmute commands

Persist volume state when muting and unmuting.
---
 linux_voice_assistant/entity.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/linux_voice_assistant/entity.py b/linux_voice_assistant/entity.py
index a05986a..30ed757 100644
--- a/linux_voice_assistant/entity.py
+++ b/linux_voice_assistant/entity.py
@@ -182,6 +182,8 @@ class MediaPlayerEntity(ESPHomeEntity):
                         self.music_player.set_volume(0)
                         self.announce_player.set_volume(0)
                         self.muted = True
+                        if hasattr(self.server, "state") and getattr(self.server, "state", None) is not None:
+                            self.server.state.persist_volume(self.volume)
                     yield self._update_state(self.state)
 
                 elif command == MediaPlayerCommand.UNMUTE:
@@ -191,6 +193,8 @@ class MediaPlayerEntity(ESPHomeEntity):
                         self.music_player.set_volume(int(self.volume * 100))
                         self.announce_player.set_volume(int(self.volume * 100))
                         self.muted = False
+                        if hasattr(self.server, "state") and getattr(self.server, "state", None) is not None:
+                            self.server.state.persist_volume(self.volume)
                     yield self._update_state(self.state)
 
             elif msg.has_volume: