From c9d766c3db169f0ef167bc4cba0dea1e73d77889 Mon Sep 17 00:00:00 2001
From: dekzter <kappa118@gmail.com>
Date: Thu, 20 Mar 2025 18:34:59 -0400
Subject: [PATCH] fixing some bugs

---
 apps/channels/api_views.py | 2 +-
 frontend/src/WebSocket.jsx | 7 +++++--
 2 files changed, 6 insertions(+), 3 deletions(-)

diff --git a/apps/channels/api_views.py b/apps/channels/api_views.py
index f6189d3e..579d9dfa 100644
--- a/apps/channels/api_views.py
+++ b/apps/channels/api_views.py
@@ -262,7 +262,7 @@ class ChannelViewSet(viewsets.ModelViewSet):
                 errors.append({"item": item, "error": str(e)})
                 continue
 
-            channel_group, _ = ChannelGroup.objects.get_or_create(name=stream.group_name)
+            channel_group = stream.channel_group
 
             # Determine channel number: if provided, use it (if free); else auto assign.
             provided_number = item.get('channel_number')
diff --git a/frontend/src/WebSocket.jsx b/frontend/src/WebSocket.jsx
index 92670fca..4f4838b9 100644
--- a/frontend/src/WebSocket.jsx
+++ b/frontend/src/WebSocket.jsx
@@ -17,7 +17,7 @@ export const WebsocketProvider = ({ children }) => {
   const [val, setVal] = useState(null);
 
   const { fetchStreams } = useStreamsStore();
-  const { setChannelStats } = useChannelsStore();
+  const { setChannelStats, fetchChannelGroups } = useChannelsStore();
   const { setRefreshProgress } = usePlaylistsStore();
 
   const ws = useRef(null);
@@ -62,7 +62,10 @@ export const WebsocketProvider = ({ children }) => {
               color: 'green.5',
             });
           } else if (event.data.progress) {
-            console.log('calling set progress');
+            if (event.data.progress == 100) {
+              fetchStreams();
+              fetchChannelGroups();
+            }
             setRefreshProgress(event.data.account, event.data.progress);
           }
           break;