import json
from graph_creator import GraphCreator
data = json.load(open('../Antimony/downloads/cristiano/metadata.json', 'r'))
gc = GraphCreator('bolt://localhost:7687', 'neo4j', '332mat332')
for media_data in data:
if media_data['Type'] == 'Video':
video_name = media_data['Name'].rstrip('.mp4')
video_name = 'video_' + video_name
gc.create_video_graph(media_data['Labels'], video_name, 0.45)
gc.close_driver()
