import os
from neo4j import GraphDatabase
from dotenv import load_dotenv

load_dotenv("d:/Mobcoder Pharam Care/.env")

URI = os.getenv("NEO4J_URI")
USER = os.getenv("NEO4J_USER")
PASSWORD = os.getenv("NEO4J_PASSWORD")

print("Connecting to Neo4j...")
try:
    driver = GraphDatabase.driver(URI, auth=(USER, PASSWORD))
    with driver.session() as session:
        # Get Node Labels
        labels = session.run("CALL db.labels() YIELD label RETURN label").value()
        print(f"Node Labels: {labels}")
        
        # Get Relationship Types
        rel_types = session.run("CALL db.relationshipTypes() YIELD relationshipType RETURN relationshipType").value()
        print(f"Relationship Types: {rel_types}")
        
        # Get total node count
        count = session.run("MATCH (n) RETURN count(n)").single()[0]
        print(f"Total Nodes: {count}")
    driver.close()
except Exception as e:
    print(f"Neo4j Error: {e}")