Refactor Weaviate client initialization into a separate function for improved code organization and resource management. The client is now created within the `search_images` function and closed after use.

app.py

@@ -11,15 +11,18 @@ load_dotenv()
 # Initialize the model
 model = SentenceTransformer('all-MiniLM-L6-v2')
 
-# Initialize Weaviate client
-client = weaviate.connect_to_weaviate_cloud(
-    cluster_url=os.getenv("WEAVIATE_URL"),
-    auth_credentials=Auth.api_key(os.getenv("WEAVIATE_API_KEY"))
-)
-
-lux_collection = client.collections.get("LuxData")
+# Move client initialization into search function
+def get_client():
+    return weaviate.connect_to_weaviate_cloud(
+        cluster_url=os.getenv("WEAVIATE_URL"),
+        auth_credentials=Auth.api_key(os.getenv("WEAVIATE_API_KEY"))
+    )
 
 def search_images(query):
+    # Create client connection
+    client = get_client()
+    lux_collection = client.collections.get("LuxData")
+    
     # Encode the query and search
     query_vector = model.encode(query)
     results = lux_collection.query.near_vector(
@@ -36,6 +39,9 @@ def search_images(query):
             "image_url": result.properties.get('image_url', "No image available")
         })
     
+    # Close the client connection
+    client.close()
+    
     return formatted_results
 
 # Create Gradio interface