Update app.py

app.py

@@ -74,6 +74,13 @@ def init_mistral_api():
 # Initialize Astra DB connection
 def init_astra_db():
     """Initialize connection to Astra DB"""
+    # Initialize collection variables at the very beginning
+    db = None
+    product_embeddings = None
+    query_analytics = None
+    product_images = None
+    astra_db_keyspace = None
+    
     try:
         # Get credentials from environment variables
         astra_db_id = os.getenv("ASTRA_DB_ID")
@@ -84,36 +91,32 @@ def init_astra_db():
         
         # Initialize the client
         db = DataAPIClient(api_endpoint=astra_db_endpoint, token=astra_db_application_token)
-
-        # Initialize collection variables
-        product_embeddings = None
-        query_analytics = None
-        product_images = None
         
         # Try to create or access collections
         try:
             product_embeddings = db.collection("product_embeddings")
             query_analytics = db.collection("query_analytics")
             product_images = db.collection("product_images")
+            print("Successfully created/accessed collections")
         except Exception as collection_error:
             print(f"Error creating collections: {collection_error}")
-            # Continue execution - the app can still work with limited functionality
-
+        
         print(f"Connected to Astra DB")
-
-        # Return DB client and collections for use in application
-        return {
-            "db": db,
-            "keyspace": astra_db_keyspace,
-            "collections": {
-                "product_embeddings": product_embeddings,
-                "query_analytics": query_analytics,
-                "product_images": product_images
-            }
-        }
+        
     except Exception as e:
         print(f"Error connecting to Astra DB: {e}")
-        return None
+        db = None
+    
+    # Always return a dictionary, even if there are errors
+    return {
+        "db": db,
+        "keyspace": astra_db_keyspace,
+        "collections": {
+            "product_embeddings": product_embeddings,
+            "query_analytics": query_analytics,
+            "product_images": product_images
+        }
+    }
         
 # Initialize AWS S3 client for accessing product catalogs
 def init_s3_client():