1
This commit is contained in:
parent
597adaa75a
commit
d595e932e3
BIN
resources/icon.png
Normal file
BIN
resources/icon.png
Normal file
Binary file not shown.
|
After Width: | Height: | Size: 34 KiB |
53
resources/icon.svg
Normal file
53
resources/icon.svg
Normal file
@ -0,0 +1,53 @@
|
|||||||
|
<svg class="svg-canvas" viewBox="0 0 800 600" width="1600" height="1200" preserveAspectRatio="none" version="1.1"
|
||||||
|
xmlns="http://www.w3.org/2000/svg"
|
||||||
|
xmlns:xlink="http://www.w3.org/1999/xlink">
|
||||||
|
<!--滤镜-->
|
||||||
|
<defs>
|
||||||
|
<filter id="filter_shadow_shape_HBt9Poo2sR" width="300%" height="300%" x="-100%" y="-100%">
|
||||||
|
<feDropShadow dx="7.431448254773942" dy="6.691306063588582" stdDeviation="1.2000000000000002" flood-color="#000000" flood-opacity="0.26"></feDropShadow>
|
||||||
|
</filter>
|
||||||
|
</defs>
|
||||||
|
|
||||||
|
|
||||||
|
<!--元素-->
|
||||||
|
|
||||||
|
<g id="shape_BG1gIWpILa" ref="shape_BG1gIWpILa" key="shape_BG1gIWpILa" mask="">
|
||||||
|
<g transform="translate(-82.01635169835797,-38.231795889017974) rotate(0,482.36560107022797,353.36560107022797) scale(1,1)" style="opacity: 1;mix-blend-mode: multiply;" filter="">
|
||||||
|
<!--普通元素-->
|
||||||
|
<svg xmlns="http://www.w3.org/2000/svg" version="1.1"
|
||||||
|
xmlns:xlink="http://www.w3.org/1999/xlink" width="964.7312021404559" height="706.7312021404559" viewBox="0 0 964.7312021404559 706.7312021404559" data-ligature="true" data-parent="shape_BG1gIWpILa">
|
||||||
|
<g data-role="text" data-width="354.00141380959474" data-height="117" transform="translate(305.3648941654306, 294.86560107022797)">
|
||||||
|
<path d="M17.66-71.89L17.66-71.89Q19.68-69.97 28.03-67.96L28.03-67.96Q30.43-66.61 30.43-63.83L30.43-63.83L30.43-63.54Q30.43-60.18 25.05-59.42L25.05-59.42L24.48-58.46L24.48-56.44Q21.79-45.79 21.79-40.89L21.79-40.89L22.08-38.20L20.35-28.70L20.64-27.26Q19.10-16.70 19.10-15.65L19.10-15.65L19.39-15.65Q24.57-15.65 44.44-15.93L44.44-15.93Q46.55-15.65 46.84-15.65L46.84-15.65Q51.64-18.91 53.37-18.91L53.37-18.91L53.94-18.91Q54.52-17.76 54.81-17.76L54.81-17.76Q54.52-14.78 54.52-13.25L54.52-13.25L54.52-12.38Q54.52-10.75 55.38-7.87L55.38-7.87Q55.38-6.72 53.08-4.90L53.08-4.90Q51.54-4.90 51.54-5.47L51.54-5.47L48.57-5.18L44.73-5.18Q41.37-5.18 36.09-5.47L36.09-5.47L35.51-5.18Q34.46-5.47 34.27-5.47L34.27-5.47Q28.12-4.90 24.19-4.90L24.19-4.90L22.08-4.90Q20.64-4.90 17.09-4.03L17.09-4.03L14.11-4.32Q10.46-3.46 10.46-3.17L10.46-3.17Q10.08-3.17 6.91-7.87L6.91-7.87Q4.22-9.89 4.22-11.42L4.22-11.42Q4.22-12.29 6.05-15.65L6.05-15.65Q6.05-17.47 8.45-21.02L8.45-21.02Q13.15-42.62 13.15-61.72L13.15-61.72Q13.44-61.72 13.44-63.25L13.44-63.25L11.71-64.69Q12.09-66.23 16.41-69.78L16.41-69.78Q16.41-71.89 17.66-71.89Z" transform="translate(0 94) " fill="rgb(50, 113, 174)" stroke="rgb(50, 113, 174)" stroke-width="0.164" data-glyph-w="63.4155967830671" data-glyph-h="95.986328125" data-glyph-ascender="84.46796875000001" data-glyph-descender="-11.518359375000001" data-kerning="0"></path>
|
||||||
|
<path d="M24.19-39.64L24.19-39.64L25.72-39.64Q27.84-39.64 30.72-35.80L30.72-35.80L31.39-35.80Q33.12-37.63 35.80-37.63L35.80-37.63Q39.64-36.19 39.64-32.54L39.64-32.54L41.47-28.99Q41.47-28.51 40.89-26.30L40.89-26.30Q40.89-26.11 41.18-24.48L41.18-24.48Q39.07-19.39 39.07-18.24L39.07-18.24L39.07-17.37Q39.07-15.07 41.18-13.25L41.18-13.25L41.75-13.25Q43.77-13.25 45.88-16.22L45.88-16.22L48.57-17.09Q49.62-17.09 50.10-13.82L50.10-13.82Q48.86-8.45 48.86-7.58L48.86-7.58Q49.43-7.29 49.82-7.29L49.82-7.29Q48.38-4.61 45.59-4.61L45.59-4.61Q44.73-4.61 43.58-4.90L43.58-4.90Q41.95-4.61 40.31-4.61L40.31-4.61Q34.08-7.29 33.12-9.02L33.12-9.02Q26.49-3.65 25.05-3.65L25.05-3.65Q25.05-3.07 20.64-2.78L20.64-2.78Q12.57-2.78 9.31-8.16L9.31-8.16Q5.76-12.48 5.76-17.95L5.76-17.95L5.76-21.21Q5.76-31.48 16.80-37.91L16.80-37.91Q22.46-39.64 24.19-39.64ZM13.15-19.49L13.15-19.49Q14.40-10.56 16.51-10.56L16.51-10.56L19.10-10.56Q23.32-10.56 27.45-19.49L27.45-19.49Q30.43-24.57 30.43-28.99L30.43-28.99Q29.08-30.24 27.16-34.65L27.16-34.65L26.30-34.65Q21.21-34.65 15.55-27.74L15.55-27.74Q13.15-23.71 13.15-19.49Z" transform="translate(63.4155967830671 94) " fill="rgb(50, 113, 174)" stroke="rgb(50, 113, 174)" stroke-width="0.164" data-glyph-w="54.872813579942104" data-glyph-h="95.986328125" data-glyph-ascender="84.46796875000001" data-glyph-descender="-11.518359375000001" data-kerning="0"></path>
|
||||||
|
<path d="M14.30-56.34L14.30-56.34Q18.24-56.34 20.25-50.68L20.25-50.68Q20.25-49.05 16.13-46.55L16.13-46.55Q12.96-47.32 12.19-47.99L12.19-47.99Q11.71-47.71 11.04-47.71L11.04-47.71Q9.21-50.20 9.21-50.68L9.21-50.68Q9.89-56.06 14.01-56.06L14.01-56.06Q14.01-56.34 14.30-56.34ZM9.21-39.93L9.50-41.47L10.17-41.47Q11.52-41.47 13.15-41.18L13.15-41.18L14.88-41.47Q16.51-41.47 17.85-39.07L17.85-39.07Q16.99-29.76 16.99-21.79L16.99-21.79Q17.28-20.25 17.28-20.06L17.28-20.06L16.99-18.24Q17.28-15.65 17.28-15.26L17.28-15.26L16.99-13.53Q17.28-10.65 17.28-8.16L17.28-8.16L17.28-5.76Q17.28-5.47 17.57-5.47L17.57-5.47Q16.80-0.96 16.41-0.96L16.41-0.96Q15.84-1.25 15.45-1.25L15.45-1.25L15.17-0.67L15.45-0.10Q15.17 0.77 14.88 0.77L14.88 0.77Q13.92 0.19 13.15 0.19L13.15 0.19Q12.86 0.67 12.86 1.34L12.86 1.34L12.48 1.34Q6.24-4.32 6.24-7.58L6.24-7.58Q8.16-20.06 9.21-38.20L9.21-38.20L9.21-39.93Z" transform="translate(118.2884103630092 94) " fill="rgb(50, 113, 174)" stroke="rgb(50, 113, 174)" stroke-width="0.164" data-glyph-w="27.804669048692098" data-glyph-h="95.986328125" data-glyph-ascender="84.46796875000001" data-glyph-descender="-11.518359375000001" data-kerning="0"></path>
|
||||||
|
<path d="M56.25-68.05L56.25-68.05Q61.34-67.29 61.34-62.97L61.34-62.97L61.34-62.68Q61.14-60.76 56.54-57.30L56.54-57.30Q55.96-56.15 55.96-55.58L55.96-55.58L53.85-55.86L53.56-55.86Q52.98-55.86 52.98-55.00L52.98-55.00Q47.90-55.58 46.17-55.58L46.17-55.58L44.06-55.58Q37.53-55.58 37.53-53.46L37.53-53.46Q37.53-53.08 36.57-52.31L36.57-52.31L36.95-51.06L36.28-49.62L36.57-47.51Q36.57-46.17 36.28-43.67L36.28-43.67Q36.57-38.20 36.57-34.46L36.57-34.46Q36.57-33.79 36.28-31.20L36.28-31.20Q36.57-30.62 36.57-30.24L36.57-30.24L35.99-24.57Q36.28-23.04 36.28-21.31L36.28-21.31Q36.28-20.45 35.99-19.29L35.99-19.29Q36.28-17.66 36.28-16.32L36.28-16.32L36.28-14.49Q34.56-0.77 31.87-0.77L31.87-0.77Q30.81-0.48 30.62-0.48L30.62-0.48Q30.04-0.48 28.60-1.15L28.60-1.15L20.83-0.48L20.83-1.44L19.68-5.86Q21.79-5.86 25.05-37.72L25.05-37.72Q25.05-40.31 25.34-52.02L25.34-52.02L25.63-53.46L25.05-53.75Q14.30-53.46 14.30-52.89L14.30-52.89L12.77-53.18L9.21-52.02Q5.09-52.02 5.09-58.55L5.09-58.55Q4.80-58.55 3.55-60.95L3.55-60.95L3.55-62.10Q3.65-62.68 4.22-62.68L4.22-62.68L4.80-62.68Q6.82-62.68 8.06-61.53L8.06-61.53L9.50-62.10L11.61-61.82Q40.60-66.23 45.88-66.23L45.88-66.23L50.30-66.23Q56.25-67.57 56.25-68.05Z" transform="translate(146.0930794117013 94) " fill="rgb(50, 113, 174)" stroke="rgb(50, 113, 174)" stroke-width="0.164" data-glyph-w="68.2149131893171" data-glyph-h="95.986328125" data-glyph-ascender="84.46796875000001" data-glyph-descender="-11.518359375000001" data-kerning="0"></path>
|
||||||
|
<path d="M26.49-40.41L26.49-40.41Q44.63-33.31 44.63-24.67L44.63-24.67L44.63-22.27Q44.63-16.32 39.83-11.81L39.83-11.81Q36.86-6.43 30.62-4.13L30.62-4.13Q23.61-2.02 21.69-2.02L21.69-2.02Q13.73-2.02 6.82-12.09L6.82-12.09Q5.95-14.78 5.95-15.74L5.95-15.74L5.95-17.76Q7.01-31.20 8.64-31.20L8.64-31.20Q15.74-40.12 21.98-40.12L21.98-40.12L24.67-39.83Q25.92-40.41 26.49-40.41ZM27.64-30.33L24.96-30.33Q24.96-28.51 18.72-28.51L18.72-28.51L17.85-28.51Q14.88-28.51 13.05-21.69L13.05-21.69L13.05-19.01Q14.21-13.05 17.28-13.05L17.28-13.05Q17.37-12.48 17.85-12.48L17.85-12.48L21.12-12.48Q28.03-12.48 32.44-22.56L32.44-22.56L32.73-25.24Q31.77-30.33 27.64-30.33L27.64-30.33Z" transform="translate(214.30799260101838 94) " fill="rgb(50, 113, 174)" stroke="rgb(50, 113, 174)" stroke-width="0.164" data-glyph-w="54.200909283067105" data-glyph-h="95.986328125" data-glyph-ascender="84.46796875000001" data-glyph-descender="-11.518359375000001" data-kerning="0"></path>
|
||||||
|
<path d="M26.49-40.41L26.49-40.41Q44.63-33.31 44.63-24.67L44.63-24.67L44.63-22.27Q44.63-16.32 39.83-11.81L39.83-11.81Q36.86-6.43 30.62-4.13L30.62-4.13Q23.61-2.02 21.69-2.02L21.69-2.02Q13.73-2.02 6.82-12.09L6.82-12.09Q5.95-14.78 5.95-15.74L5.95-15.74L5.95-17.76Q7.01-31.20 8.64-31.20L8.64-31.20Q15.74-40.12 21.98-40.12L21.98-40.12L24.67-39.83Q25.92-40.41 26.49-40.41ZM27.64-30.33L24.96-30.33Q24.96-28.51 18.72-28.51L18.72-28.51L17.85-28.51Q14.88-28.51 13.05-21.69L13.05-21.69L13.05-19.01Q14.21-13.05 17.28-13.05L17.28-13.05Q17.37-12.48 17.85-12.48L17.85-12.48L21.12-12.48Q28.03-12.48 32.44-22.56L32.44-22.56L32.73-25.24Q31.77-30.33 27.64-30.33L27.64-30.33Z" transform="translate(268.5089018840855 94) " fill="rgb(50, 113, 174)" stroke="rgb(50, 113, 174)" stroke-width="0.164" data-glyph-w="54.200909283067105" data-glyph-h="95.986328125" data-glyph-ascender="84.46796875000001" data-glyph-descender="-11.518359375000001" data-kerning="0"></path>
|
||||||
|
<path d="M18.33-68.73L18.33-68.73Q22.56-67.86 22.56-66.71L22.56-66.71Q21.31-61.34 21.31-59.80L21.31-59.80L21.31-57.50Q20.16-35.04 18.72-21.12L18.72-21.12L18.72-19.10L19.58-19.10Q19.58-19.49 24.28-22.36L24.28-22.36Q27.64-25.34 28.22-25.34L28.22-25.34L28.22-24.76Q25.72-19.20 23.71-11.33L23.71-11.33Q20.16-7.68 20.16-5.38L20.16-5.38Q20.16-2.98 13.34-2.40L13.34-2.40Q13.34-2.11 13.05-2.11L13.05-2.11L5.86-5.95Q5.28-7.77 4.99-7.77L4.99-7.77Q5.57-9.31 5.57-13.15L5.57-13.15L5.57-13.44Q8.64-13.44 10.65-52.41L10.65-52.41Q11.23-62.20 11.23-64.60L11.23-64.60Q10.75-64.89 10.08-64.89L10.08-64.89L10.08-65.46Q10.65-67.77 13.63-68.44L13.63-68.44Q14.11-68.44 18.33-68.73Z" transform="translate(322.7098111671526 94) " fill="rgb(50, 113, 174)" stroke="rgb(50, 113, 174)" stroke-width="0.164" data-glyph-w="35.2916026424421" data-glyph-h="95.986328125" data-glyph-ascender="84.46796875000001" data-glyph-descender="-11.518359375000001" data-kerning="0"></path>
|
||||||
|
</g>
|
||||||
|
</svg>
|
||||||
|
</g>
|
||||||
|
</g>
|
||||||
|
<g id="shape_HBt9Poo2sR" ref="shape_HBt9Poo2sR" key="shape_HBt9Poo2sR" mask="">
|
||||||
|
<g transform="translate(290.23123855735,120) rotate(0,109.76876144264,68.206715750285) scale(1,1)" style="opacity: 1;mix-blend-mode: undefined;" filter="url(#filter_shadow_shape_HBt9Poo2sR)">
|
||||||
|
<!--普通元素-->
|
||||||
|
<svg version="1.1" id="圖層_1"
|
||||||
|
xmlns="http://www.w3.org/2000/svg"
|
||||||
|
xmlns:xlink="http://www.w3.org/1999/xlink" x="0px" y="0px" viewBox="0 -8.068921089172363 315.3833923339844 195.9689178466797" style="enable-background:new 0 0 304.3 187.9;" xml:space="preserve" width="219.53752288528" height="136.41343150057" class="style-removed" preserveAspectRatio="none" data-parent="shape_HBt9Poo2sR">
|
||||||
|
|
||||||
|
<g>
|
||||||
|
<path d="M214.9,124.7L214.9,124.7c1.3,4.5,2.7,9.1,4.2,13.7c12.6,40.3,27.1,42.9,27.1,42.9l4.4,0.4h0.3
|
||||||
|
c0,0,15.8-3.6,26.7-45c7.4-28.2,11.5-43.9,16.5-62h-64.6C225,91.1,220.3,107.5,214.9,124.7z" fill="rgb(50, 113, 174)" stroke-width="0"></path>
|
||||||
|
<path d="M88.6,6.5c-0.5-0.2-0.9-0.4-1.3-0.6h-2.8c0,0-13.2-1.6-22,30.7C52.8,72.3,44.9,105,33,141.8
|
||||||
|
c-14.7,45.5-33,46.1-33,46.1h72.9c0,0,16.6-3.6,27.5-46.3c9.1-35.9,12.6-52.2,19.5-77.5c-1.5-5.6-3.1-11.7-5-18.5
|
||||||
|
C105.9,13.5,90.3,7.2,88.6,6.5z" fill="rgb(50, 113, 174)" stroke-width="0"></path>
|
||||||
|
<path d="M212.7,140.4c-0.5-1.5-1-3.1-1.4-4.6c-11.1-36.2-18.8-68.7-28.1-104.1C175.3,1.4,163.6,0,161.6,0
|
||||||
|
c-0.2,0-0.3,0-0.3,0H90.8c-0.1,0-0.2,0.2-0.1,0.2c1.5,0.5,4.9,2,9.2,5.6c6.7,5.7,15.4,16.8,21.3,37.9c0.7,2.7,1.5,5.2,2.1,7.7
|
||||||
|
c8.6,31.4,11.7,47.7,21.8,88.6c10.9,44.1,27.5,47.8,27.5,47.8h72.8c-0.7-0.1-5-0.6-11-6.3C229.4,177.3,220.5,165.4,212.7,140.4z" fill="rgb(50, 113, 174)" stroke-width="0"></path>
|
||||||
|
|
||||||
|
<ellipse transform="matrix(0.8694 -0.4941 0.4941 0.8694 17.8966 137.5734)" cx="269.2" cy="34.9" rx="37" ry="28.4" fill="rgb(50, 113, 174)" stroke-width="0"></ellipse>
|
||||||
|
</g>
|
||||||
|
</svg>
|
||||||
|
</g>
|
||||||
|
</g>
|
||||||
|
</svg>
|
||||||
|
After Width: | Height: | Size: 11 KiB |
BIN
resources/icon11.ico
Normal file
BIN
resources/icon11.ico
Normal file
Binary file not shown.
|
After Width: | Height: | Size: 183 KiB |
BIN
resources/icon_1.png
Normal file
BIN
resources/icon_1.png
Normal file
Binary file not shown.
|
After Width: | Height: | Size: 35 KiB |
BIN
resources/icon_2.png
Normal file
BIN
resources/icon_2.png
Normal file
Binary file not shown.
|
After Width: | Height: | Size: 1.2 MiB |
Binary file not shown.
|
After Width: | Height: | Size: 5.3 MiB |
Binary file not shown.
|
After Width: | Height: | Size: 5.0 MiB |
Binary file not shown.
|
After Width: | Height: | Size: 5.7 MiB |
Binary file not shown.
|
After Width: | Height: | Size: 5.6 MiB |
Binary file not shown.
|
After Width: | Height: | Size: 5.4 MiB |
172
src/define/define_string.js
Normal file
172
src/define/define_string.js
Normal file
@ -0,0 +1,172 @@
|
|||||||
|
export const DEFINE_STRING = {
|
||||||
|
SAVE_DEFINE_CONFIG_JSON_BY_PROPERTY: "SAVE_DEFINE_CONFIG_JSON_BY_PROPERTY",
|
||||||
|
GET_DEFINE_CONFIG_JSON_BY_PROPERTY: "GET_DEFINE_CONFIG_JSON_BY_PROPERTY",
|
||||||
|
GET_IMAGE_GENERATE_CATEGORY: "GET_IMAGE_GENERATE_CATEGORY",
|
||||||
|
SHOW_MAIN_NOTIFICATION: "SHOW_MAIN_NOTIFICATION",
|
||||||
|
CHECK_MACHINE_ID: "CHECK_MACHINE_ID",
|
||||||
|
GET_CUSTOMIZE_GPT_PROMPT: "GET_CUSTOMIZE_GPT_PROMPT",
|
||||||
|
GENERATE_GPT_EXAMPLE_OUT: "GENERATE_GPT_EXAMPLE_OUT",
|
||||||
|
GET_PERMISSION: "GET_PERMISSION",
|
||||||
|
SAVE_IMAGE_TO_OTHER_FOLDER: "SAVE_IMAGE_TO_OTHER_FOLDER",
|
||||||
|
GET_IMAGE_AUTO_SAVE_SETTING: "GET_IMAGE_AUTO_SAVE_SETTING",
|
||||||
|
SAVE_IMAGE_AUTO_SAVE_SETTING: "SAVE_IMAGE_AUTO_SAVE_SETTING",
|
||||||
|
GET_AUTO_SAVE_IMAGE_CLASSIFY_OPTIONS: "GET_AUTO_SAVE_IMAGE_CLASSIFY_OPTIONS",
|
||||||
|
MODIFY_GENERATE_TASK_STATUS: "MODIFY_GENERATE_TASK_STATUS",
|
||||||
|
DELETE_BACK_TASK: "DELETE_BACK_TASK",
|
||||||
|
SAVE_VIDEO_SRT_AND_AUDIO_MESSAGE: "SAVE_VIDEO_SRT_AND_AUDIO_MESSAGE",
|
||||||
|
SAVE_KEY_FRAME_SETTING: "SAVE_KEY_FRAME_SETTING",
|
||||||
|
MODIFY_SAMPLE_SETTING: "MODIFY_SAMPLE_SETTING",
|
||||||
|
GET_SETTING_Dafault_DATA: "GET_SETTING_Dafault_DATA",
|
||||||
|
GET_DRAFT_FILE_LIST: "GET_DRAFT_FILE_LIST",
|
||||||
|
GET_FRAME: "GET_FRAME",
|
||||||
|
PYTHON_ERROR: "PYTHON_ERROR",
|
||||||
|
PYTHON_CLOSE: "PYTHON_CLOSE",
|
||||||
|
PYTHON_OUTPUT: "PYTHON_OUTPUT",
|
||||||
|
RESTART_GENERATE: "RESTART_GENERATE",
|
||||||
|
ALIGN_DRAFT_IMG: "ALIGN_DRAFT_IMG",
|
||||||
|
ALIGN_DRAFT_IMG_TO_TEXT: "ALIGN_DRAFT_IMG_TO_TEXT",
|
||||||
|
REGENERATE_IMAGE_RETUN: "REGENERATE_IMAGE_RETUN",
|
||||||
|
GET_SUBFOLDER_LIST: "GET_SUBFOLDER_LIST",
|
||||||
|
REFRASH_IMAGWE_DATA: "REFRASH_IMAGWE_DATA",
|
||||||
|
GET_IMAGE_PROMPTLIST: "GET_IMAGE_PROMPTLIST",
|
||||||
|
SELECT_FILE: "SELECT_FILE",
|
||||||
|
IMPROVE_IMAGE_RESOULTION: "IMPROVE_IMAGE_RESOULTION",
|
||||||
|
GET_BACKGROUND_MUSIC_CONFIG_LIST: "GET_BACKGROUND_MUSIC_CONFIG_LIST",
|
||||||
|
ADD_BACKGROUND_MUSIC_FOLDER: "ADD_BACKGROUND_MUSIC_FOLDER",
|
||||||
|
DELETE_CLIP_SETTING: "DELETE_CLIP_SETTING",
|
||||||
|
DELETE_FRIENDLY_REMINDER: "DELETE_FRIENDLY_REMINDER",
|
||||||
|
MODIFY_INPUT_CROP_JSON: "MODIFY_INPUT_CROP_JSON",
|
||||||
|
PUSH_BACK_PROMPT: "PUSH_BACK_PROMPT",
|
||||||
|
GET_FRIENDLY_REMINDER_DRAFT: "GET_FRIENDLY_REMINDER_DRAFT",
|
||||||
|
GET_FRIENDLY_REMINDER_LIST: "GET_FRIENDLY_REMINDER_LIST",
|
||||||
|
AUTO_GENERATION_VIDEO: "AUTO_GENERATION_VIDEO",
|
||||||
|
GET_PROJECT_WORD: "GET_PROJECT_WORD",
|
||||||
|
AIMODIFY_ONE_WORD: "AIMODIFY_ONE_WORD",
|
||||||
|
IMPORT_SRT_AND_GET_TIME: "IMPORT_SRT_AND_GET_TIME",
|
||||||
|
SAVE_NEW_WORD: "SAVE_NEW_WORD",
|
||||||
|
SAVE_COPYWRITING_INFOMATION: "SAVE_COPYWRITING_INFOMATION",
|
||||||
|
SAVE_SD_CONFIG: "SAVE_SD_CONFIG",
|
||||||
|
SAVE_GENERAL_SETTING: "SAVE_GENERAL_SETTING",
|
||||||
|
GET_VIDEO_CONFIG_MESSAGE: "GET_VIDEO_CONFIG_MESSAGE",
|
||||||
|
GET_SYSTEM_INSTALL_FONTNAME: "GET_SYSTEM_INSTALL_FONTNAME",
|
||||||
|
SAVE_ASS_CONFIG: "SAVE_ASS_CONFIG",
|
||||||
|
DELETE_VIDEO_CONFIG: "DELETE_VIDEO_CONFIG",
|
||||||
|
SHOW_NEW_WINDOW: "SHOW_NEW_WINDOW",
|
||||||
|
GET_DRAFT_FILE_LIST: "GET_DRAFT_FILE_LIST",
|
||||||
|
SELECT_FOLDER: "SELECT_FOLDER",
|
||||||
|
GET_DRAFT_TEXT_STYLE: "GET_DRAFT_TEXT_STYLE",
|
||||||
|
GET_TEXT_STYLE_LIST: "GET_TEXT_STYLE_LIST",
|
||||||
|
DELETE_DRAFT_TEXT_STYLE: "DELETE_DRAFT_TEXT_STYLE",
|
||||||
|
ADD_DRAFT: "ADD_DRAFT",
|
||||||
|
RETURN_IMAGE_PROMPT: "RETURN_IMAGE_PROMPT",
|
||||||
|
RE_GENERATE_IAMGE_ONE: "RE_GENERATE_IAMGE_ONE",
|
||||||
|
INIT_SD_CONFIG: "INIT_SD_CONFIG",
|
||||||
|
ADD_IMAGE_TASK_LIST: "ADD_IMAGE_TASK_LIST",
|
||||||
|
GET_GENERATE_TASK_LIST: "GET_GENERATE_TASK_LIST",
|
||||||
|
DELETE_IMAGE_TASK_LIST: "DELETE_IMAGE_TASK_LIST",
|
||||||
|
GENERATE_IMAGWE_IN_SELECT_TASK: "GENERATE_IMAGWE_IN_SELECT_TASK",
|
||||||
|
GET_MACHINE_ID: "GET_MACHINE_ID",
|
||||||
|
QUIT_APP: "QUIT_APP",
|
||||||
|
GET_BAD_PROMPT: "GET_BAD_PROMPT",
|
||||||
|
SAVE_BAD_PROMPT: "SAVE_BAD_PROMPT",
|
||||||
|
DELETE_BAD_PROMPT: "DELETE_BAD_PROMPT",
|
||||||
|
ADD_WEBUI_JSON: "ADD_WEBUI_JSON",
|
||||||
|
OPEN_GPT_BUY_URL: "OPEN_GPT_BUY_URL",
|
||||||
|
GET_IAMGE_PROMPT_LIST: "GET_IAMGE_PROMPT_LIST",
|
||||||
|
GET_ADETAILER_LIST: "GET_ADETAILER_LIST",
|
||||||
|
SAVE_DETAILER_CONFIG: "SAVE_DETAILER_CONFIG",
|
||||||
|
OPEN_URL: "OPEN_URL",
|
||||||
|
GET_VERSION: "GET_VERSION",
|
||||||
|
GET_FRAME_RETUN: "GET_FRAME_RETUN",
|
||||||
|
DOWNLOAD_MODEL: "DOWNLOAD_MODEL",
|
||||||
|
START_STORY_BOARDING: "START_STORY_BOARDING",
|
||||||
|
SHOW_MESSAGE_DIALOG: "SHOW_MESSAGE_DIALOG",
|
||||||
|
SHOW_GLOABAL_MESSAGE_DIALOG: "SHOW_GLOABAL_MESSAGE_DIALOG",
|
||||||
|
IMAGE_TASK_STATUS_REFRESH: "IMAGE_TASK_STATUS_REFRESH",
|
||||||
|
SAVE_TRIAL_END_TIME: "SAVE_TRIAL_END_TIME",
|
||||||
|
DOWNLOAD_IMAGE_FILE: "DOWNLOAD_IMAGE_FILE",
|
||||||
|
OPEN_FOLDER: "OPEN_FOLDER",
|
||||||
|
VIDEO_GENERATE_STATUS_REFRESH: "VIDEO_GENERATE_STATUS_REFRESH",
|
||||||
|
AUTO_CONDITION_CHECK: "AUTO_CONDITION_CHECK",
|
||||||
|
MODIFY_IMAGE_TASK_LIST: "MODIFY_IMAGE_TASK_LIST",
|
||||||
|
ACTION_AUTO_VIDEO_TASK: "ACTION_AUTO_VIDEO_TASK",
|
||||||
|
GET_VIDEO_GENERATE_CONFIG: "GET_VIDEO_GENERATE_CONFIG",
|
||||||
|
TRANSLATE_PROMPT: "TRANSLATE_PROMPT",
|
||||||
|
TRANSLATE_RETURN_NOW: "TRANSLATE_RETURN_NOW",
|
||||||
|
TRANSLATE_RETURN_REFRESH: "TRANSLATE_RETURN_REFRESH",
|
||||||
|
GET_SHOW_MESSAGE: "GET_SHOW_MESSAGE",
|
||||||
|
AUTO_ANALYZE_CHARACTER: "AUTO_ANALYZE_CHARACTER",
|
||||||
|
GET_CONFIG_JSON: "GET_CONFIG_JSON",
|
||||||
|
ORIGINAL_ADD_WEBUI_JSON: "ORIGINAL_ADD_WEBUI_JSON",
|
||||||
|
GET_PROMPT_JSON: "GET_PROMPT_JSON",
|
||||||
|
GPT_PROMPT: "GPT_PROMPT",
|
||||||
|
GPT_GENERATE_PROMPT_RETURN: "GPT_GENERATE_PROMPT_RETURN",
|
||||||
|
AUTO_SAVE_DATA_JSON: "AUTO_SAVE_DATA_JSON",
|
||||||
|
ORIGINAL_SD_SINGLE_IMAGE_GENERATE: "ORIGINAL_SD_SINGLE_IMAGE_GENERATE",
|
||||||
|
SD_ORIGINAL_GENERATE_IMAGE_RETURN: "SD_ORIGINAL_GENERATE_IMAGE_RETURN",
|
||||||
|
GET_STYLE_IMAGE_SUB_LIST: "GET_STYLE_IMAGE_SUB_LIST",
|
||||||
|
GET_IMAGE_STYLE_INFOMATION: "GET_IMAGE_STYLE_INFOMATION",
|
||||||
|
GET_IMAGE_STYLE_MENU: "GET_IMAGE_STYLE_MENU",
|
||||||
|
GET_GPT_BUSINESS_OPTION: "GET_GPT_BUSINESS_OPTION",
|
||||||
|
GET_GPT_MODEL_OPTION: "GET_GPT_MODEL_OPTION",
|
||||||
|
GET_GPT_AUTO_INFERENCE_OPTIONS: "GET_GPT_AUTO_INFERENCE_OPTIONS",
|
||||||
|
SAVE_DYNAMIC_GPT_OPTION: "SAVE_DYNAMIC_GPT_OPTION",
|
||||||
|
DELETE_DYNAMIC_GPT_OPTION: "DELETE_DYNAMIC_GPT_OPTION",
|
||||||
|
TEST_GPT_CONNECTION: "TEST_GPT_CONNECTION",
|
||||||
|
SAVE_WORD_TXT: "SAVE_WORD_TXT",
|
||||||
|
GET_KEY_FRAME_CONFIG_DATA: "GET_KEY_FRAME_CONFIG_DATA",
|
||||||
|
GET_KEYFRAME_OPTIONS: "GET_KEYFRAME_OPTIONS",
|
||||||
|
|
||||||
|
QUEUE_BATCH: {
|
||||||
|
SD_ORIGINAL_GENERATE_IMAGE: "SD_ORIGINAL_GENERATE_IMAGE",
|
||||||
|
SD_ORIGINAL_GPT_PROMPT: "SD_ORIGINAL_GPT_PROMPT",
|
||||||
|
SD_BACKSTEP_GENERATE_IMAGE: "SD_BACKSTEP_GENERATE_IMAGE",
|
||||||
|
MJ_ORIGINAL_GENERATE_IMAGE: "MJ_ORIGINAL_GENERATE_IMAGE",
|
||||||
|
LOCAL_IMAGE_IMPROVE: "LOCAL_IMAGE_IMPROVE",
|
||||||
|
AUTO_VIDEO_GENERATE: "AUTO_VIDEO_GENERATE",
|
||||||
|
AUTO_VIDEO_GENERATE_SINGLE: "AUTO_VIDEO_GENERATE_SINGLE",
|
||||||
|
TRANSLATE_PROMPT: "TRANSLATE_PROMPT",
|
||||||
|
TRANSLATE_RETURN_NOW_TASK: "TRANSLATE_RETURN_NOW_TASK",
|
||||||
|
IMAGE_SAVE_TO_OTHER_FOLDER: "IMAGE_SAVE_TO_OTHER_FOLDER",
|
||||||
|
SAVE_FILE_QUEUE: "SAVE_FILE_QUEUE",
|
||||||
|
},
|
||||||
|
PERMISSIONS: {
|
||||||
|
NORMAL_PERMISSION: "NORMAL_PERMISSION",
|
||||||
|
AUTO_SAVE_IMAGE_PERMISSION: "AUTO_SAVE_IMAGE_PERMISSION",
|
||||||
|
},
|
||||||
|
MJ: {
|
||||||
|
SAVE_WORD_SRT: "SAVE_WORD_SRT",
|
||||||
|
GET_MJ_CONFIG_SRT_INFORMATION: "GET_MJ_CONFIG_SRT_INFORMATION",
|
||||||
|
GET_TAG_DATA_BY_TYPE_AND_PROPERTY: "GET_TAG_DATA_BY_TYPE_AND_PROPERTY",
|
||||||
|
SAVE_TAG_PROPERTY_DATA: "SAVE_TAG_PROPERTY_DATA",
|
||||||
|
DELETE_TAG_PROPERTY_DATA: "DELETE_TAG_PROPERTY_DATA",
|
||||||
|
GET_TAG_SELECT_MODEL: "GET_TAG_SELECT_MODEL",
|
||||||
|
TRANSLATE_RETURN_NOW_TASK: "TRANSLATE_RETURN_NOW_TASK",
|
||||||
|
ORIGINAL_MJ_IMAGE_GENERATE: "ORIGINAL_MJ_IMAGE_GENERATE",
|
||||||
|
GET_CHANNEL_ROBOTS: "GET_CHANNEL_ROBOTS",
|
||||||
|
GET_MJ_GENERATE_CATEGORY: "GET_MJ_GENERATE_CATEGORY",
|
||||||
|
IMAGE_SPLIT: "IMAGE_SPLIT",
|
||||||
|
ADD_MJ_BAD_PROMPT: "ADD_MJ_BAD_PROMPT",
|
||||||
|
MJ_BAD_PROMPT_CHECK: "MJ_BAD_PROMPT_CHECK",
|
||||||
|
GET_GENERATED_MJ_IMAGE_AND_SPLIT: "GET_GENERATED_MJ_IMAGE_AND_SPLIT",
|
||||||
|
DOWNLOAD_IMAGE_URL_AND_SPLIT: "DOWNLOAD_IMAGE_URL_AND_SPLIT"
|
||||||
|
},
|
||||||
|
DISCORD: {
|
||||||
|
OPERATE_REFRASH_DISCORD_URL: "OPERATE_REFRASH_DISCORD_URL",
|
||||||
|
GET_DISCORD_WINDOW_URL: "GET_DISCORD_WINDOW_URL",
|
||||||
|
CREATE_MESSAGE: "CREATE_MESSAGE",
|
||||||
|
UPDATE_MESSAGE: "UPDATE_MESSAGE",
|
||||||
|
DELETE_MESSAGE: "DELETE_MESSAGE",
|
||||||
|
MAIN_DISCORD_MESSAGE_CHANGE: "MAIN_DISCORD_MESSAGE_CHANGE",
|
||||||
|
},
|
||||||
|
DISCORD_REQUEST_LISTENER_TYPE: {
|
||||||
|
INPUT_MODEL_IMAGINE_REQUEST: "INPUT_MODEL_IMAGINE_REQUEST",
|
||||||
|
},
|
||||||
|
DISCORD_SIMPLE_DATA_TYPE: {
|
||||||
|
URL: "URL",
|
||||||
|
TOKEN: "TOKEN",
|
||||||
|
},
|
||||||
|
MAIN: {
|
||||||
|
OPEN_DISCORD_WINDOW: "OPEN_DISCORD_WINDOW"
|
||||||
|
}
|
||||||
|
}
|
||||||
2265
src/define/iamgeStyleDefine.js
Normal file
2265
src/define/iamgeStyleDefine.js
Normal file
File diff suppressed because it is too large
Load Diff
72
src/main/IPCEvent/settingIpc.js
Normal file
72
src/main/IPCEvent/settingIpc.js
Normal file
@ -0,0 +1,72 @@
|
|||||||
|
const {
|
||||||
|
ipcMain
|
||||||
|
} = require("electron")
|
||||||
|
import { DEFINE_STRING } from '../../define/define_string'
|
||||||
|
import {
|
||||||
|
Setting
|
||||||
|
} from '../setting/setting'
|
||||||
|
let setting = new Setting(global);
|
||||||
|
|
||||||
|
function SettingIpc() {
|
||||||
|
|
||||||
|
// 获取背景音乐配置列表
|
||||||
|
ipcMain.handle(DEFINE_STRING.GET_BACKGROUND_MUSIC_CONFIG_LIST, async (event) => await setting.GetBackGroundMusicConfigList());
|
||||||
|
|
||||||
|
// 获取剪映关键帧配置列表
|
||||||
|
ipcMain.handle(DEFINE_STRING.GET_KEYFRAME_OPTIONS, async (event) => await setting.GetKeyFrameOptions());
|
||||||
|
// 保存剪映关键帧配置
|
||||||
|
ipcMain.handle(DEFINE_STRING.SAVE_KEY_FRAME_SETTING, async (event, value) => await setting.SaveKeyFrameSetting(value));
|
||||||
|
// 监听添加背景音乐文件
|
||||||
|
ipcMain.handle(DEFINE_STRING.ADD_BACKGROUND_MUSIC_FOLDER, async (event, value) => await setting.AddBackgroundMusicFolder(value))
|
||||||
|
// 删除剪映的样式设置
|
||||||
|
ipcMain.handle(DEFINE_STRING.DELETE_DRAFT_TEXT_STYLE, async (event, value) => await setting.deleteClipSetting("text_style", value));
|
||||||
|
|
||||||
|
// 删除剪映的背景音乐设置
|
||||||
|
ipcMain.handle(DEFINE_STRING.DELETE_CLIP_SETTING, async (event, value) => await setting.deleteClipSetting("background_music_setting", value));
|
||||||
|
// 删除剪映的友情提示设置
|
||||||
|
ipcMain.handle(DEFINE_STRING.DELETE_FRIENDLY_REMINDER, async (event, value) => await setting.deleteClipSetting("friendly_reminder_setting", value));
|
||||||
|
|
||||||
|
// 监听获取SD配置任务
|
||||||
|
ipcMain.handle(DEFINE_STRING.INIT_SD_CONFIG, async (event, value) => await setting.InitSDConfig());
|
||||||
|
|
||||||
|
// 获取主页显示信息
|
||||||
|
ipcMain.handle(DEFINE_STRING.GET_SHOW_MESSAGE, async (event) => await setting.GetShowMessage())
|
||||||
|
// 获取关键帧的配置数据
|
||||||
|
ipcMain.handle(DEFINE_STRING.GET_KEY_FRAME_CONFIG_DATA, async (event) => await setting.GetKeyFrameConfigData());
|
||||||
|
|
||||||
|
// 删除后台队列任务
|
||||||
|
ipcMain.handle(DEFINE_STRING.DELETE_BACK_TASK, async (event, value) => await setting.RemoveTask(value));
|
||||||
|
|
||||||
|
// 获取自动保存图片的分类方式
|
||||||
|
ipcMain.handle(DEFINE_STRING.GET_AUTO_SAVE_IMAGE_CLASSIFY_OPTIONS, async (event) => await setting.GetAutoSaveImageClassifyOptions());
|
||||||
|
|
||||||
|
// 保存图片自动保存的配置
|
||||||
|
ipcMain.handle(DEFINE_STRING.SAVE_IMAGE_AUTO_SAVE_SETTING, async (event, value) => await setting.SaveImageAutoSaveSetting(value));
|
||||||
|
|
||||||
|
// 获取当前的自动保存图片的设置
|
||||||
|
ipcMain.handle(DEFINE_STRING.GET_IMAGE_AUTO_SAVE_SETTING, async (event) => await setting.GetImageAutoSaveSetting());
|
||||||
|
|
||||||
|
// 开始手动保存图片
|
||||||
|
ipcMain.handle(DEFINE_STRING.SAVE_IMAGE_TO_OTHER_FOLDER, async (event, value) => await setting.SaveImageToOtherFolder(value));
|
||||||
|
// 检查机器码是否存在
|
||||||
|
ipcMain.handle(DEFINE_STRING.CHECK_MACHINE_ID, async (event, value) => await setting.CheckMachineId(value));
|
||||||
|
|
||||||
|
//修改剪映草稿配置
|
||||||
|
ipcMain.handle(DEFINE_STRING.MODIFY_SAMPLE_SETTING, async (event, value) => await setting.ModifySampleSetting(value));
|
||||||
|
|
||||||
|
// 获取选择角色场景模式的options
|
||||||
|
ipcMain.handle(DEFINE_STRING.MJ.GET_TAG_SELECT_MODEL, async (event) => await setting.GetRoleSceneModeOptions());
|
||||||
|
|
||||||
|
// 获取当前生成图片的生图方式（sd,mj,d3）
|
||||||
|
ipcMain.handle(DEFINE_STRING.GET_IMAGE_GENERATE_CATEGORY, async (event) => await setting.GetImageGenerateCategory());
|
||||||
|
|
||||||
|
// // 获取指定的配置文件里面指定的属性的数据
|
||||||
|
ipcMain.handle(DEFINE_STRING.GET_DEFINE_CONFIG_JSON_BY_PROPERTY, async (event, value) => await setting.GetDefineConfigJsonByProperty(value))
|
||||||
|
|
||||||
|
// // 保存指定的配置文件里面指定的属性的数据
|
||||||
|
ipcMain.handle(DEFINE_STRING.SAVE_DEFINE_CONFIG_JSON_BY_PROPERTY, async (event, value) => await setting.SaveDefineConfigJsonByProperty(value))
|
||||||
|
}
|
||||||
|
|
||||||
|
export {
|
||||||
|
SettingIpc
|
||||||
|
}
|
||||||
28
src/main/IPCEvent/writingIpc.js
Normal file
28
src/main/IPCEvent/writingIpc.js
Normal file
@ -0,0 +1,28 @@
|
|||||||
|
import {
|
||||||
|
ipcMain
|
||||||
|
} from "electron";
|
||||||
|
import { DEFINE_STRING } from '../../define/define_string'
|
||||||
|
import {
|
||||||
|
Writing
|
||||||
|
} from '../backPrompt/writing'
|
||||||
|
let writing = new Writing(global);
|
||||||
|
|
||||||
|
function WritingIpc() {
|
||||||
|
// 监听分镜时间的保存
|
||||||
|
ipcMain.handle(DEFINE_STRING.SAVE_COPYWRITING_INFOMATION, async (event, value) => await writing.SaveCopywritingInformation(value));
|
||||||
|
|
||||||
|
// 监听获取当前项目下面的分镜文案
|
||||||
|
ipcMain.handle(DEFINE_STRING.GET_PROJECT_WORD, async (event, value) => await writing.GetProjectWord());
|
||||||
|
|
||||||
|
// 获取config配置文件数据
|
||||||
|
ipcMain.handle(DEFINE_STRING.GET_CONFIG_JSON, async (event, value) => await writing.GetConfigJson(value));
|
||||||
|
|
||||||
|
// 将指定的文案txt数组写入到指定的文件中
|
||||||
|
ipcMain.handle(DEFINE_STRING.SAVE_WORD_TXT, async (event, value) => await writing.SaveWordTxt(value));
|
||||||
|
|
||||||
|
// 监听获取字幕时间
|
||||||
|
ipcMain.handle(DEFINE_STRING.IMPORT_SRT_AND_GET_TIME, async (event, value) => await writing.ImportSrtAndGetTime(value))
|
||||||
|
}
|
||||||
|
export {
|
||||||
|
WritingIpc
|
||||||
|
}
|
||||||
33
src/main/Original/TagCustomize.js
Normal file
33
src/main/Original/TagCustomize.js
Normal file
@ -0,0 +1,33 @@
|
|||||||
|
import { TagDefine } from "../../define/tagDefine";
|
||||||
|
export class TagCustomize {
|
||||||
|
constructor(global) {
|
||||||
|
this.global = global;
|
||||||
|
this.tagDefine = new TagDefine(global);
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 获取默认的tag
|
||||||
|
* @param {*} value
|
||||||
|
*/
|
||||||
|
async GetTagDataByTypeAndProperty(value) {
|
||||||
|
return await this.tagDefine.getTagDataByTypeAndProperty(value[0], value[1]);
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 保存数据到指定的tag中
|
||||||
|
* @param {*} value
|
||||||
|
* @returns
|
||||||
|
*/
|
||||||
|
async SaveTagPropertyData(value) {
|
||||||
|
return await this.tagDefine.saveTagPropertyData(value);
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 删除指定的tag数据
|
||||||
|
* @param {*} value
|
||||||
|
* @returns
|
||||||
|
*/
|
||||||
|
async DeleteTagPropertyData(value) {
|
||||||
|
return await this.tagDefine.deleteTagPropertyData(value);
|
||||||
|
}
|
||||||
|
}
|
||||||
493
src/main/backPrompt/imageGenerate.js
Normal file
493
src/main/backPrompt/imageGenerate.js
Normal file
@ -0,0 +1,493 @@
|
|||||||
|
|
||||||
|
import { Tools } from "../tools";
|
||||||
|
import path from "path";
|
||||||
|
import { DEFINE_STRING } from "../../define/define_string";
|
||||||
|
import { define } from "../../define/define";
|
||||||
|
import { PublicMethod } from "../Public/publicMethod";
|
||||||
|
import { SD } from "../Public/SD"
|
||||||
|
const util = require('util');
|
||||||
|
const { spawn, exec } = require('child_process');
|
||||||
|
import axios from "axios";
|
||||||
|
const sharp = require('sharp');
|
||||||
|
const execAsync = util.promisify(exec);
|
||||||
|
const { v4: uuidv4 } = require('uuid'); // 引入UUID库来生成唯一标识符
|
||||||
|
let fspromises = require("fs").promises;
|
||||||
|
import { MD5 } from "crypto-js";
|
||||||
|
import { ImageSetting } from "../../define/setting/imageSetting";
|
||||||
|
|
||||||
|
|
||||||
|
export class ImageGenerate {
|
||||||
|
constructor(global) {
|
||||||
|
this.global = global;
|
||||||
|
this.tools = new Tools();
|
||||||
|
this.pm = new PublicMethod(global);
|
||||||
|
this.sd = new SD(global);
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 获取指定名称的提示词json文件数据
|
||||||
|
* @param {*} value
|
||||||
|
*/
|
||||||
|
async GetPromptJson(value) {
|
||||||
|
try {
|
||||||
|
let json_path = path.join(this.global.config.project_path, "tmp/input_crop/" + value + ".json");
|
||||||
|
let isExist = await this.tools.checkExists(json_path);
|
||||||
|
if (!isExist) {
|
||||||
|
throw new Error("提示词文件不存在");
|
||||||
|
}
|
||||||
|
let res = JSON.parse(await fspromises.readFile(json_path, 'utf-8'));
|
||||||
|
return {
|
||||||
|
value: 1,
|
||||||
|
data: res
|
||||||
|
}
|
||||||
|
} catch (error) {
|
||||||
|
return {
|
||||||
|
code: 0,
|
||||||
|
message: error.toString()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 修改指定任务队列的状态
|
||||||
|
* @param {*} value 0 使用什么查找（id。out_folder,） 1查找的值 2 状态值
|
||||||
|
* @returns
|
||||||
|
*/
|
||||||
|
async ModifyGenerateTaskStatus(value) {
|
||||||
|
try {
|
||||||
|
await this.pm.ModifyTaskStatus(value[0], value[1], value[2]);
|
||||||
|
return {
|
||||||
|
code: 1
|
||||||
|
}
|
||||||
|
} catch (error) {
|
||||||
|
return {
|
||||||
|
code: 0,
|
||||||
|
message: error.toString()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 修改本地的参数
|
||||||
|
* @param {修改json的参数} value
|
||||||
|
*/
|
||||||
|
async ModifyInpurCropJson(value) {
|
||||||
|
try {
|
||||||
|
value = JSON.parse(value);
|
||||||
|
let prompt = value.prompt;
|
||||||
|
// 重绘幅度
|
||||||
|
let denoising_strength = value.denoising_strength;
|
||||||
|
let json_path = path.join(path.join(global.config.project_path, 'tmp/input_crop'), value.name + ".json");
|
||||||
|
let old_json = JSON.parse(await fspromises.readFile(json_path, 'utf-8'));
|
||||||
|
old_json.webui_config.prompt = prompt;
|
||||||
|
old_json.webui_config.denoising_strength = denoising_strength;
|
||||||
|
old_json.model = value.model;
|
||||||
|
old_json.adetailer = value.adetailer;
|
||||||
|
old_json.chinese_prompt = value.chinese_prompt;
|
||||||
|
|
||||||
|
if (value.modify_old) {
|
||||||
|
old_json.webui_config.init_images = value.modify_old;
|
||||||
|
}
|
||||||
|
|
||||||
|
await fspromises.writeFile(json_path, JSON.stringify(old_json));
|
||||||
|
return {
|
||||||
|
code: 1
|
||||||
|
}
|
||||||
|
} catch (error) {
|
||||||
|
throw new Error(error);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 返回文件夹里的所有的图片，以及其他信息
|
||||||
|
* @param {图片文件夹} value
|
||||||
|
* @returns
|
||||||
|
*/
|
||||||
|
async getFolderImageList(mainWindow, value, isReturn) {
|
||||||
|
try {
|
||||||
|
let img_dir = path.normalize(value);
|
||||||
|
let base_name = path.basename(img_dir);
|
||||||
|
let tmp = path.dirname(img_dir);
|
||||||
|
let input_path = path.join(tmp, "input_crop")
|
||||||
|
// console.log(tmp, input_path)
|
||||||
|
let files = await fspromises.readdir(img_dir)
|
||||||
|
let imageFiles = files.filter(file => /\.(png)$/i.test(file));
|
||||||
|
imageFiles.sort();
|
||||||
|
|
||||||
|
let res = [];
|
||||||
|
let task_list_json = JSON.parse(await fspromises.readFile(path.join(this.global.config.project_path, 'scripts/task_list.json'), 'utf-8'));
|
||||||
|
// 处理数据
|
||||||
|
for (let i = 0; i < imageFiles.length; i++) {
|
||||||
|
const img = imageFiles[i];
|
||||||
|
let imput_path = path.join(input_path, img);
|
||||||
|
let img_json = path.join(input_path, img + '.json');
|
||||||
|
// 判断文件是不是存在
|
||||||
|
let prompt_json = JSON.parse(await fspromises.readFile(img_json));
|
||||||
|
|
||||||
|
let output_list = await this.pm.getSubFolderList(path.join(this.global.config.project_path, "tmp"), 'start', 'output_crop');
|
||||||
|
let output_image_list = [];
|
||||||
|
for (let i = 0; i < output_list.length; i++) {
|
||||||
|
const item = output_list[i];
|
||||||
|
// 找到找到文件对应的任务列表的样式和lora
|
||||||
|
let task_list = task_list_json.task_list.filter(a => a.out_folder == item);
|
||||||
|
let output_1 = path.join(tmp, item);
|
||||||
|
let obj = {
|
||||||
|
id: uuidv4(),
|
||||||
|
img_path: path.join(output_1, img),
|
||||||
|
image_style: task_list[0].image_style,
|
||||||
|
lora: task_list[0].lora
|
||||||
|
}
|
||||||
|
output_image_list.push(obj);
|
||||||
|
|
||||||
|
}
|
||||||
|
let main_task_list = task_list_json.task_list.filter(a => a.out_folder == base_name);
|
||||||
|
let obj = {
|
||||||
|
name: img,
|
||||||
|
id: uuidv4(),
|
||||||
|
sampler_name: prompt_json.webui_config.sampler_name,
|
||||||
|
prompt: prompt_json.webui_config.prompt,
|
||||||
|
chinese_prompt: prompt_json.chinese_prompt,
|
||||||
|
negative_prompt: prompt_json.webui_config.negative_prompt,
|
||||||
|
denoising_strength: prompt_json.webui_config.denoising_strength,
|
||||||
|
seed: prompt_json.webui_config.seed,
|
||||||
|
image: path.join(img_dir, img),
|
||||||
|
oldImage: prompt_json.webui_config.init_images,
|
||||||
|
output_list: output_image_list,
|
||||||
|
model: prompt_json.model,
|
||||||
|
image_style: main_task_list[0].image_style,
|
||||||
|
lora: main_task_list[0].lora,
|
||||||
|
image_style_list: main_task_list[0].image_style_list,
|
||||||
|
adetailer: prompt_json.adetailer,
|
||||||
|
width: prompt_json.webui_config.width,
|
||||||
|
height: prompt_json.webui_config.height
|
||||||
|
}
|
||||||
|
res.push(obj)
|
||||||
|
// 获取提示词
|
||||||
|
}
|
||||||
|
if (isReturn) {
|
||||||
|
return {
|
||||||
|
code: 1,
|
||||||
|
windowID: mainWindow.id,
|
||||||
|
value: res
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
mainWindow.webContents.send(DEFINE_STRING.RETURN_IMAGE_PROMPT, {
|
||||||
|
code: 1,
|
||||||
|
windowID: mainWindow.id,
|
||||||
|
value: res
|
||||||
|
})
|
||||||
|
}
|
||||||
|
} catch (error) {
|
||||||
|
return {
|
||||||
|
code: 0,
|
||||||
|
message: `Error message ${error.toString()}`
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 将反推的图片的信息添加到一个json文件中
|
||||||
|
*/
|
||||||
|
async AddWebuiJson() {
|
||||||
|
try {
|
||||||
|
return await this.pm.AddWebuiJson();
|
||||||
|
} catch (error) {
|
||||||
|
return {
|
||||||
|
code: 0,
|
||||||
|
message: error.toString()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 修改任务队列数据
|
||||||
|
* @param {替换后的值} value
|
||||||
|
*/
|
||||||
|
async ModifyImageTaskList(value) {
|
||||||
|
try {
|
||||||
|
let res = await this.pm.ModifyImageTaskList(value);
|
||||||
|
if (res) {
|
||||||
|
return {
|
||||||
|
code: 1,
|
||||||
|
message: "替换成功"
|
||||||
|
}
|
||||||
|
}
|
||||||
|
} catch (error) {
|
||||||
|
return {
|
||||||
|
code: 0,
|
||||||
|
message: error.toString()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 获取生成图片的任务队列（可以添加部分判断）
|
||||||
|
*/
|
||||||
|
async GetGenerateTaskList() {
|
||||||
|
try {
|
||||||
|
return await this.pm.GetImageTask()
|
||||||
|
} catch (error) {
|
||||||
|
return {
|
||||||
|
code: 0,
|
||||||
|
message: error.message
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 生成任务中的队列
|
||||||
|
* @param {传入所有执行生图队列的任务} value
|
||||||
|
*/
|
||||||
|
async GenerateImageInSelectTask(value) {
|
||||||
|
try {
|
||||||
|
let batch = DEFINE_STRING.QUEUE_BATCH.SD_BACKSTEP_GENERATE_IMAGE
|
||||||
|
let taskPath = path.join(this.global.config.project_path, 'scripts/task_list.json');
|
||||||
|
let task_list_json = JSON.parse(await fspromises.readFile(taskPath, 'utf-8'));
|
||||||
|
let auto_save_image = await this.tools.getJsonFilePropertyValue(define.img_base, "auto_save_image", {}, false);
|
||||||
|
let png_files = [];
|
||||||
|
// 读取队列中的任务，然后添加队列
|
||||||
|
for (let i = 0; i < value.length; i++) {
|
||||||
|
const element = value[i];
|
||||||
|
let task_list = task_list_json.task_list.filter(item => item.id == element)[0];
|
||||||
|
let seed = -1;
|
||||||
|
let images = await this.tools.getFilesWithExtensions(path.join(this.global.config.project_path, 'tmp/input_crop'), '.png');
|
||||||
|
await fspromises.mkdir(path.join(this.global.config.project_path, 'tmp/' + task_list.out_folder), { recursive: true });
|
||||||
|
if (images.length <= 0) {
|
||||||
|
throw new Error("未检测到抽帧图片。请检查");
|
||||||
|
}
|
||||||
|
if (images.length > auto_save_image.save_match_count) {
|
||||||
|
png_files = await this.tools.getFilesWithExtensions(auto_save_image.main_save_folder, '.png');
|
||||||
|
}
|
||||||
|
this.global.requestQuene.enqueue(async () => {
|
||||||
|
let res = await this.sd.OneImageGeneration(images[0], task_list, seed);
|
||||||
|
let tmp_seed = -1;
|
||||||
|
if (seed == -1) {
|
||||||
|
tmp_seed = res;
|
||||||
|
}
|
||||||
|
for (let j = 1; j < images.length; j++) {
|
||||||
|
const item = images[j];
|
||||||
|
|
||||||
|
let has_permission = false;
|
||||||
|
// 判断权限
|
||||||
|
let permission = this.global.permission;
|
||||||
|
|
||||||
|
if (permission && permission.length >= 0) {
|
||||||
|
if (permission.indexOf(DEFINE_STRING.PERMISSIONS.AUTO_SAVE_IMAGE_PERMISSION) >= 0) {
|
||||||
|
has_permission = true;
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
has_permission = true;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (auto_save_image.save_match_count && j >= auto_save_image.save_match_count && has_permission && auto_save_image.auto_match) {
|
||||||
|
// 现在随机匹配视频
|
||||||
|
// 获取指定的文件夹中的图片
|
||||||
|
let randomData = png_files[Math.floor(Math.random() * png_files.length)];
|
||||||
|
let base_name = path.basename(item);
|
||||||
|
let copy_path = path.join(this.global.config.project_path, 'tmp/' + task_list.out_folder, base_name);
|
||||||
|
await this.tools.copyFileOrDirectory(randomData, copy_path);
|
||||||
|
} else {
|
||||||
|
this.global.requestQuene.enqueue(async () => {
|
||||||
|
await this.sd.OneImageGeneration(item, task_list, tmp_seed);
|
||||||
|
}, `${task_list.out_folder}_${images[j]}`, batch, task_list.out_folder)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
}, `${task_list.out_folder}_${images[0]}`, batch, task_list.out_folder)
|
||||||
|
// }
|
||||||
|
task_list.status = 'queue';
|
||||||
|
|
||||||
|
// 开始批次执行（全部刷新）
|
||||||
|
this.global.newWindow[0].win.webContents.send(DEFINE_STRING.IMAGE_TASK_STATUS_REFRESH, task_list);
|
||||||
|
|
||||||
|
// 判断是不是还有批次执行的任务
|
||||||
|
// 监听子批次完成（修改当前批次的状态）
|
||||||
|
this.global.requestQuene.setSubBatchCompletionCallback(batch, task_list.out_folder, async (failedTasks) => {
|
||||||
|
console.log(failedTasks)
|
||||||
|
if (failedTasks.length > 0) {
|
||||||
|
|
||||||
|
} else {
|
||||||
|
// 判断是不是有错误。没有错误的话。直接修改状态。有错误直接记录错误（写入一个就行）
|
||||||
|
task_list_json = JSON.parse(await fspromises.readFile(taskPath, 'utf-8'));
|
||||||
|
// 全部执行完毕之后，修改
|
||||||
|
task_list_json.task_list.map(a => {
|
||||||
|
if (a.out_folder == task_list.out_folder) {
|
||||||
|
a.status = "ok";
|
||||||
|
}
|
||||||
|
})
|
||||||
|
// 写入
|
||||||
|
await fspromises.writeFile(taskPath, JSON.stringify(task_list_json));
|
||||||
|
|
||||||
|
this.global.newWindow[0].win.webContents.send(DEFINE_STRING.IMAGE_TASK_STATUS_REFRESH, {
|
||||||
|
out_folder: task_list.out_folder,
|
||||||
|
status: "ok"
|
||||||
|
});
|
||||||
|
|
||||||
|
this.global.newWindow[0].win.webContents.send(DEFINE_STRING.VIDEO_GENERATE_STATUS_REFRESH, {
|
||||||
|
out_folder: task_list.out_folder,
|
||||||
|
status: "ok"
|
||||||
|
})
|
||||||
|
}
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
// 监听总批次完成
|
||||||
|
this.global.requestQuene.setBatchCompletionCallback(batch, (failedTasks) => {
|
||||||
|
if (failedTasks.length > 0) {
|
||||||
|
let message = `
|
||||||
|
批次生成任务都已完成。
|
||||||
|
但是以下任务执行失败：
|
||||||
|
`
|
||||||
|
failedTasks.forEach(({ taskId, error }) => {
|
||||||
|
message += `${taskId}-, \n 错误信息: ${error}` + '\n';
|
||||||
|
});
|
||||||
|
|
||||||
|
this.global.newWindow[0].win.webContents.send(DEFINE_STRING.SHOW_MESSAGE_DIALOG, {
|
||||||
|
code: 0,
|
||||||
|
message: message
|
||||||
|
})
|
||||||
|
} else {
|
||||||
|
this.global.newWindow[0].win.webContents.send(DEFINE_STRING.SHOW_MESSAGE_DIALOG, {
|
||||||
|
code: 1,
|
||||||
|
message: "所有生成任务完成"
|
||||||
|
})
|
||||||
|
}
|
||||||
|
})
|
||||||
|
// 写回
|
||||||
|
await fspromises.writeFile(taskPath, JSON.stringify(task_list_json));
|
||||||
|
return {
|
||||||
|
code: 1,
|
||||||
|
}
|
||||||
|
} catch (error) {
|
||||||
|
return {
|
||||||
|
code: 0,
|
||||||
|
message: error.toString()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 批量高清文件夹
|
||||||
|
* @param {项目下面的文件夹的值} value
|
||||||
|
*/
|
||||||
|
async ImproveResolution(value) {
|
||||||
|
try {
|
||||||
|
let batch = DEFINE_STRING.QUEUE_BATCH.LOCAL_IMAGE_IMPROVE;
|
||||||
|
|
||||||
|
// 加入队列
|
||||||
|
for (let i = 0; i < value.length; i++) {
|
||||||
|
const folder = value[i];
|
||||||
|
this.global.requestQuene.enqueue(async () => {
|
||||||
|
await this.pm.ImproveFolder(folder);
|
||||||
|
}, folder, batch)
|
||||||
|
|
||||||
|
// 添加队列后修改状态
|
||||||
|
await this.pm.ModifyTaskStatus('out_folder', folder, "video_improving");
|
||||||
|
this.global.newWindow[0].win.webContents.send(DEFINE_STRING.VIDEO_GENERATE_STATUS_REFRESH, {
|
||||||
|
out_folder: folder,
|
||||||
|
status: "video_improving"
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
global.requestQuene.setBatchCompletionCallback(batch, async (failedTasks) => {
|
||||||
|
|
||||||
|
let permission = this.global.permission;
|
||||||
|
// 判断权限
|
||||||
|
if (permission && permission.length >= 0) {
|
||||||
|
if (permission.indexOf(DEFINE_STRING.PERMISSIONS.AUTO_SAVE_IMAGE_PERMISSION) >= 0) {
|
||||||
|
// 在高清前，先将要高清的图片保存
|
||||||
|
let res = await ImageSetting.SaveImageToOtherFolder(value, null);
|
||||||
|
if (res.code == 0) {
|
||||||
|
return res;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
// 在高清前，先将要高清的图片保存
|
||||||
|
let res = await ImageSetting.SaveImageToOtherFolder(value, null);
|
||||||
|
if (res.code == 0) {
|
||||||
|
return res;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
if (failedTasks.length > 0) {
|
||||||
|
let message = `
|
||||||
|
高清任务都已完成。
|
||||||
|
但是以下任务执行失败：
|
||||||
|
`
|
||||||
|
failedTasks.forEach(({ taskId, error }) => {
|
||||||
|
message += `${taskId}-, \n 错误信息: ${error}` + '\n';
|
||||||
|
});
|
||||||
|
|
||||||
|
this.global.newWindow[0].win.webContents.send(DEFINE_STRING.SHOW_MESSAGE_DIALOG, {
|
||||||
|
code: 0,
|
||||||
|
message: message
|
||||||
|
})
|
||||||
|
} else {
|
||||||
|
this.global.newWindow[0].win.webContents.send(DEFINE_STRING.SHOW_MESSAGE_DIALOG, {
|
||||||
|
code: 1,
|
||||||
|
message: "所有任务高清完成"
|
||||||
|
})
|
||||||
|
}
|
||||||
|
})
|
||||||
|
return {
|
||||||
|
code: 1
|
||||||
|
}
|
||||||
|
} catch (error) {
|
||||||
|
return {
|
||||||
|
code: 0,
|
||||||
|
message: error.toString()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 下载文件到指定位置
|
||||||
|
* @param {*} value 0: 源图片地址 1: 类型 2: 目标文件地址
|
||||||
|
*/
|
||||||
|
async DownloadImageFile(value) {
|
||||||
|
try {
|
||||||
|
let type = value[1];
|
||||||
|
let img_file = value[0];
|
||||||
|
let target_file = value[2];
|
||||||
|
if (type == "cover") {
|
||||||
|
// 下载封面
|
||||||
|
// 判断封面是不是存在
|
||||||
|
let cover_path = path.join(this.global.config.project_path, 'data/封面');
|
||||||
|
let isExist = await this.tools.checkExists(cover_path);
|
||||||
|
if (!isExist) {
|
||||||
|
await fspromises.mkdir(cover_path, { recursive: true });
|
||||||
|
}
|
||||||
|
// 复制
|
||||||
|
await fspromises.copyFile(img_file, path.join(cover_path, Date.now() + '.png'));
|
||||||
|
return {
|
||||||
|
code: 1,
|
||||||
|
message: "保存成功！"
|
||||||
|
}
|
||||||
|
} else if (type == "replace") {
|
||||||
|
if (!target_file) {
|
||||||
|
throw new Error("目标文件地址不存在");
|
||||||
|
}
|
||||||
|
await fspromises.copyFile(path.normalize(decodeURIComponent(img_file)), path.normalize(decodeURIComponent(target_file)));
|
||||||
|
await this.tools.copyFileOrDirectory(decodeURIComponent(img_file), decodeURIComponent(target_file).replace("output_crop_1", "input_crop"));
|
||||||
|
return {
|
||||||
|
code: 1,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
else {
|
||||||
|
return {
|
||||||
|
code: 0,
|
||||||
|
message: "参数错误"
|
||||||
|
}
|
||||||
|
}
|
||||||
|
} catch (error) {
|
||||||
|
return {
|
||||||
|
code: 0,
|
||||||
|
message: error.toString()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
556
src/main/backPrompt/videoGenerate.js
Normal file
556
src/main/backPrompt/videoGenerate.js
Normal file
@ -0,0 +1,556 @@
|
|||||||
|
import path from "path";
|
||||||
|
import { define } from "../../define/define";
|
||||||
|
import { Tools } from "../tools";
|
||||||
|
import { PublicMethod } from "../Public/publicMethod";
|
||||||
|
import { func } from '../func'
|
||||||
|
import { DEFINE_STRING } from "../../define/define_string";
|
||||||
|
const util = require('util');
|
||||||
|
const { spawn, exec } = require('child_process');
|
||||||
|
const execAsync = util.promisify(exec);
|
||||||
|
const fspromises = require("fs").promises;
|
||||||
|
import { SD } from "../Public/SD"
|
||||||
|
|
||||||
|
export class VideoGenerate {
|
||||||
|
constructor(global) {
|
||||||
|
this.global = global;
|
||||||
|
this.tools = new Tools();
|
||||||
|
this.pm = new PublicMethod(global);
|
||||||
|
this.sd = new SD(global);
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 获取之前的基础配置
|
||||||
|
*/
|
||||||
|
async GetVideoGenerateConfig() {
|
||||||
|
try {
|
||||||
|
let res = await this.tools.getJsonFilePropertyValue(path.join(this.global.config.project_path, "scripts/config.json"), "video_config", {}, false);
|
||||||
|
return {
|
||||||
|
code: 1,
|
||||||
|
data: res
|
||||||
|
}
|
||||||
|
} catch (error) {
|
||||||
|
return {
|
||||||
|
code: 0,
|
||||||
|
message: error.toString()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 保存合成视频的基础信息（Srt，audio，background）
|
||||||
|
* @param {*} value 保存合成视频的基础信息（Srt，audio，background）
|
||||||
|
*/
|
||||||
|
async SaveVideoSrtAndAudioMessage(value) {
|
||||||
|
try {
|
||||||
|
value = JSON.parse(value);
|
||||||
|
await this.tools.writeJsonFilePropertyValue(path.join(this.global.config.project_path, "scripts/config.json"), "video_config", value);
|
||||||
|
return {
|
||||||
|
code: 1
|
||||||
|
}
|
||||||
|
} catch (error) {
|
||||||
|
return {
|
||||||
|
code: 0,
|
||||||
|
message: error.toString()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
*全自动合成视频的任务（出图、高清、合成视频）
|
||||||
|
* @param {传入的需要执行的task_list的任务} value
|
||||||
|
*/
|
||||||
|
async ActionAutoVideoTask(value) {
|
||||||
|
try {
|
||||||
|
// 开始添加队列任务
|
||||||
|
// 生图
|
||||||
|
// 将当前的所有任务添加到队列中
|
||||||
|
await this.pm.AddWebuiJson();
|
||||||
|
let batch = DEFINE_STRING.QUEUE_BATCH.AUTO_VIDEO_GENERATE;
|
||||||
|
let taskPath = path.join(this.global.config.project_path, "scripts/task_list.json");
|
||||||
|
|
||||||
|
// // 修改数据（保存srt字幕文件位置。配音文件位置。背景音乐文件）
|
||||||
|
// this.global.fileQueue.enqueue(async () => {
|
||||||
|
// let config_json = JSON.parse(await fspromises.readFile(path.join(this.global.config.project_path, "scripts/config.json"), 'utf-8'));
|
||||||
|
// config_json.srt_path = value[1].srt_path;
|
||||||
|
// config_json.audio_path = value[1].audio_path;
|
||||||
|
// config_json.background_music = value[1].background_music;
|
||||||
|
// await fspromises.writeFile(path.join(this.global.config.project_path, "scripts/config.json"), JSON.stringify(config_json));
|
||||||
|
// })
|
||||||
|
|
||||||
|
await this.tools.writeJsonFilePropertyValue(path.join(this.global.config.project_path, "scripts/config.json"), "video_config", value[1]);
|
||||||
|
|
||||||
|
// 便利所有的队列任务
|
||||||
|
for (let i = 0; i < value[0].length; i++) {
|
||||||
|
// 将所有的数据天添加到队列（总的大队列，有很多的小队列）
|
||||||
|
// 将所有生图任务添加到队列中
|
||||||
|
const task_list = value[0][i];
|
||||||
|
let seed = -1;
|
||||||
|
let subBatchId = `${task_list.out_folder}_image`
|
||||||
|
let images = await this.tools.getFilesWithExtensions(path.join(this.global.config.project_path, 'tmp/input_crop'), '.png');
|
||||||
|
await fspromises.mkdir(path.join(this.global.config.project_path, 'tmp/' + task_list.out_folder), { recursive: true });
|
||||||
|
if (images.length <= 0) {
|
||||||
|
throw new Error("未检测到抽帧图片。请检查");
|
||||||
|
}
|
||||||
|
this.global.requestQuene.enqueue(async () => {
|
||||||
|
let res = await this.sd.OneImageGeneration(images[0], task_list, seed);
|
||||||
|
let tmp_seed = -1;
|
||||||
|
if (seed == -1) {
|
||||||
|
tmp_seed = res;
|
||||||
|
}
|
||||||
|
for (let j = 1; j < images.length; j++) {
|
||||||
|
const element = images[j];
|
||||||
|
this.global.requestQuene.enqueue(async () => {
|
||||||
|
await this.sd.OneImageGeneration(element, task_list, tmp_seed);
|
||||||
|
}, `${task_list.out_folder}_${images[j]}`, batch, subBatchId)
|
||||||
|
}
|
||||||
|
|
||||||
|
}, `${task_list.out_folder}_${images[0]}`, batch, subBatchId)
|
||||||
|
// }
|
||||||
|
task_list.status = 'queue';
|
||||||
|
task_list["isAuto"] = "true";
|
||||||
|
await this.pm.ModifyImageTaskList([task_list]);
|
||||||
|
// 修改状态
|
||||||
|
// await this.pm.ModifyTaskStatus("id", task_list.id, "queue");
|
||||||
|
// this.global.fileQueue.enqueue(async () => {
|
||||||
|
// let task_json = JSON.parse(await fspromises.readFile(taskPath, 'utf-8'));
|
||||||
|
// let index = task_json.task_list.findIndex(item => item.id == task_list.id);
|
||||||
|
// task_json.task_list[index] = task_list;
|
||||||
|
// await fspromises.writeFile(taskPath, JSON.stringify(task_json));
|
||||||
|
// })
|
||||||
|
this.global.newWindow[0].win.webContents.send(DEFINE_STRING.VIDEO_GENERATE_STATUS_REFRESH, task_list)
|
||||||
|
|
||||||
|
// 判断是不是还有批次执行的任务
|
||||||
|
// 监听子批次完成（修改当前批次的状态）
|
||||||
|
this.global.requestQuene.setSubBatchCompletionCallback(batch, subBatchId, async (failedTasks) => {
|
||||||
|
console.log(failedTasks)
|
||||||
|
if (failedTasks.length > 0) {
|
||||||
|
// 之前的任务出现错误
|
||||||
|
// 执行错误
|
||||||
|
} else {
|
||||||
|
// 判断是不是有错误。没有错误的话。直接修改状态。有错误直接记录错误（写入一个就行）
|
||||||
|
task_list.status = "ok";
|
||||||
|
await this.pm.ModifyTaskStatus("id", task_list.id, 'ok');
|
||||||
|
this.global.newWindow[0].win.webContents.send(DEFINE_STRING.VIDEO_GENERATE_STATUS_REFRESH, task_list)
|
||||||
|
console.log(subBatchId + "生图执行完毕。可以开始执行高清")
|
||||||
|
|
||||||
|
// 添加高清队列
|
||||||
|
// task_list
|
||||||
|
this.global.requestQuene.enqueue(async () => {
|
||||||
|
await this.pm.ImproveFolder(task_list.out_folder);
|
||||||
|
}, `${task_list.out_folder}_improve`, batch, `${task_list.out_folder}_improve`);
|
||||||
|
// 添加队列后修改状态
|
||||||
|
task_list.status = "video_improving"
|
||||||
|
await this.pm.ModifyTaskStatus('out_folder', task_list.out_folder, "video_improving");
|
||||||
|
this.global.newWindow[0].win.webContents.send(DEFINE_STRING.VIDEO_GENERATE_STATUS_REFRESH, task_list)
|
||||||
|
|
||||||
|
// 监听高清任务完成
|
||||||
|
this.global.requestQuene.setSubBatchCompletionCallback(batch, `${task_list.out_folder}_improve`, async (failedTasks) => {
|
||||||
|
console.log(failedTasks)
|
||||||
|
if (failedTasks.length > 0) {
|
||||||
|
// 之前的任务出现错误
|
||||||
|
// 执行错误
|
||||||
|
} else {
|
||||||
|
|
||||||
|
console.log(task_list.out_folder + "高清完成，可以开始合成视频");
|
||||||
|
// 添加生成视频队列
|
||||||
|
this.global.requestQuene.enqueue(async () => {
|
||||||
|
let video_config = JSON.parse(await fspromises.readFile(define.video_config, 'utf-8'));
|
||||||
|
await this.AutoGeneretionOneVide(task_list.out_folder, video_config, value);
|
||||||
|
}, `${task_list.out_folder}_video`, batch, `${task_list.out_folder}_video`);
|
||||||
|
task_list.status = "video_queue";
|
||||||
|
// 添加后修改状态
|
||||||
|
await this.pm.ModifyTaskStatus('out_folder', task_list.out_folder, "video_queue");
|
||||||
|
|
||||||
|
this.global.newWindow[0].win.webContents.send(DEFINE_STRING.VIDEO_GENERATE_STATUS_REFRESH, task_list)
|
||||||
|
|
||||||
|
// 监听生成视频任务完成
|
||||||
|
this.global.requestQuene.setSubBatchCompletionCallback(batch, `${task_list.out_folder}_video`, async (failedTasks) => {
|
||||||
|
console.log(failedTasks)
|
||||||
|
if (failedTasks.length > 0) {
|
||||||
|
// 之前的任务出现错误
|
||||||
|
// 执行错误
|
||||||
|
} else {
|
||||||
|
console.log(task_list.out_folder + "合成视频完成");
|
||||||
|
// 添加生成视频队列
|
||||||
|
task_list.status = "video_ok";
|
||||||
|
// 添加后修改状态
|
||||||
|
await this.pm.ModifyTaskStatus('out_folder', task_list.out_folder, "video_ok");
|
||||||
|
this.global.newWindow[0].win.webContents.send(DEFINE_STRING.VIDEO_GENERATE_STATUS_REFRESH, task_list)
|
||||||
|
}
|
||||||
|
})
|
||||||
|
|
||||||
|
}
|
||||||
|
})
|
||||||
|
}
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
// 监听总批次完成
|
||||||
|
this.global.requestQuene.setBatchCompletionCallback(batch, async (failedTasks) => {
|
||||||
|
if (failedTasks.length > 0) {
|
||||||
|
let message = `
|
||||||
|
批次生成任务都已完成。
|
||||||
|
但是以下任务执行失败：
|
||||||
|
`
|
||||||
|
failedTasks.forEach(({ taskId, error }) => {
|
||||||
|
message += `${taskId}-, \n 错误信息: ${error}` + '\n';
|
||||||
|
});
|
||||||
|
|
||||||
|
this.global.newWindow[0].win.webContents.send(DEFINE_STRING.SHOW_MESSAGE_DIALOG, {
|
||||||
|
code: 0,
|
||||||
|
message: message
|
||||||
|
})
|
||||||
|
} else {
|
||||||
|
console.log("所有的自动生成任务完成");
|
||||||
|
this.global.newWindow[0].win.webContents.send(DEFINE_STRING.SHOW_MESSAGE_DIALOG, {
|
||||||
|
code: 1,
|
||||||
|
message: "所有自动任务完成"
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
this.global.fileQueue.enqueue(async () => {
|
||||||
|
// 读取最新的数据删除
|
||||||
|
// 将自动化标识删除
|
||||||
|
let task_json = JSON.parse(await fspromises.readFile(taskPath, 'utf-8'));
|
||||||
|
for (let i = 0; i < task_json.task_list.length; i++) {
|
||||||
|
let task_list = task_json.task_list[i];
|
||||||
|
task_list["isAuto"] = "false";
|
||||||
|
await this.pm.ModifyImageTaskList([task_list]);
|
||||||
|
}
|
||||||
|
})
|
||||||
|
})
|
||||||
|
return {
|
||||||
|
code: 1,
|
||||||
|
message: "自动任务添加到队列中"
|
||||||
|
}
|
||||||
|
|
||||||
|
} catch (error) {
|
||||||
|
console.error(error)
|
||||||
|
return {
|
||||||
|
code: 0,
|
||||||
|
message: error.message
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 获取文件夹下面的子文件夹，添加判断条件
|
||||||
|
* @param {value} value
|
||||||
|
* @returns
|
||||||
|
*/
|
||||||
|
async getSubFolderList(value) {
|
||||||
|
try {
|
||||||
|
let folder = await this.pm.getSubFolderList(path.join(this.global.config.project_path, "tmp"), value[0], value[1]);
|
||||||
|
// 找到所有的数据（查询状态返回）
|
||||||
|
let task_path = path.join(this.global.config.project_path, "scripts/task_list.json");
|
||||||
|
let isExist = await this.tools.checkExists(task_path);
|
||||||
|
let data = [];
|
||||||
|
if (!isExist) {
|
||||||
|
for (let i = 0; i < folder.length; i++) {
|
||||||
|
const element = folder[i];
|
||||||
|
let obj = {
|
||||||
|
folder: element,
|
||||||
|
status: "unkown error"
|
||||||
|
}
|
||||||
|
data.push(obj);
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
let task_list = JSON.parse(await fspromises.readFile(task_path, 'utf-8'))["task_list"];
|
||||||
|
// 查询状态
|
||||||
|
for (let i = 0; i < folder.length; i++) {
|
||||||
|
const element = folder[i];
|
||||||
|
let index = task_list.findIndex(item => item.out_folder == element);
|
||||||
|
if (index < 0) {
|
||||||
|
let obj = {
|
||||||
|
folder: element,
|
||||||
|
status: "unkown error"
|
||||||
|
}
|
||||||
|
data.push(obj);
|
||||||
|
} else {
|
||||||
|
let status = task_list[index].status;
|
||||||
|
let obj = {
|
||||||
|
folder: element,
|
||||||
|
status: status
|
||||||
|
}
|
||||||
|
data.push(obj);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return {
|
||||||
|
code: 1,
|
||||||
|
data: data
|
||||||
|
}
|
||||||
|
} catch (error) {
|
||||||
|
return {
|
||||||
|
code: 0,
|
||||||
|
message: `Error Message ${error}`
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 自动生成一个视频
|
||||||
|
* @param {视频生成图片的文件夹} element
|
||||||
|
* @param {视频配置} video_config
|
||||||
|
* @param {参数。第一个值为数组，为生成的图片文件。第二个参数是配置} value
|
||||||
|
*/
|
||||||
|
async AutoGeneretionOneVide(element, video_config, value) {
|
||||||
|
try {
|
||||||
|
let background_music = "";
|
||||||
|
// 读取背景音乐的路径
|
||||||
|
if (value[1].background_music != "") {
|
||||||
|
let background_music_config = (await func.getClipSetting("background_music_setting")).value;
|
||||||
|
let background_music_filter = background_music_config.filter(item => item.id == value[1].background_music);
|
||||||
|
if (background_music_filter.length <= 0) {
|
||||||
|
throw new Error("背景音乐对应的配置没有找到");
|
||||||
|
} else {
|
||||||
|
background_music = background_music_filter[0].folder_path;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// 随机获取字幕设置
|
||||||
|
let ass_config = video_config.assConfig;
|
||||||
|
let ass_random_index = Math.floor(Math.random() * ass_config.length);
|
||||||
|
let watermark_config = video_config.watermarkConfig;
|
||||||
|
let watermark_random_index = Math.floor(Math.random() * watermark_config.length);
|
||||||
|
|
||||||
|
// 对每个视频生成配置文件并将其添加到配置文件中
|
||||||
|
let obj = {
|
||||||
|
srt_path: value[1].srt_path,
|
||||||
|
// 字幕样式（需要随机）
|
||||||
|
srt_style: ass_config[ass_random_index],
|
||||||
|
audio_path: value[1].audio_path,
|
||||||
|
background_music_folder: background_music,
|
||||||
|
// 水印设置（需要随机）
|
||||||
|
friendly_reminder: watermark_config[watermark_random_index],
|
||||||
|
video_resolution_x: video_config.video_resolution_x,
|
||||||
|
video_resolution_y: video_config.video_resolution_y,
|
||||||
|
outpue_file: path.join(this.global.config.project_path, this.global.config.project_name + element + ".mp4"),
|
||||||
|
image_folder: path.join(this.global.config.project_path, 'tmp/' + element),
|
||||||
|
srt_config: path.join(this.global.config.project_path, 'scripts/config.json'),
|
||||||
|
mp4_file_txt: path.join(this.global.config.project_path, `scripts/${element}.txt`),
|
||||||
|
status: "no",
|
||||||
|
audio_sound_size: video_config.audioSoundSize,
|
||||||
|
background_music_sound_size: video_config.backgroundMusicSoundSize,
|
||||||
|
keyFrame: video_config.keyframe,
|
||||||
|
frameRate: video_config.frameRate,
|
||||||
|
bitRate: video_config.bitRate
|
||||||
|
}
|
||||||
|
// 将配置文件写入
|
||||||
|
let project_config_path = path.join(this.global.config.project_path, `scripts/${element}.json`);
|
||||||
|
await fspromises.writeFile(project_config_path, JSON.stringify(obj));
|
||||||
|
// let task_list = JSON.parse(await fspromises.readFile(path.join(this.global.config.project_path,'scripts/task_')));
|
||||||
|
let scriptPath = path.join(define.scripts_path, 'Lai.exe');
|
||||||
|
// 执行生成图片的脚本
|
||||||
|
let script = `cd "${define.scripts_path}" && "${scriptPath}" -c "${project_config_path.replaceAll('\\', '/')}"`;
|
||||||
|
const output = await execAsync(script, { maxBuffer: 1024 * 1024 * 10, encoding: 'utf-8' });
|
||||||
|
if (output.stderr != '') {
|
||||||
|
obj.status = "video_error";
|
||||||
|
obj.stdout = output.stdout;
|
||||||
|
obj.stderr = output.stderr;
|
||||||
|
await this.pm.ModifyTaskStatus('out_folder', element, "video_error");
|
||||||
|
throw new Error(output.stderr);
|
||||||
|
} else {
|
||||||
|
obj.status = "video_ok";
|
||||||
|
obj.stdout = output.stdout;
|
||||||
|
obj.stderr = output.stderr;
|
||||||
|
await this.pm.ModifyTaskStatus('out_folder', element, "video_ok");
|
||||||
|
// 将写出的视频中的exif数据删除
|
||||||
|
// await this.tools.deletePngAndDeleteExifData(obj.outpue_file, path.join(this.global.config.project_path, this.global.config.project_name + "_" + element.split('_')[element.split('_').length - 1] + ".mp4"));
|
||||||
|
}
|
||||||
|
|
||||||
|
this.global.newWindow[0].win.webContents.send(DEFINE_STRING.VIDEO_GENERATE_STATUS_REFRESH, {
|
||||||
|
out_folder: element,
|
||||||
|
status: obj.status
|
||||||
|
})
|
||||||
|
} catch (error) {
|
||||||
|
//手动修改
|
||||||
|
let task_list_json = JSON.parse(await fspromises.readFile(path.join(this.global.config.project_path, "scripts/task_list.json"), 'utf-8'));
|
||||||
|
let index = task_list_json.task_list.findIndex(item => item.out_folder == element);
|
||||||
|
if (index < 0) {
|
||||||
|
throw new Error("未找到对应的任务");
|
||||||
|
}
|
||||||
|
task_list_json.task_list[index].status = "video_error";
|
||||||
|
task_list_json.task_list[index].errorMessage = error.toString();
|
||||||
|
// 写回
|
||||||
|
await fspromises.writeFile(path.join(this.global.config.project_path, "scripts/task_list.json"), JSON.stringify(task_list_json));
|
||||||
|
this.global.newWindow[0].win.webContents.send(DEFINE_STRING.VIDEO_GENERATE_STATUS_REFRESH, {
|
||||||
|
out_folder: element,
|
||||||
|
status: "video_error"
|
||||||
|
})
|
||||||
|
throw error;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 添加自动合成任务到队列中去
|
||||||
|
* @param {} value
|
||||||
|
*/
|
||||||
|
async AutoGeneretionVideo(value) {
|
||||||
|
try {
|
||||||
|
// 先检查所有的条件
|
||||||
|
let batch = DEFINE_STRING.QUEUE_BATCH.AUTO_VIDEO_GENERATE_SINGLE;
|
||||||
|
// console.log(value);
|
||||||
|
let video_config = JSON.parse(await fspromises.readFile(define.video_config, 'utf-8'));
|
||||||
|
let res = await this.CheckVideoGenerattionAllCondition(video_config, value[1], value[0]);
|
||||||
|
if (res.code == 0) {
|
||||||
|
this.global.newWindow[0].win.webContents.send(DEFINE_STRING.SHOW_MESSAGE_DIALOG, res);
|
||||||
|
return res;
|
||||||
|
}
|
||||||
|
for (let i = 0; i < value[0].length; i++) {
|
||||||
|
const element = value[0][i];
|
||||||
|
this.global.requestQuene.enqueue(async () => {
|
||||||
|
await this.AutoGeneretionOneVide(element, video_config, value)
|
||||||
|
}, element, batch);
|
||||||
|
|
||||||
|
// 添加队列后修改状态
|
||||||
|
await this.pm.ModifyTaskStatus('out_folder', element, "video_queue");
|
||||||
|
this.global.newWindow[0].win.webContents.send(DEFINE_STRING.VIDEO_GENERATE_STATUS_REFRESH, {
|
||||||
|
out_folder: element,
|
||||||
|
status: "video_queue"
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
this.global.requestQuene.setBatchCompletionCallback(batch, (failedTasks) => {
|
||||||
|
if (failedTasks.length > 0) {
|
||||||
|
let message = `
|
||||||
|
生成视频任务都已完成。
|
||||||
|
但是以下任务执行失败：
|
||||||
|
`
|
||||||
|
failedTasks.forEach(({ taskId, error }) => {
|
||||||
|
message += `${taskId}-, \n 错误信息: ${error}` + '\n';
|
||||||
|
});
|
||||||
|
|
||||||
|
this.global.newWindow[0].win.webContents.send(DEFINE_STRING.SHOW_MESSAGE_DIALOG, {
|
||||||
|
code: 0,
|
||||||
|
message: message
|
||||||
|
})
|
||||||
|
} else {
|
||||||
|
this.global.newWindow[0].win.webContents.send(DEFINE_STRING.SHOW_MESSAGE_DIALOG, {
|
||||||
|
code: 1,
|
||||||
|
message: "所有生成视频任务已完成"
|
||||||
|
})
|
||||||
|
}
|
||||||
|
})
|
||||||
|
return {
|
||||||
|
code: 1
|
||||||
|
}
|
||||||
|
} catch (error) {
|
||||||
|
return {
|
||||||
|
code: 0,
|
||||||
|
message: error.toString()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 检测生成视频条件的必要条件
|
||||||
|
* @param {生成视频配置文件} video_config
|
||||||
|
* @param {生成视频的基础配置} simpleData
|
||||||
|
* @param {输出文件夹位置，为null，检查input文件里面的图片文件是不是和配置文件中的图片数量一致} out_folder
|
||||||
|
* @returns
|
||||||
|
*/
|
||||||
|
async CheckVideoGenerattionAllCondition(video_config, simpleData, out_folder) {
|
||||||
|
let message = "";
|
||||||
|
if (video_config == null) {
|
||||||
|
video_config = JSON.parse(await fspromises.readFile(define.video_config, 'utf-8'));
|
||||||
|
}
|
||||||
|
// 基础信息检测
|
||||||
|
if (video_config.audioSoundSize == null) {
|
||||||
|
message += "配音大小不能为空" + '\n';
|
||||||
|
}
|
||||||
|
if (video_config.backgroundMusicSoundSize == null) {
|
||||||
|
message += "背景音乐音量大学不能为空" + '\n';
|
||||||
|
}
|
||||||
|
if (video_config.video_resolution_x == null) {
|
||||||
|
message += "生成视频的宽度不能为空" + '\n';
|
||||||
|
}
|
||||||
|
if (video_config.video_resolution_y == null) {
|
||||||
|
message += "生成视频的高度不能为空" + '\r\n';
|
||||||
|
}
|
||||||
|
if (video_config.offsetValue == null) {
|
||||||
|
message += "视频的上下偏移量不能为空" + '\n';
|
||||||
|
}
|
||||||
|
if (video_config.frameRate == null) {
|
||||||
|
message += "生成视频的帧率不能为空" + '\n';
|
||||||
|
}
|
||||||
|
if (video_config.bitRate == null) {
|
||||||
|
message += "生成视频码率的不能为空" + '\n';
|
||||||
|
}
|
||||||
|
|
||||||
|
// 判断字幕列表中是不是有数据
|
||||||
|
if (video_config.assConfig == null || video_config.assConfig.length <= 0) {
|
||||||
|
message += "字幕设置最少包含一条" + '\n';
|
||||||
|
}
|
||||||
|
if (video_config.watermarkConfig == null || video_config.watermarkConfig.length <= 0) {
|
||||||
|
message += "水印设置最少包含一条" + '\n';
|
||||||
|
}
|
||||||
|
// 判断背景音乐文件夹中是不是存在。并且检查其中是不是有音乐文件
|
||||||
|
if (simpleData.background_music == "") {
|
||||||
|
message += "背景音乐文件夹路径不能为空" + '\n';
|
||||||
|
}
|
||||||
|
// 判断背景音乐文件夹是不是存在
|
||||||
|
if (await this.tools.checkExists(simpleData.background_music)) {
|
||||||
|
message += "背景音乐文件夹不存在" + '\n';
|
||||||
|
}
|
||||||
|
// 判断里面是不是有MP3或者是wav
|
||||||
|
let clip_json = JSON.parse(await fspromises.readFile(define.clip_setting, 'utf-8'))["background_music_setting"];
|
||||||
|
let background_music_obj = clip_json.filter(item => item.id == simpleData.background_music)[0];
|
||||||
|
let mp3_file = await this.tools.getFilesWithExtensions(background_music_obj.folder_path, '.mp3');
|
||||||
|
let wav_file = await this.tools.getFilesWithExtensions(background_music_obj.folder_path, '.wav');
|
||||||
|
if (mp3_file.length <= 0 && wav_file.length <= 0) {
|
||||||
|
message += "背景文件夹中没有 MP3 或 WAV 文件" + '\n';
|
||||||
|
}
|
||||||
|
|
||||||
|
let config_path = path.join(this.global.config.project_path, 'scripts/config.json');
|
||||||
|
let isE = await this.tools.checkExists(config_path);
|
||||||
|
if (!isE) {
|
||||||
|
message += "配置文件不存在。请先导入字幕文件。并调整时间轴。"
|
||||||
|
} else {
|
||||||
|
// 判断文案时间信息和图片信息是不是相同
|
||||||
|
let config_json = JSON.parse(await fspromises.readFile(config_path, 'utf-8'));
|
||||||
|
let len = config_json["srt_time_information"].length;
|
||||||
|
if (out_folder == null) {
|
||||||
|
// 判断输入文件中的配置文件中的数量是不是对上
|
||||||
|
let img_l = await this.tools.getFilesWithExtensions(path.join(this.global.config.project_path, "tmp/input_crop"), '.png');
|
||||||
|
if (img_l.length != len) {
|
||||||
|
message += `input_crop 文件里面图片和文案信息对不上。检查是不是图片数量不对`
|
||||||
|
}
|
||||||
|
// 判断是不是有tag文件
|
||||||
|
let tag_txt = await this.tools.getFilesWithExtensions(path.join(this.global.config.project_path, "tmp/input_crop"), '.txt');
|
||||||
|
let c_j = await this.tools.getFilesWithExtensions(path.join(this.global.config.project_path, "tmp/input_crop"), '.json');
|
||||||
|
if (tag_txt.length > 0) {
|
||||||
|
if (tag_txt.length != img_l.length) {
|
||||||
|
message += "反推的tag文件和图片的数量对不上。" + "\n";
|
||||||
|
}
|
||||||
|
} else if (c_j.length > 0) {
|
||||||
|
if (c_j.length != img_l.length) {
|
||||||
|
message += "已存在的配置文件和图片的数量对不上。" + "\n";
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
message += "好像没有反推呢" + "\n";
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
// 判断当前的输入的 out_folder 中图片数量是不是可以和配置文件中能否对上
|
||||||
|
for (let i = 0; i < out_folder.length; i++) {
|
||||||
|
const element = out_folder[i];
|
||||||
|
let image_l = await this.tools.getFilesWithExtensions(path.join(this.global.config.project_path, "tmp/" + element), '.png');
|
||||||
|
if (image_l.length != len) {
|
||||||
|
message += `${element} 文件里面图片和文案信息对不上。检查是不是图片数量不对`
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if (message != '') {
|
||||||
|
let res = {
|
||||||
|
code: 0,
|
||||||
|
message: message
|
||||||
|
}
|
||||||
|
this.global.newWindow[0].win.webContents.send(DEFINE_STRING.SHOW_MESSAGE_DIALOG, res)
|
||||||
|
return res;
|
||||||
|
}
|
||||||
|
return {
|
||||||
|
code: 1,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
301
src/main/backPrompt/writing.js
Normal file
301
src/main/backPrompt/writing.js
Normal file
@ -0,0 +1,301 @@
|
|||||||
|
let path = require("path");
|
||||||
|
let fspromises = require("fs").promises;
|
||||||
|
import { Tools } from "../tools";
|
||||||
|
import { DEFINE_STRING } from "../../define/define_string";
|
||||||
|
import { PublicMethod } from "../Public/publicMethod";
|
||||||
|
import { define } from "../../define/define";
|
||||||
|
import { get, has } from "lodash";
|
||||||
|
import { ClipSetting } from "../../define/setting/clipSetting";
|
||||||
|
const { v4: uuidv4 } = require('uuid'); // 引入UUID库来生成唯一标识符
|
||||||
|
let tools = new Tools();
|
||||||
|
|
||||||
|
export class Writing {
|
||||||
|
constructor(global) {
|
||||||
|
this.global = global
|
||||||
|
this.pm = new PublicMethod(global);
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 将文案信息写入到本地的文案文件中
|
||||||
|
* @param {*} value
|
||||||
|
*/
|
||||||
|
async SaveWordTxt(value) {
|
||||||
|
try {
|
||||||
|
let word_path = path.join(global.config.project_path, "文案.txt");
|
||||||
|
await tools.writeArrayToFile(value, word_path);
|
||||||
|
return {
|
||||||
|
code: 1,
|
||||||
|
message: "保存成功"
|
||||||
|
}
|
||||||
|
} catch (error) {
|
||||||
|
throw new Error(error);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 将分镜的时间信息添加道配置文件中
|
||||||
|
* @param {*} value 是一个数组，0 ：写入的数据 1：写入的属性 2：是否需要解析
|
||||||
|
*/
|
||||||
|
async SaveCopywritingInformation(value) {
|
||||||
|
try {
|
||||||
|
return await this.pm.SaveConfigJsonProperty(value);
|
||||||
|
} catch (error) {
|
||||||
|
return {
|
||||||
|
code: 0,
|
||||||
|
message: error.toString()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 获取Config.json文件中指定的属性
|
||||||
|
* @param {Array} value 传入的值 0 ： 需要获取的属性 1： 返回的默认值
|
||||||
|
* @returns
|
||||||
|
*/
|
||||||
|
async GetConfigJson(value) {
|
||||||
|
try {
|
||||||
|
return await this.pm.GetConfigJson(value, false);
|
||||||
|
} catch (error) {
|
||||||
|
return {
|
||||||
|
code: 0,
|
||||||
|
message: error.toString()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 获取当前项目下面的文案
|
||||||
|
*/
|
||||||
|
async GetProjectWord() {
|
||||||
|
try {
|
||||||
|
// 先判断当前的项目文件下面是不是又配置文件。没有才读取文案
|
||||||
|
let srt_config_path = path.join(global.config.project_path, "scripts/config.json");
|
||||||
|
let isExist = await tools.checkExists(srt_config_path);
|
||||||
|
let data = null;
|
||||||
|
let isImformation = false;
|
||||||
|
if (isExist) {
|
||||||
|
let config_1 = JSON.parse(await fspromises.readFile(srt_config_path));
|
||||||
|
isImformation = has(config_1, 'srt_time_information');
|
||||||
|
if (isImformation) {
|
||||||
|
data = JSON.parse(await fspromises.readFile(srt_config_path)).srt_time_information;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if (!isExist || !isImformation) {
|
||||||
|
let word_path = path.join(global.config.project_path, "文案.txt");
|
||||||
|
let isExistWord = await tools.checkExists(word_path);
|
||||||
|
if (!isExistWord) {
|
||||||
|
return {
|
||||||
|
code: 0,
|
||||||
|
message: "没有文案文件"
|
||||||
|
}
|
||||||
|
}
|
||||||
|
let data = await fspromises.readFile(word_path, { encoding: 'utf-8' });
|
||||||
|
let lines = data.split(/\r?\n/);
|
||||||
|
// 打印或返回这个数组
|
||||||
|
// console.log(lines);
|
||||||
|
// 判断是不是有洗稿后的文件
|
||||||
|
let new_srt_path = path.join(global.config.project_path, "new_word.txt");
|
||||||
|
let isExistAfterGPTWord = await tools.checkExists(new_srt_path);
|
||||||
|
let after_data = null;
|
||||||
|
if (isExistAfterGPTWord) {
|
||||||
|
after_data = (await fspromises.readFile(new_srt_path, { encoding: 'utf-8' })).split(/\r?\n/);
|
||||||
|
}
|
||||||
|
// 判断抽帧文件是不是存在
|
||||||
|
// 返回图片信息
|
||||||
|
let old_image_path_list = await tools.getFilesWithExtensions(path.join(global.config.project_path, "tmp/input_crop"), '.png');
|
||||||
|
let res = [];
|
||||||
|
let lastId = '';
|
||||||
|
// 处理数据
|
||||||
|
for (let i = 0; i < lines.length; i++) {
|
||||||
|
const line = lines[i];
|
||||||
|
let id = uuidv4();
|
||||||
|
let after_gpt = null;
|
||||||
|
if (after_data != null) {
|
||||||
|
after_gpt = after_data[i];
|
||||||
|
}
|
||||||
|
|
||||||
|
let img_path = null;
|
||||||
|
if (old_image_path_list != null) {
|
||||||
|
img_path = old_image_path_list[i];
|
||||||
|
}
|
||||||
|
let obj = {
|
||||||
|
no: i + 1,
|
||||||
|
id: id,
|
||||||
|
lastId: lastId,
|
||||||
|
word: line,
|
||||||
|
old_image: img_path,
|
||||||
|
after_gpt: after_gpt,
|
||||||
|
start_time: null,
|
||||||
|
end_time: null,
|
||||||
|
timeLimit: null,
|
||||||
|
subValue: []
|
||||||
|
}
|
||||||
|
res.push(obj);
|
||||||
|
lastId = id;
|
||||||
|
}
|
||||||
|
return {
|
||||||
|
code: 1,
|
||||||
|
type: 0,
|
||||||
|
data: res
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
let data = JSON.parse(await fspromises.readFile(srt_config_path)).srt_time_information;
|
||||||
|
return {
|
||||||
|
code: 1,
|
||||||
|
type: 1,
|
||||||
|
data: data
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
} catch (error) {
|
||||||
|
throw new Error(error);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 搭导入srt。然后加载时间轴。完全匹配失败的将会还是会导入然后手动手动切换
|
||||||
|
* @param {文案洗稿界面信息} textData
|
||||||
|
*/
|
||||||
|
async ImportSrtAndGetTime(data) {
|
||||||
|
let textData = data[0];
|
||||||
|
let init_num = textData.length;
|
||||||
|
let srt_path = data[1];
|
||||||
|
let current_text = "";
|
||||||
|
try {
|
||||||
|
if (!srt_path) {
|
||||||
|
// 获取项目下面的所有的srt
|
||||||
|
let srtfiles = await tools.getFilesWithExtensions(global.config.project_path, '.srt');
|
||||||
|
if (srtfiles.length <= 0) {
|
||||||
|
throw new Error("没有SRT文件");
|
||||||
|
}
|
||||||
|
srt_path = srtfiles[0];
|
||||||
|
}
|
||||||
|
let srt_data = (await fspromises.readFile(srt_path, 'utf-8')).toString("utf-8");
|
||||||
|
const entries = srt_data.replace(/\r\n/g, '\n').split('\n\n');
|
||||||
|
let data = entries.map(entry => {
|
||||||
|
const lines = entry.split('\n');
|
||||||
|
if (lines.length >= 3) {
|
||||||
|
const times = lines[1];
|
||||||
|
const text = lines.slice(2).join(' ');
|
||||||
|
const [start, end] = times.split(' --> ').map(time => {
|
||||||
|
const [hours, minutes, seconds] = time.split(':');
|
||||||
|
const [sec, millis] = seconds.split(',');
|
||||||
|
return ((parseInt(hours) * 3600 + parseInt(minutes) * 60 + parseInt(sec)) * 1000 + parseInt(millis));
|
||||||
|
});
|
||||||
|
return { start, end, text, id: uuidv4() };
|
||||||
|
}
|
||||||
|
}).filter(entry => entry);
|
||||||
|
// 开始匹配（洗稿后的）
|
||||||
|
let srt_list = [];
|
||||||
|
let srt_obj = null
|
||||||
|
let text_count = 0;
|
||||||
|
let tmp_str = "";
|
||||||
|
for (let i = 0; i < data.length;) {
|
||||||
|
let srt_value = data[i].text;
|
||||||
|
current_text = `字幕： “${srt_value}” 和文案第${text_count + 1} 行数据 “${textData[text_count].after_gpt}” 数据不匹配（检查一下上下文）`;
|
||||||
|
let start_time = data[i].start;
|
||||||
|
let end_time = data[i].end;
|
||||||
|
let obj = {
|
||||||
|
start_time,
|
||||||
|
end_time,
|
||||||
|
srt_value,
|
||||||
|
id: data[i].id
|
||||||
|
};
|
||||||
|
// 判断当前字幕是不是在当前句
|
||||||
|
// 不能用简单的包含，而是将数据进行去除特殊符号拼接后判断是不是相同
|
||||||
|
tmp_str += srt_value;
|
||||||
|
if (tools.removePunctuationIncludingEllipsis(textData[text_count].after_gpt).startsWith(tools.removePunctuationIncludingEllipsis(tmp_str))) {
|
||||||
|
if (srt_obj == null) {
|
||||||
|
srt_obj = {}
|
||||||
|
srt_obj.id = uuidv4();
|
||||||
|
srt_obj.start_time = start_time;
|
||||||
|
srt_obj.value = srt_value;
|
||||||
|
srt_obj.subValue = [obj];
|
||||||
|
}
|
||||||
|
else {
|
||||||
|
srt_obj.value = srt_obj.value + srt_value;
|
||||||
|
srt_obj.subValue = [...srt_obj.subValue, obj];
|
||||||
|
}
|
||||||
|
textData[text_count].start_time = srt_obj.start_time;
|
||||||
|
textData[text_count].subValue = srt_obj.subValue
|
||||||
|
srt_list.push(obj);
|
||||||
|
i++;
|
||||||
|
} else {
|
||||||
|
// 判断下一句文件是不是以当当前巨开头。是的话继续。不是的话。直接返回后面的所有信息
|
||||||
|
if (tools.removePunctuationIncludingEllipsis(textData[text_count + 1].after_gpt).startsWith(tools.removePunctuationIncludingEllipsis(srt_value))) {
|
||||||
|
textData[text_count].end_time = srt_list[srt_list.length - 1].end_time;
|
||||||
|
text_count++;
|
||||||
|
srt_obj = null;
|
||||||
|
tmp_str = ""
|
||||||
|
} else {
|
||||||
|
// 将下面的数据直接 添加到textData后面。
|
||||||
|
// 修改当前行数据的结束事件为
|
||||||
|
if (srt_list.length > 0) {
|
||||||
|
textData[text_count].end_time = srt_list[srt_list.length - 1].end_time;
|
||||||
|
text_count++;
|
||||||
|
}
|
||||||
|
// 将后面的数据直接添加
|
||||||
|
let lastId = textData[textData.length - 1].id;
|
||||||
|
for (let j = i; j < data.length; j++) {
|
||||||
|
// 直接修改原有数据
|
||||||
|
if (text_count < init_num) {
|
||||||
|
textData[text_count].subValue = [{
|
||||||
|
start_time: data[j].start,
|
||||||
|
end_time: data[j].end,
|
||||||
|
id: data[j].id,
|
||||||
|
srt_value: data[j].text
|
||||||
|
}]
|
||||||
|
textData[text_count].start_time = data[j].start;
|
||||||
|
textData[text_count].end_time = data[j].end;
|
||||||
|
text_count++;
|
||||||
|
}
|
||||||
|
else {
|
||||||
|
let id = uuidv4();
|
||||||
|
// 添加
|
||||||
|
let obj = {
|
||||||
|
no: j + 1,
|
||||||
|
id: id,
|
||||||
|
word: null,
|
||||||
|
lastId: lastId,
|
||||||
|
old_image: path.normalize(define.zhanwei_image),
|
||||||
|
after_gpt: null,
|
||||||
|
start_time: data[j].start,
|
||||||
|
end_time: data[j].end,
|
||||||
|
subValue: [{
|
||||||
|
start_time: data[j].start,
|
||||||
|
end_time: data[j].end,
|
||||||
|
id: data[j].id,
|
||||||
|
srt_value: data[j].text
|
||||||
|
}]
|
||||||
|
}
|
||||||
|
lastId = id;
|
||||||
|
textData.push(obj);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
this.global.newWindow[0].win.webContents.send(DEFINE_STRING.SHOW_MESSAGE_DIALOG, {
|
||||||
|
code: 0,
|
||||||
|
message: current_text
|
||||||
|
})
|
||||||
|
return {
|
||||||
|
code: 1,
|
||||||
|
data: textData
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
// 最后对齐
|
||||||
|
textData[textData.length - 1].end_time = srt_list[srt_list.length - 1].end_time
|
||||||
|
// 返回数据
|
||||||
|
return {
|
||||||
|
code: 1,
|
||||||
|
data: textData
|
||||||
|
}
|
||||||
|
} catch (error) {
|
||||||
|
return {
|
||||||
|
code: 0,
|
||||||
|
message: error.toString()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
490
src/main/discord/discordApi.js
Normal file
490
src/main/discord/discordApi.js
Normal file
@ -0,0 +1,490 @@
|
|||||||
|
const axios = require('axios');
|
||||||
|
// import { Midjourney as mApi} from "midjourney";
|
||||||
|
// const MJapi2 = require('midjourney');
|
||||||
|
// const { midjourney } = require('midjourney')
|
||||||
|
// const Midjourney = require('midjourney');
|
||||||
|
// ES5 的模块引入方式
|
||||||
|
const fetch = require("node-fetch");
|
||||||
|
|
||||||
|
// ES6 的模块引入方式
|
||||||
|
// import fetch from "node-fetch";
|
||||||
|
|
||||||
|
export class DiscordAPI {
|
||||||
|
constructor(mj_setting) {
|
||||||
|
// https://discord.com/api/v9/channels/1208362852482809939/messages?limit=20
|
||||||
|
this.apiClient = axios.create({
|
||||||
|
baseURL: 'https://discord.com'
|
||||||
|
});
|
||||||
|
this.DiscordBaseUrl = 'https://discord.com';
|
||||||
|
this.ServerId = mj_setting.serviceID;
|
||||||
|
this.ChannelId = mj_setting.channelID;
|
||||||
|
this.userToken = mj_setting.token;
|
||||||
|
this.botId = mj_setting.select_robot?.botId;
|
||||||
|
this.commandId = mj_setting.select_robot?.commandId;
|
||||||
|
this.versionId = mj_setting.select_robot?.versionId;
|
||||||
|
this.versionName = mj_setting.select_robot?.versionName;
|
||||||
|
this.botName = mj_setting.select_robot?.botName;
|
||||||
|
}
|
||||||
|
|
||||||
|
// 提交任务
|
||||||
|
async imagine(data) {
|
||||||
|
|
||||||
|
// let req_data = {
|
||||||
|
// "token": this.userToken,
|
||||||
|
// "method": "post",
|
||||||
|
// "api_url": "/mj/submit/imagine",
|
||||||
|
// "data":data
|
||||||
|
// }
|
||||||
|
|
||||||
|
// const response = await this.apiClient.post('/api/v3/req_mj_api', req_data);
|
||||||
|
// await this.interactions(data.prompt);
|
||||||
|
return await this.interactions(data.prompt)
|
||||||
|
// return {
|
||||||
|
// code:1,
|
||||||
|
// result:'taskid_'+new Date().getTime()
|
||||||
|
// }
|
||||||
|
// return response.data;
|
||||||
|
|
||||||
|
}
|
||||||
|
async channelList() {
|
||||||
|
axios.get(`https://discord.com/api/v9/channels/${this.ChannelId}/messages?limit=20`, {
|
||||||
|
method: 'get',
|
||||||
|
headers: {
|
||||||
|
"Authorization": this.userToken
|
||||||
|
}
|
||||||
|
})
|
||||||
|
.then(response => {
|
||||||
|
// 请求成功处理
|
||||||
|
console.log(typeof response.data);
|
||||||
|
let eList = []
|
||||||
|
let flg = false;
|
||||||
|
let job_id = '';
|
||||||
|
let type = '';
|
||||||
|
if (response && response.data) {
|
||||||
|
try {
|
||||||
|
response.data.forEach(element => {
|
||||||
|
flg = false;
|
||||||
|
type = '';
|
||||||
|
if (element.attachments && element.attachments.length) {
|
||||||
|
const arr = element.attachments[0].filename.split("_");
|
||||||
|
job_id = arr[arr.length - 1].replace(".png", '');
|
||||||
|
if (element.components) {
|
||||||
|
element.components.forEach(e2 => {
|
||||||
|
e2.components.forEach(e3 => {
|
||||||
|
if (e3.label == 'U1') {
|
||||||
|
flg = true;
|
||||||
|
type = 'U1'
|
||||||
|
} else if (e3.label && e3.label.indexOf('Upscale') > -1) {
|
||||||
|
flg = true;
|
||||||
|
type = 'Upscale'
|
||||||
|
}
|
||||||
|
})
|
||||||
|
});
|
||||||
|
}
|
||||||
|
let t = eList.find((e) => {
|
||||||
|
return e.filename == element.attachments[0].filename;
|
||||||
|
})
|
||||||
|
|
||||||
|
if (!t) {
|
||||||
|
eList.push(
|
||||||
|
{
|
||||||
|
flg: flg, job_id: job_id, filename: element.attachments[0].filename,
|
||||||
|
url: element.attachments[0].url,
|
||||||
|
proxy_url: element.attachments[0].proxy_url,
|
||||||
|
type: type,
|
||||||
|
timestamp: element.timestamp
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
// console.log({flg:flg,job_id:job_id,filename:element.attachments[0].filename})
|
||||||
|
} else {
|
||||||
|
console.log({ flg: flg })
|
||||||
|
}
|
||||||
|
|
||||||
|
});
|
||||||
|
} catch (error) {
|
||||||
|
console.log('异常2', error)
|
||||||
|
}
|
||||||
|
// console.log(eList)
|
||||||
|
// 8e7406df-bf0c-4e3d-8e49-b2bb8e2c263d
|
||||||
|
// 5abedc71-ba80-4756-8ddc-489c927d3acd
|
||||||
|
}
|
||||||
|
}).catch(error => {
|
||||||
|
// 请求失败处理
|
||||||
|
console.error(error);
|
||||||
|
});
|
||||||
|
}
|
||||||
|
// 混合
|
||||||
|
async blend(data) {
|
||||||
|
|
||||||
|
let req_data = {
|
||||||
|
"token": this.userToken,
|
||||||
|
"method": "post",
|
||||||
|
"api_url": "/mj/submit/blend",
|
||||||
|
"data": data
|
||||||
|
}
|
||||||
|
|
||||||
|
const response = await this.apiClient.post('/api/v3/req_mj_api', req_data);
|
||||||
|
return response.data;
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
// 反推
|
||||||
|
async describe(data) {
|
||||||
|
|
||||||
|
let req_data = {
|
||||||
|
"token": this.userToken,
|
||||||
|
"method": "post",
|
||||||
|
"api_url": "/mj/submit/describe",
|
||||||
|
"data": data
|
||||||
|
}
|
||||||
|
|
||||||
|
const response = await this.apiClient.post('/api/v3/req_mj_api', req_data);
|
||||||
|
return response.data;
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
// 获取任务
|
||||||
|
async getTaskId(task_id) {
|
||||||
|
let req_data = {
|
||||||
|
"token": this.userToken,
|
||||||
|
"method": "get",
|
||||||
|
"api_url": `/mj/task/${task_id}/fetch`,
|
||||||
|
"data": {}
|
||||||
|
}
|
||||||
|
const response = await this.apiClient.post('/api/v3/req_mj_api', req_data);
|
||||||
|
return response.data;
|
||||||
|
}
|
||||||
|
|
||||||
|
//获取seed
|
||||||
|
async imageSeed(task_id) {
|
||||||
|
let req_data = {
|
||||||
|
"token": this.userToken,
|
||||||
|
"method": "get",
|
||||||
|
"api_url": `/mj/task/${task_id}/image-seed`,
|
||||||
|
"data": {}
|
||||||
|
}
|
||||||
|
const response = await this.apiClient.post('/api/v3/req_mj_api', req_data);
|
||||||
|
return response.data;
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
//账号创建
|
||||||
|
async account_create(data) {
|
||||||
|
let req_data = {
|
||||||
|
"token": this.userToken,
|
||||||
|
"method": "post",
|
||||||
|
"api_url": '/mj/account/create',
|
||||||
|
"data": data
|
||||||
|
}
|
||||||
|
const response = await this.apiClient.post('/api/v3/req_mj_api', req_data, { timeout: 20000 });
|
||||||
|
return response.data;
|
||||||
|
}
|
||||||
|
|
||||||
|
//账号创建
|
||||||
|
async account_fetch(cid) {
|
||||||
|
let req_data = {
|
||||||
|
"token": this.userToken,
|
||||||
|
"method": "get",
|
||||||
|
"api_url": `/mj/account/${cid}/fetch`,
|
||||||
|
"data": {}
|
||||||
|
}
|
||||||
|
const response = await this.apiClient.post('/api/v3/req_mj_api', req_data);
|
||||||
|
return response.data;
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
//账号同步信息
|
||||||
|
async account_asyn_info(cid) {
|
||||||
|
let req_data = {
|
||||||
|
"token": this.userToken,
|
||||||
|
"method": "post",
|
||||||
|
"api_url": `/mj/account/${cid}/sync-info`,
|
||||||
|
"data": {}
|
||||||
|
}
|
||||||
|
const response = await this.apiClient.post('/api/v3/req_mj_api', req_data, { timeout: 20000 });
|
||||||
|
return response.data;
|
||||||
|
}
|
||||||
|
|
||||||
|
//账号删除信息
|
||||||
|
async account_del_info(cid) {
|
||||||
|
let req_data = {
|
||||||
|
"token": this.userToken,
|
||||||
|
"method": "delete",
|
||||||
|
"api_url": `/mj/account/${cid}/delete`,
|
||||||
|
"data": {}
|
||||||
|
}
|
||||||
|
const response = await this.apiClient.post('/api/v3/req_mj_api', req_data);
|
||||||
|
return response.data;
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
//执行动作
|
||||||
|
async action(data) {
|
||||||
|
let req_data = {
|
||||||
|
"token": this.userToken,
|
||||||
|
"method": "post",
|
||||||
|
"api_url": '/mj/submit/action',
|
||||||
|
"data": data
|
||||||
|
}
|
||||||
|
const response = await this.apiClient.post('/api/v3/req_mj_api', req_data);
|
||||||
|
return response.data;
|
||||||
|
}
|
||||||
|
|
||||||
|
// 确认弹窗
|
||||||
|
async modal(data) {
|
||||||
|
let req_data = {
|
||||||
|
"token": this.userToken,
|
||||||
|
"method": "post",
|
||||||
|
"api_url": '/mj/submit/modal',
|
||||||
|
"data": data
|
||||||
|
}
|
||||||
|
const response = await this.apiClient.post('/api/v3/req_mj_api', req_data);
|
||||||
|
return response.data;
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
async interactions(prompt) {
|
||||||
|
// 直接自己调用
|
||||||
|
prompt = prompt.trim();
|
||||||
|
// prompt = "4k,8k,best quality, masterpiece, woman, divorced, leaving fast-paced city life, serene expression, walking away from cityscape, bustling streets, entering tranquil countryside, peaceful surroundings, rejuvenating atmosphere, , --niji 5 --ar 4:3"
|
||||||
|
var payload = {};
|
||||||
|
if (this.botName == 'niji') {
|
||||||
|
payload = {
|
||||||
|
"type": 2,
|
||||||
|
"application_id": this.botId,
|
||||||
|
"guild_id": this.ServerId,
|
||||||
|
"channel_id": this.ChannelId,
|
||||||
|
"session_id": this.userToken,
|
||||||
|
"data": {
|
||||||
|
"version": this.versionId,
|
||||||
|
"id": this.commandId,
|
||||||
|
"name": "imagine",
|
||||||
|
"type": 1,
|
||||||
|
"options": [
|
||||||
|
{
|
||||||
|
"type": 3,
|
||||||
|
"name": "prompt",
|
||||||
|
"value": prompt
|
||||||
|
}
|
||||||
|
],
|
||||||
|
"application_command": {
|
||||||
|
"id": this.commandId,
|
||||||
|
"type": 1,
|
||||||
|
"application_id": this.botId,
|
||||||
|
"version": this.versionId,
|
||||||
|
"name": "imagine",
|
||||||
|
"description": "Create images with Midjourney",
|
||||||
|
"options": [
|
||||||
|
{
|
||||||
|
"type": 3,
|
||||||
|
"name": "prompt",
|
||||||
|
"description": "The prompt to imagine",
|
||||||
|
"required": true,
|
||||||
|
"description_localized": "The prompt to imagine",
|
||||||
|
"name_localized": "prompt"
|
||||||
|
}
|
||||||
|
],
|
||||||
|
"integration_types": [
|
||||||
|
0
|
||||||
|
],
|
||||||
|
"global_popularity_rank": 1,
|
||||||
|
"description_localized": "Create images with Midjourney",
|
||||||
|
"name_localized": "imagine"
|
||||||
|
},
|
||||||
|
"attachments": [
|
||||||
|
|
||||||
|
]
|
||||||
|
},
|
||||||
|
// "nonce": "1210857131343872000",
|
||||||
|
"analytics_location": "slash_ui"
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
payload = {
|
||||||
|
"type": 2,
|
||||||
|
// "application_id":"1022952195194359889",//niji
|
||||||
|
"application_id": this.botId,
|
||||||
|
"guild_id": this.ServerId,
|
||||||
|
"channel_id": this.ChannelId,
|
||||||
|
"session_id": this.userToken,
|
||||||
|
"data": {
|
||||||
|
"version": this.versionId,
|
||||||
|
"id": this.commandId,
|
||||||
|
"name": "imagine", "type": 1,
|
||||||
|
"options":
|
||||||
|
[{
|
||||||
|
"type": 3,
|
||||||
|
"name": "prompt",
|
||||||
|
"value": prompt
|
||||||
|
}],
|
||||||
|
"application_command": {
|
||||||
|
"id": this.commandId,
|
||||||
|
"type": 1,
|
||||||
|
"application_id": this.botId,
|
||||||
|
"version": this.versionId,
|
||||||
|
"name": "imagine",
|
||||||
|
"description": "Create images with Niji journey",
|
||||||
|
"options": [{ "type": 3, "name": "prompt", "description": "The prompt to imagine", "required": true, "description_localized": "The prompt to imagine", "name_localized": "prompt" }],
|
||||||
|
"integration_types": [0], "global_popularity_rank": 1, "description_localized": "Create images with Niji journey", "name_localized": "imagine"
|
||||||
|
},
|
||||||
|
"attachments": []
|
||||||
|
},
|
||||||
|
"analytics_location": "slash_ui"
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
let response = {
|
||||||
|
status: 200,
|
||||||
|
data: {}
|
||||||
|
};
|
||||||
|
|
||||||
|
|
||||||
|
try {
|
||||||
|
const headers = {
|
||||||
|
"Content-Type": "application/json",
|
||||||
|
Authorization: this.userToken,
|
||||||
|
};
|
||||||
|
await fetch(`${this.DiscordBaseUrl}/api/v9/interactions`, {
|
||||||
|
method: "POST",
|
||||||
|
body: JSON.stringify(payload),
|
||||||
|
headers: headers,
|
||||||
|
}).then(res => res.json()).then(res => {
|
||||||
|
response.data = res;
|
||||||
|
}).catch(e => {
|
||||||
|
console.error("请求失败了,详细信息:" + JSON.stringify(e));
|
||||||
|
response = {
|
||||||
|
status: 500,
|
||||||
|
data: JSON.stringify(e)
|
||||||
|
};
|
||||||
|
});
|
||||||
|
console.log('response结果')
|
||||||
|
console.log(response)
|
||||||
|
// if (response.status == 204) {
|
||||||
|
// //成功
|
||||||
|
|
||||||
|
// }
|
||||||
|
if (response.status >= 400) {
|
||||||
|
console.error("api.error.config", {
|
||||||
|
payload: JSON.stringify(payload)
|
||||||
|
});
|
||||||
|
}
|
||||||
|
return {
|
||||||
|
code: response.status,
|
||||||
|
response: response
|
||||||
|
};
|
||||||
|
}
|
||||||
|
catch (error) {
|
||||||
|
console.error(error);
|
||||||
|
return 500;
|
||||||
|
}
|
||||||
|
return;
|
||||||
|
const client = new Midjourney.Midjourney({
|
||||||
|
ServerId: this.ServerId,
|
||||||
|
ChannelId: this.ChannelId,
|
||||||
|
SalaiToken: this.userToken,
|
||||||
|
Debug: true,
|
||||||
|
fetch: fetch,
|
||||||
|
Ws: true, //enable ws is required for remix mode (and custom zoom)
|
||||||
|
});
|
||||||
|
await client.init();
|
||||||
|
console.log('mjmj_begin2', prompt);
|
||||||
|
// const prompt =
|
||||||
|
// "Christmas dinner with spaghetti with family in a cozy house, we see interior details , simple blue&white illustration";
|
||||||
|
//imagine
|
||||||
|
const Imagine = await client.Imagine(
|
||||||
|
prompt,
|
||||||
|
(uri, progress) => {
|
||||||
|
client.Close();
|
||||||
|
|
||||||
|
console.log("loading", uri, "progress", progress);
|
||||||
|
return
|
||||||
|
}
|
||||||
|
);
|
||||||
|
console.log(Imagine);
|
||||||
|
if (!Imagine) {
|
||||||
|
console.log("no message");
|
||||||
|
console.log('mjmj_end2')
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
console.log('mjmj_end')
|
||||||
|
client.Close();
|
||||||
|
return
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 获取频道内的机器人
|
||||||
|
* @returns 返回机器人列表
|
||||||
|
*/
|
||||||
|
async getBotList() {
|
||||||
|
try {
|
||||||
|
|
||||||
|
const headers = {
|
||||||
|
'Host': 'discord.com',
|
||||||
|
'Connection': 'keep-alive',
|
||||||
|
'authorization': this.userToken,
|
||||||
|
}
|
||||||
|
|
||||||
|
await fetch('https://discord.com/api/v9/guilds/1182523906855284826/application-command-index', {
|
||||||
|
method: 'GET',
|
||||||
|
headers: headers,
|
||||||
|
})
|
||||||
|
.then(response => {
|
||||||
|
response.json()
|
||||||
|
}
|
||||||
|
)
|
||||||
|
.then(data => {
|
||||||
|
console.log(data)
|
||||||
|
})
|
||||||
|
.catch(error => {
|
||||||
|
console.error('Error:', error)
|
||||||
|
})
|
||||||
|
|
||||||
|
|
||||||
|
} catch (error) {
|
||||||
|
throw new Error(error);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
async getMjMsgList() {
|
||||||
|
// const headers = {
|
||||||
|
// "Content-Type": "application/json",
|
||||||
|
// Authorization:this.userToken,
|
||||||
|
// };
|
||||||
|
let response = {
|
||||||
|
status: 200,
|
||||||
|
data: {}
|
||||||
|
};
|
||||||
|
// // `https://discord.com/api/v9/channels/${mj_channelId}/messages?limit=10
|
||||||
|
// await fetch(`${this.DiscordBaseUrl}/api/v9/channels/${this.ChannelId}/messages?limit=50`, {
|
||||||
|
// method: "GET",
|
||||||
|
// headers: headers,
|
||||||
|
// }).then(res => res.json()).then(res => {
|
||||||
|
// response.data=res;
|
||||||
|
// }).catch(e => {
|
||||||
|
// console.error("请求失败了,详细信息:" + JSON.stringify(e));
|
||||||
|
// response ={
|
||||||
|
// status:500,
|
||||||
|
// data:JSON.stringify(e)
|
||||||
|
// };
|
||||||
|
// });
|
||||||
|
// console.log('getMjMsgList_response结果')
|
||||||
|
// // console.log(response)
|
||||||
|
|
||||||
|
// return response;
|
||||||
|
axios.get(`https://discord.com/api/v9/channels/${this.ChannelId}/messages?limit=20`, {
|
||||||
|
method: 'get',
|
||||||
|
headers: {
|
||||||
|
"Authorization": this.userToken
|
||||||
|
}
|
||||||
|
}).then(res => {
|
||||||
|
response.data = res;
|
||||||
|
}).catch(error => {
|
||||||
|
// 请求失败处理
|
||||||
|
console.error(error);
|
||||||
|
});
|
||||||
|
|
||||||
|
}
|
||||||
|
}
|
||||||
28
src/preload/discordIndex.js
Normal file
28
src/preload/discordIndex.js
Normal file
@ -0,0 +1,28 @@
|
|||||||
|
import { contextBridge, ipcRenderer } from 'electron'
|
||||||
|
import { DEFINE_STRING } from '../define/define_string.js';
|
||||||
|
// Custom APIs for renderer
|
||||||
|
|
||||||
|
let events = [];
|
||||||
|
const api = {
|
||||||
|
// 创建MJ消息
|
||||||
|
CreateMessage: (value) => ipcRenderer.send(DEFINE_STRING.DISCORD.CREATE_MESSAGE, value),
|
||||||
|
|
||||||
|
// MJ消息更新
|
||||||
|
UpdateMessage: (value) => ipcRenderer.send(DEFINE_STRING.DISCORD.UPDATE_MESSAGE, value),
|
||||||
|
|
||||||
|
// MJ消息删除
|
||||||
|
DeleteMessage: (value) => ipcRenderer.send(DEFINE_STRING.DISCORD.DELETE_MESSAGE, value),
|
||||||
|
}
|
||||||
|
// Use `contextBridge` APIs to expose Electron APIs to
|
||||||
|
// renderer only if context isolation is enabled, otherwise
|
||||||
|
// just add to the DOM global.
|
||||||
|
if (process.contextIsolated) {
|
||||||
|
try {
|
||||||
|
contextBridge.exposeInMainWorld('api', api)
|
||||||
|
} catch (error) {
|
||||||
|
console.error(error)
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
window.api = api
|
||||||
|
}
|
||||||
|
|
||||||
Loading…
x
Reference in New Issue
Block a user