update app.py

app.py

@@ -8,6 +8,11 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
 from loguru import logger
 
 from utils import get_tmt_client, getTextTrans_tmt
+import subprocess
+
+result = subprocess.run(['pip', 'list'], check=True)
+print(f'pip list = {result}')
+
 tmt_client = get_tmt_client()
 
 # language_translation_model = hub.Module(directory=f'./baidu_translate')
@@ -33,6 +38,8 @@ space_ids = {
             "spaces/runwayml/stable-diffusion-v1-5": "SD 1.5",
             "spaces/stabilityai/stable-diffusion-1": "SD 1.0",
             "dalle_mini_tab": "Dalle mini",
+            "spaces/ai-forever/Kandinsky2.1": "Kandinsky2.1",
+            "spaces/yizhangliu/openjourney-v2": "openjourney",            
             "spaces/IDEA-CCNL/Taiyi-Stable-Diffusion-Chinese": "Taiyi(太乙)",
             }
 
@@ -135,6 +142,7 @@ start_work = """async() => {
             valueSetter.call(element, value);
       }
     }
+
     window['tab_advanced'] = 0;
     
     var gradioEl = document.querySelector('body > gradio-app').shadowRoot;
@@ -144,8 +152,10 @@ start_work = """async() => {
     
     if (typeof window['gradioEl'] === 'undefined') {
         window['gradioEl'] = gradioEl;
+
         tabitems = window['gradioEl'].querySelectorAll('.tabitem');
         tabitems_title = window['gradioEl'].querySelectorAll('#tab_demo')[0].children[0].children[0].children;
+
         window['dalle_mini_block'] = null;
         window['dalle_mini_iframe'] = null;
         for (var i = 0; i < tabitems.length; i++) {   
@@ -163,6 +173,15 @@ start_work = """async() => {
                 } else if (tabitems_title[i].innerText.indexOf('SD 2') >= 0) {
                     tabitems[i].children[0].children[1].children[3].children[0].click();
                 }
+            } else if (tabitems_title[i].innerText.indexOf('ernie') >= 0) {
+                tabitems[i].children[0].children[0].style.display='none';
+                tabitems[i].children[0].children[1].children[0].style.display='none';
+                tabitems[i].children[0].children[1].children[1].style.display='none';
+                tabitems[i].children[0].children[1].children[4].style.display='none';
+                tabitems[i].children[0].children[1].children[5].style.display='none';
+                tabitems[i].children[0].children[1].children[6].style.display='none';
+                tabitems[i].children[0].children[1].children[7].style.display='none';
+                tabitems[i].children[0].children[1].children[8].style.display='none';
             } else if (tabitems_title[i].innerText.indexOf('Taiyi') >= 0) {
                 tabitems[i].children[0].children[0].children[1].style.display='none';
                 tabitems[i].children[0].children[0].children[0].children[0].children[1].style.display='none';
@@ -170,7 +189,15 @@ start_work = """async() => {
                 tabitems[i].childNodes[0].children[0].children[1].style.display='none';
             } else if (tabitems_title[i].innerText.indexOf('Dalle mini') >= 0) {
                 window['dalle_mini_block']=  tabitems[i];
-            }
+            } else if (tabitems_title[i].innerText.indexOf('openjourney') >= 0) {
+                tabitems[i].children[0].children[0].style.display='none';
+                tabitems[i].children[0].children[1].children[0].style.display='none';
+                tabitems[i].children[0].children[1].children[1].children[0].children[1].style.display='none';
+            } else if (tabitems_title[i].innerText.indexOf('Kandinsky') >= 0) {
+                tabitems[i].children[0].children[0].style.display='none';
+                tabitems[i].children[0].children[1].children[0].style.display='none';
+                tabitems[i].children[0].children[1].children[2].style.display='none';
+            }          
         }  
         
         tab_demo = window['gradioEl'].querySelectorAll('#tab_demo')[0];
@@ -186,6 +213,7 @@ start_work = """async() => {
         }
         page1.style.display = "none";
         page2.style.display = "block";  
+
         prompt_work = window['gradioEl'].querySelectorAll('#prompt_work');
         for (var i = 0; i < prompt_work.length; i++) {
             prompt_work[i].style.display='none';
@@ -226,9 +254,18 @@ start_work = """async() => {
                                 if (tabitems_title[i].innerText.indexOf('SD') >= 0) {
                                     text_value = window['gradioEl'].querySelectorAll('#prompt_work')[0].querySelectorAll('textarea')[0].value;
                                     inputText = tabitems[i].children[0].children[1].children[0].querySelectorAll('.gr-text-input')[0];
+                                } else if (tabitems_title[i].innerText.indexOf('ernie') >= 0) {
+                                    text_value = window['gradioEl'].querySelectorAll('#prompt_work_zh')[0].querySelectorAll('textarea')[0].value;
+                                    inputText = tabitems[i].children[0].children[1].children[0].querySelectorAll('.gr-text-input')[0];
                                 } else if (tabitems_title[i].innerText.indexOf('Taiyi') >= 0) {
                                     text_value = window['gradioEl'].querySelectorAll('#prompt_work_zh')[0].querySelectorAll('textarea')[0].value;
                                     inputText = tabitems[i].children[0].children[0].children[1].querySelectorAll('.gr-text-input')[0];
+                                } else if (tabitems_title[i].innerText.indexOf('openjourney') >= 0) {
+                                    text_value = window['gradioEl'].querySelectorAll('#prompt_work')[0].querySelectorAll('textarea')[0].value;
+                                    inputText = tabitems[i].children[0].children[1].children[0].children[0].querySelectorAll('.gr-text-input')[0]
+                                } else if (tabitems_title[i].innerText.indexOf('Kandinsky') >= 0) {
+                                    text_value = window['gradioEl'].querySelectorAll('#prompt_work')[0].querySelectorAll('textarea')[0].value;
+                                    inputText = tabitems[i].children[0].children[1].children[0].children[0].querySelectorAll('.gr-text-input')[0]
                                 }
                                 if (inputText) {
                                     setNativeValue(inputText, text_value);
@@ -239,7 +276,7 @@ start_work = """async() => {
                             setTimeout(function() {
                                 btns = window['gradioEl'].querySelectorAll('button');
                                 for (var i = 0; i < btns.length; i++) {
-                                    if (['Generate image','Run', '生成图像(Generate)'].includes(btns[i].innerText)) {
+                                    if (['Generate image','Run','生成图像(Generate)','Submit', 'Generate video'].includes(btns[i].innerText)) {
                                         btns[i].click();                
                                     }
                                 }
@@ -272,10 +309,21 @@ switch_tab_advanced = """async() => {
                 if (j != 1) {
                     tabitems[i].childNodes[0].children[1].children[j].style.display=action;
                 }
-            }            
+            }  
+        } else if (tabitems_title[i].innerText.indexOf('ernie') >= 0) {
+            tabitems[i].children[0].children[1].children[0].style.display=action;
+            tabitems[i].children[0].children[1].children[1].style.display=action;
+            tabitems[i].children[0].children[1].children[4].style.display=action;
+            tabitems[i].children[0].children[1].children[5].style.display=action;
+            tabitems[i].children[0].children[1].children[6].style.display=action;
+            tabitems[i].children[0].children[1].children[7].style.display=action;            
         } else if (tabitems_title[i].innerText.indexOf('Taiyi') >= 0) {
             tabitems[i].children[0].children[0].children[1].style.display=action;
-        }
+        } else if (tabitems_title[i].innerText.indexOf('openjourney') >= 0) {
+            tabitems[i].children[0].children[1].children[0].style.display=action;
+        } else if (tabitems_title[i].innerText.indexOf('Kandinsky') >= 0) {
+            tabitems[i].children[0].children[1].children[0].style.display=action;
+        }           
     }
     return false;
 }"""
@@ -391,5 +439,4 @@ with gr.Blocks(title='Text-to-Image') as demo:
                 draw_btn_0.click(fn=prompt_draw_2, inputs=[prompt_input0], outputs=[prompt_work, prompt_work_zh])
                 draw_btn_1.click(fn=prompt_draw_2, inputs=[prompt_input1], outputs=[prompt_work, prompt_work_zh])                
 
-demo.queue()
-demo.launch()
+demo.queue(concurrency_count=128).launch()

requirements.txt

@@ -2,8 +2,9 @@ ftfy
 spacy
 transformers 
 torch
-gradio
+gradio==3.14.0
 paddlepaddle==2.3.2
 paddlehub
+requests
 loguru
 tencentcloud-sdk-python