Update app.py

app.py

@@ -60,11 +60,17 @@ def load_images_from_directory():
     return Load_Data().from_folder([REFERENCE_DATASET_PATH])
 
 def initialize_search_engine(image_list):
+    if not image_list:
+        raise RuntimeError("No images found to initialize the search engine.")
+
     search_setup = Search_Setup(
         image_list=image_list, model_name='vgg19', pretrained=True, image_count=len(image_list)
     )
-    with patch('builtins.input', return_value='no'):
-        search_setup.run_index()
+    try:
+        with patch('builtins.input', return_value='no'):
+            search_setup.run_index()
+    except Exception as e:
+        raise RuntimeError(f"Error during search engine initialization: {e}")
     return search_setup
 
 # App Initialization
@@ -77,7 +83,10 @@ def initialize_app():
     image_list = load_images_from_directory()
     return initialize_search_engine(image_list)
 
-search_setup = initialize_app()
+try:
+    search_setup = initialize_app()
+except Exception as e:
+    raise RuntimeError(f"Application initialization failed: {e}")
 
 # Routes
 @app.get("/health")