fix update

main.py

@@ -3,6 +3,9 @@ from pydantic import BaseModel
 import os
 import asyncio
 from smart_search import SmartSearch
+# from dotenv import load_dotenv
+
+# load_dotenv()
 
 app = FastAPI()
 LOAD_BALANCER_URL = os.getenv("LOAD_BALANCER_URL")
@@ -16,7 +19,13 @@ search_system_lock = asyncio.Lock()
 @app.on_event("startup")
 async def startup_event():
     # Start the background task for updating data and initializing
-    asyncio.create_task(search_system.update_data_and_initialize(search_system_lock))
+    asyncio.create_task(update_data_periodically())
+
+async def update_data_periodically():
+    while True:
+        async with search_system_lock:
+            await search_system.update_data()
+        await asyncio.sleep(120)  # Sleep for 2 minutes (120 seconds)
 
 class Query(BaseModel):
     query: str

smart_search.py

@@ -49,12 +49,8 @@ class SmartSearch:
         self.index = self.create_index(self.films, self.tv_series)
         self.is_initialized = True
 
-    async def update_data_and_initialize(self, lock: asyncio.Lock):
-        while True:
-            await self.initialize()
-            async with lock:
-                self.is_initialized = True
-            await asyncio.sleep(120)  # Sleep for 2 minutes (120 seconds)
+    async def update_data(self):
+        await self.initialize()
 
     def create_index(self, films: Dict[str, str], tv_series: Dict[str, Dict[str, Union[str, List[Dict[str, str]]]]]) -> Dict[str, Union[Dict[str, str], Dict[str, Dict[str, Union[str, List[Dict[str, str]]]]]]]:
         return {
@@ -75,7 +71,7 @@ class SmartSearch:
         # Search TV series
         tv_series = self.index['tv_series']
         for series, data in tv_series.items():
-            if query in series or fuzz.partial_ratio(query, series) >50:  # Adjusted threshold
+            if query in series or fuzz.partial_ratio(query, series) > 50:  # Adjusted threshold
                 results['tv_series'].append(data['original'])
             else:
                 for episode in data['episodes']: