Add video transcription, fix model names
This commit is contained in:
+2
-1
@@ -1,6 +1,6 @@
|
||||
import { app } from "../../scripts/app.js";
|
||||
import { api } from "../../scripts/api.js";
|
||||
import { promptEnhanceHandler, syncDimensionsNodeHandler, searchNodeHandler, APIKeyHandler, captionNodeHandler, mediaUUIDHandler, handleCustomErrors } from "./utils.js";
|
||||
import { promptEnhanceHandler, syncDimensionsNodeHandler, searchNodeHandler, APIKeyHandler, captionNodeHandler, mediaUUIDHandler, videoTranscriptionHandler, handleCustomErrors } from "./utils.js";
|
||||
import { RUNWARE_NODE_TYPES, RUNWARE_NODE_PROPS, SEARCH_TERMS } from "./types.js";
|
||||
|
||||
const nodeInitList = [];
|
||||
@@ -10,6 +10,7 @@ app.registerExtension({
|
||||
api.addEventListener('runwareError', handleCustomErrors);
|
||||
api.addEventListener('runwareImageCaption', captionNodeHandler);
|
||||
api.addEventListener('runwareMediaUUID', mediaUUIDHandler);
|
||||
api.addEventListener('runwareVideoTranscription', videoTranscriptionHandler);
|
||||
},
|
||||
|
||||
async nodeCreated(node) {
|
||||
|
||||
@@ -88,6 +88,24 @@ function captionNodeHandler(msgEvent) {
|
||||
return false;
|
||||
}
|
||||
|
||||
function videoTranscriptionHandler(msgEvent) {
|
||||
const transcriptionData = msgEvent.detail;
|
||||
const transcriptionText = transcriptionData.transcriptionText;
|
||||
const transcriptionNodeID = parseInt(transcriptionData.nodeID);
|
||||
|
||||
if(transcriptionData.success) {
|
||||
const transcriptionNode = app.graph.getNodeById(transcriptionNodeID);
|
||||
if(transcriptionNode !== null && transcriptionNode !== undefined) {
|
||||
// Find the prompt widget (where transcription is displayed)
|
||||
const promptWidget = transcriptionNode.widgets.find(widget => widget.name === "prompt");
|
||||
if(promptWidget && promptWidget.inputEl) {
|
||||
promptWidget.inputEl.value = transcriptionText;
|
||||
}
|
||||
}
|
||||
}
|
||||
return false;
|
||||
}
|
||||
|
||||
function notifyUser(message, type="info", title = "Runware", life = 4.5) {
|
||||
app.extensionManager.toast.add({
|
||||
severity: type, // 'info', 'success', 'warn', 'error' \\
|
||||
@@ -606,6 +624,7 @@ export {
|
||||
searchNodeHandler,
|
||||
mediaUUIDHandler,
|
||||
captionNodeHandler,
|
||||
videoTranscriptionHandler,
|
||||
handleCustomErrors,
|
||||
APIKeyHandler,
|
||||
};
|
||||
Reference in New Issue
Block a user