How to connect Azure Databricks to Neo4j using Python code

I am having a file in Azure, for which I have to generate nodes and relationships in Neo4j.

Is there a way to call Neo4j in Databricks so nodes can be generated based on file availability.

Saw below type of codes, but when executed - getting error as 'no module named 'Neo4j'

Or, any other way to automatically load data to Neo4j from Azure Storage/Databricks

from neo4j import GraphDatabase

uri = "bolt://localhost:7687"
driver = GraphDatabase.driver(uri, auth=("neo4j", "password"))

def print_friends_of(tx, name):
for record in tx.run("MATCH (a:Person)-[:KNOWS]->(f) "
"WHERE a.name = {name} "
"RETURN f.name", name=name):
print(record["f.name"])

with driver.session() as session:
session.read_transaction(print_friends_of, "Alice")