From 23ff263641b60fc2d4ec434725b2eaab123c34a3 Mon Sep 17 00:00:00 2001
From: Hy4ri <hiari45@gmail.com>
Date: Fri, 19 Jun 2026 19:56:50 +0300
Subject: [PATCH] Fix WebSocket connection stuck on subsequent launches (Issue
 #45)

---
 .../java/com/m57/hermescontrol/ui/chat/ChatViewModel.kt  | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/app/src/main/java/com/m57/hermescontrol/ui/chat/ChatViewModel.kt b/app/src/main/java/com/m57/hermescontrol/ui/chat/ChatViewModel.kt
index 96e7c08f..e497e085 100644
--- a/app/src/main/java/com/m57/hermescontrol/ui/chat/ChatViewModel.kt
+++ b/app/src/main/java/com/m57/hermescontrol/ui/chat/ChatViewModel.kt
@@ -79,6 +79,15 @@ class ChatViewModel(
                 handleWsEvent(event)
             }
         }
+
+        if (wsClient.isConnected) {
+            _uiState.update { it.copy(isConnected = true, isLoading = false) }
+            addSystemMessage("Connected to Hermes")
+            loadSessions()
+            if (_uiState.value.currentSessionId == null) {
+                createNewSession()
+            }
+        }
     }
 
     private fun handleWsEvent(event: WsEvent) {