Update app.py

app.py

@@ -84,14 +84,22 @@ def init_astra_db():
         
         # Initialize the client
         db = DataAPIClient(api_endpoint=astra_db_endpoint, token=astra_db_application_token)
-        
-        # Get or create collections
-        product_embeddings = db.collection("product_embeddings")
-        query_analytics = db.collection("query_analytics")
-        product_images = db.collection("product_images")
-        
-        print(f"Connected to Astra DB: {db.list_collection_names()}")
-        
+
+        try:
+            product_embeddings = db.collection("product_embeddings")
+            query_analytics = db.collection("query_analytics")
+            product_images = db.collection("product_images")
+        except Exception as collection_error:
+            print(f"Error creating collections: {collection_error}")
+
+        print(f"Connected to Astra DB")
+
+        try:
+            collections = db.list_collections()
+            print(f"Available collections: {collections}")
+        except AttributeError:
+            print("Unable to list collections (method not available)")
+
         # Return DB client and collections for use in application
         return {
             "db": db,