try fix 2

LoadBalancer.py

@@ -78,37 +78,45 @@ class LoadBalancer:
         
     def get_reports(self):
         reports = self.instances_api.fetch_reports()
+        
+        # Initialize temporary JSON data holders
+        temp_film_store = {}
+        temp_tv_store = {}
+
         for instance_url in self.instances[:]:  # Copy list to avoid modification during iteration
             if instance_url in reports:
                 report = reports[instance_url]
                 logging.info(f"Report from {instance_url}: {report}")
-                self.process_report(instance_url, report)
+                self.process_report(instance_url, report, temp_film_store, temp_tv_store)
             else:
                 logging.error(f"Failed to get report from {instance_url}. Removing instance.")
                 self.remove_instance(instance_url)
 
-    def process_report(self, instance_url, report):
+        # Compare and update actual JSON files after processing all reports
+        self.update_json_files(temp_film_store, temp_tv_store)
+
+    def process_report(self, instance_url, report, temp_film_store, temp_tv_store):
         film_store = report.get('film_store', {})
         tv_store = report.get('tv_store', {})
         cache_size = report.get('cache_size')
 
         logging.info(f"Processing report from {instance_url}")
-        # logging.info(f"Film Store: {film_store}")
-        # logging.info(f"TV Store: {tv_store}")
-        # logging.info(f"Cache Size: {cache_size}")
 
-        # Process films
+        # Update temporary film store
         for title, path in film_store.items():
             url = f"{instance_url}/api/film/{title.replace(' ', '%20')}"
-            self.update_film_store_json(title, url)
+            temp_film_store[title] = url
 
- 
-        # Process TV shows
+        # Update temporary TV store
         for title, seasons in tv_store.items():
+            if title not in temp_tv_store:
+                temp_tv_store[title] = {}
             for season, episodes in seasons.items():
+                if season not in temp_tv_store[title]:
+                    temp_tv_store[title][season] = {}
                 for episode, path in episodes.items():
                     url = f"{instance_url}/api/tv/{title.replace(' ', '%20')}/{season.replace(' ', '%20')}/{episode.replace(' ', '%20')}"
-                    self.update_tv_store_json(title, season, episode, url)
+                    temp_tv_store[title][season][episode] = url
 
         logging.info("Film and TV Stores processed successfully.")
         self.update_instances_health(instance=instance_url, cache_size=cache_size)
@@ -124,6 +132,51 @@ class LoadBalancer:
         logging.info("Stopping polling.")
         self.stop_event.set()
 
+    def update_json_files(self, temp_film_store, temp_tv_store):
+        # Load current data
+        film_store_data = self.read_json(self.FILM_STORE_JSON_PATH)
+        tv_store_data = self.read_json(self.TV_STORE_JSON_PATH)
+
+        # Compare and update film store
+        for title in list(film_store_data.keys()):
+            if title not in temp_film_store:
+                del film_store_data[title]
+        
+        for title, url in temp_film_store.items():
+            film_store_data[title] = url
+
+        with open(self.FILM_STORE_JSON_PATH, 'w') as json_file:
+            json.dump(film_store_data, json_file, indent=2)
+
+        print(f'Film store updated.')
+
+        # Compare and update TV store
+        for title in list(tv_store_data.keys()):
+            if title not in temp_tv_store:
+                del tv_store_data[title]
+            else:
+                for season in list(tv_store_data[title].keys()):
+                    if season not in temp_tv_store[title]:
+                        del tv_store_data[title][season]
+                    else:
+                        for episode in list(tv_store_data[title][season].keys()):
+                            if episode not in temp_tv_store[title][season]:
+                                del tv_store_data[title][season][episode]
+
+        for title, seasons in temp_tv_store.items():
+            if title not in tv_store_data:
+                tv_store_data[title] = {}
+            for season, episodes in seasons.items():
+                if season not in tv_store_data[title]:
+                    tv_store_data[title][season] = {}
+                for episode, url in episodes.items():
+                    tv_store_data[title][season][episode] = url
+
+        with open(self.TV_STORE_JSON_PATH, 'w') as json_file:
+            json.dump(tv_store_data, json_file, indent=2)
+
+        print(f'TV store updated.')
+
 ######################################################################
 
     @staticmethod