48 lines
1.6 KiB
JavaScript
48 lines
1.6 KiB
JavaScript
import { ComfyWidgets } from "../../../scripts/widgets.js";
|
|
import { fitHeight } from "./utils.js";
|
|
import { formatAndDisplayJSON } from "./gtUIUtils.js";
|
|
import { hideWidget, showWidget } from "./utils.js";
|
|
import { app } from "../../../scripts/app.js";
|
|
export function setupExtractionNodes(nodeType, nodeData, app) {
|
|
if (nodeData.name.includes("Extraction"))
|
|
{
|
|
setupExtractionTypeAttrr(nodeType, nodeData, app);
|
|
}
|
|
}
|
|
|
|
function setupExtractionTypeAttrr(nodeType, nodeData, app) {
|
|
const onNodeCreated = nodeType.prototype.onNodeCreated
|
|
nodeType.prototype.onNodeCreated = function() {
|
|
const me = onNodeCreated?.apply(this);
|
|
const widget_extraction_type = this.widgets.find(w => w.name === 'extraction_type');
|
|
const widget_column_names = this.widgets.find(w=> w.name === 'column_names');
|
|
const widget_template_schema = this.widgets.find(w=> w.name === 'template_schema');
|
|
|
|
// Hide both widgets
|
|
widget_extraction_type.callback = async() => {
|
|
hideWidget(this, widget_column_names);
|
|
hideWidget(this, widget_template_schema);
|
|
|
|
switch (widget_extraction_type.value) {
|
|
case "csv":
|
|
showWidget(widget_column_names);
|
|
fitHeight(this, true);
|
|
break;
|
|
case "json":
|
|
showWidget(widget_template_schema);
|
|
fitHeight(this, true);
|
|
break;
|
|
default:
|
|
fitHeight(this, true);
|
|
break;
|
|
}
|
|
}
|
|
|
|
setTimeout(() => { widget_extraction_type.callback() }, 5);
|
|
return me;
|
|
// setupMessageStyle(this.message);
|
|
};
|
|
|
|
}
|
|
|