unstable fix

LoadBalancer.py

@@ -34,6 +34,8 @@ class LoadBalancer:
 
         # Initialize file structure and start prefetching
         self.file_structure = indexer()
+
+        # Start prefetching in the FastAPI event loop
         self.start_prefetching()
 
         # Start polling and file checking in separate threads
@@ -42,11 +44,9 @@ class LoadBalancer:
         polling_thread.start()
 
     def start_prefetching(self):
-        """Start the metadata prefetching in a separate thread."""
-        loop = asyncio.new_event_loop()
-        asyncio.set_event_loop(loop)
-        loop.run_in_executor(None, self.prefetch_metadata)
-        
+        """Start the metadata prefetching in the FastAPI event loop."""
+        asyncio.create_task(self.prefetch_metadata())
+
     async def prefetch_metadata(self):
         """Prefetch metadata for all items in the file structure."""
         tasks = []

app.py

@@ -9,10 +9,13 @@ CACHE_DIR = os.getenv("CACHE_DIR")
 TOKEN = os.getenv("TOKEN")
 REPO = os.getenv("REPO")
 
-load_balancer = LoadBalancer(cache_dir=CACHE_DIR, token=TOKEN, repo=REPO)
-
 app = FastAPI()
 
+@app.on_event("startup")
+async def startup_event():
+    global load_balancer
+    load_balancer = LoadBalancer(cache_dir=CACHE_DIR, token=TOKEN, repo=REPO)
+
 @app.get("/")
 def greet_json():
     return {"Version": "0.0.1 Alpha"}