def onboard_file_bundle(kiara: Kiara, uploaded_files, aliases: typing.Optional[typing.Iterable[str]]=None):
if not uploaded_files:
return
if aliases is None:
aliases = []
if isinstance(aliases, str):
aliases = [aliases]
bundle_aliases = [f"{x}_uploaded_files" for x in aliases]
if isinstance(uploaded_files, UploadedFile):
uploaded_files = [uploaded_files]
with tempfile.TemporaryDirectory() as tmpdirname:
for uf in uploaded_files:
path = os.path.join(tmpdirname, uf.name)
with open(path, 'wb') as f:
f.write(uf.getbuffer())
inputs = {
"path": tmpdirname,
"aliases": aliases,
"file_bundle_aliases": bundle_aliases
}
kiara.run("table.import.from_local_folder", inputs=inputs)
shutil.rmtree(tmpdirname, ignore_errors=True)
def import_bytes(kiara: Kiara, uploaded_file: UploadedFile):
with tempfile.TemporaryDirectory() as tmpdirname:
path = os.path.join(tmpdirname, uploaded_file.name)
with open(path, "wb") as f:
f.write(uploaded_file.getbuffer())
if uploaded_file.name.endswith(".csv"):
kiara.run(
"table.import.from_local_file",
inputs={
"path": path,
"aliases": [uploaded_file.name[0:-4]],
"file_aliases": [uploaded_file.name],
},
)
else:
kiara.run(
"import.local_file",
inputs={"path": path, "aliases": [uploaded_file.name]},
)
def create_graph(kiara: Kiara):
def create_graph(alias, edges, nodes, source_column, target_column,
weight_column, node_index):
if not alias:
return ("No alias specified, doing nothing...", None)
all_graph_aliases = find_all_aliases_of_type(
kiara, value_type="network_graph")
if alias in all_graph_aliases:
return (f"Alias '{alias}' already registered.", None)
if not edges:
return ("No edges table specified, doing nothing...", None)
inputs = {
"edges_table_id": edges,
"nodes_table_id": nodes,
"source_column": source_column,
"target_column": target_column,
"weight_column": weight_column,
"index_column_name": node_index,
}
try:
workflow = kiara.create_workflow(pipeline_file)
workflow.inputs.set_values(**inputs)
graph_value = workflow.outputs.get_value_obj("graph")
graph_value.save([alias])
return (f"", graph_value)
except Exception as e:
return (f"Error creating graph: {e}", None)
return ("CREATED GRAPH", None)
def get_table_column_names(table_id):
if not table_id:
return []
md = kiara.data_store.get_metadata_for_id(table_id)
if not md:
return []
return md.metadata["table"]["metadata_item"]["column_names"]
def find_likely_index(options: typing.Iterable, keyword: str):
for idx, alias in enumerate(options):
if keyword.lower() in alias.lower():
return idx
return 0
graph = None
st.write("Create a new graph")
graph_alias = st.text_input("The alias for the graph")
all_table_aliases = find_all_aliases_of_type(kiara, value_type="table")
default_edge_table = find_likely_index(all_table_aliases, "edge")
default_node_table = find_likely_index(all_table_aliases, "node")
select_edges = st.selectbox("Edges",
all_table_aliases,
index=default_edge_table)
select_nodes = st.selectbox("Nodes",
all_table_aliases,
index=default_node_table)
edge_column_names = get_table_column_names(select_edges)
nodes_column_names = get_table_column_names(select_nodes)
default_source_name = find_likely_index(edge_column_names, "source")
default_target_name = find_likely_index(edge_column_names, "target")
default_weight_name = find_likely_index(edge_column_names, "weight")
default_id_name = find_likely_index(nodes_column_names, "id")
source_column_name = st.selectbox("Source column name",
edge_column_names,
index=default_source_name)
target_column_name = st.selectbox("Target column name",
edge_column_names,
index=default_target_name)
weight_column_name = st.selectbox("Weight column name",
edge_column_names,
index=default_weight_name)
nodes_index_name = st.selectbox("Nodes table_index",
nodes_column_names,
index=default_id_name)
create_button = st.button(label="Create graph")
if create_button:
result, graph = create_graph(
alias=graph_alias,
edges=select_edges,
nodes=select_nodes,
source_column=source_column_name,
target_column=target_column_name,
weight_column=weight_column_name,
node_index=nodes_index_name,
)
st.write(result)
if graph is None:
return
st.write("Graph properties")
props = kiara.run("network.graph.properties",
inputs={"graph": graph},
resolve_result=True)
st.write(props)