Update app.py

app.py

@@ -13,108 +13,34 @@ from pathlib import Path
   #  ]
 
 models = [
-    "",
-    "runwayml/stable-diffusion-v1-5",
-    "CompVis/stable-diffusion-v1-4",
-    "claudfuen/photorealistic-fuen-v1",
-    "andite/anything-v4.0",
-    "naclbit/trinart_stable_diffusion_v2",
-    "nitrosocke/Arcane-Diffusion",
-    "nitrosocke/archer-diffusion",
-    "nitrosocke/elden-ring-diffusion",
-    "nitrosocke/redshift-diffusion",
-    "nitrosocke/spider-verse-diffusion", 
-    "nitrosocke/mo-di-diffusion",
-    "nitrosocke/classic-anim-diffusion",
-    "dreamlike-art/dreamlike-photoreal-1.0",
-    "dreamlike-art/dreamlike-photoreal-2.0",    
-    "wavymulder/wavyfusion",
-    "wavymulder/Analog-Diffusion",
-    "prompthero/midjourney-v4-diffusion",
-    "prompthero/openjourney",
-    "dallinmackay/Van-Gogh-diffusion",
-    "hakurei/waifu-diffusion",
-    "DGSpitzer/Cyberpunk-Anime-Diffusion",
-    "Fictiverse/Stable_Diffusion_BalloonArt_Model",
-    "dallinmackay/Tron-Legacy-diffusion",
-    "AstraliteHeart/pony-diffusion",
-    "nousr/robo-diffusion",
-    "Linaqruf/anything-v3",
-    "Omnibus/maximum_diffusion_fast",
-    "",
-]
-model_1=models[1]
-model_2=models[2]
-model_3=models[3]
-model_4=models[4]
-
-model_5=models[5]
-model_6=models[6]
-model_7=models[9]
-model_8=models[13]
-
-model_9=models[14]
-model_10=models[15]
-model_11=models[16]
-model_12=models[17]
-
-text_gen=gr.Interface.load("spaces/Omnibus/MagicPrompt-Stable-Diffusion_link",live=True, preprocess=True)
-
-proc1=gr.Interface.load(f"models/{model_1}",live=False,preprocess=True, postprocess=False)
-proc2=gr.Interface.load(f"models/{model_2}",live=False,preprocess=True, postprocess=False)
-proc3=gr.Interface.load(f"models/{model_3}",live=False,preprocess=True, postprocess=False)
-proc4=gr.Interface.load(f"models/{model_4}",live=False,preprocess=True, postprocess=False)
-
-proc5=gr.Interface.load(f"models/{model_5}",live=False,preprocess=True, postprocess=False)
-proc6=gr.Interface.load(f"models/{model_6}",live=False,preprocess=True, postprocess=False)
-proc7=gr.Interface.load(f"models/{model_7}",live=False,preprocess=True, postprocess=False)
-proc8=gr.Interface.load(f"models/{model_8}",live=False,preprocess=True, postprocess=False)
-
-proc9=gr.Interface.load(f"models/{model_9}",live=False,preprocess=True, postprocess=False)
-proc10=gr.Interface.load(f"models/{model_10}",live=False,preprocess=True, postprocess=False)
-proc11=gr.Interface.load(f"models/{model_11}",live=False,preprocess=True, postprocess=False)
-proc12=gr.Interface.load(f"models/{model_12}",live=False,preprocess=True, postprocess=False)
-
-def get_prompts(prompt_text):
-    return text_gen(prompt_text)
-def send_it1(inputs,proc1=proc1):
-        output1=proc1(inputs)
-        return(output1)
-def send_it2(inputs,proc2=proc2):
-        output2=proc2(inputs)
-        return(output2)
-def send_it3(inputs,proc3=proc3):
-        output3=proc3(inputs)
-        return(output3)
-def send_it4(inputs,proc4=proc4):
-        output4=proc4(inputs)
-        return(output4)
-
-def send_it5(inputs,proc5=proc5):
-        output5=proc5(inputs)
-        return(output5)
-def send_it6(inputs,proc6=proc6):
-        output6=proc6(inputs)
-        return(output6)
-def send_it7(inputs,proc7=proc7):
-        output7=proc7(inputs)
-        return(output7)
-def send_it8(inputs,proc8=proc8):
-        output8=proc8(inputs)
-        return(output8)
-
-def send_it9(inputs,proc9=proc9):
-        output9=proc9(inputs)
-        return(output9)
-def send_it10(inputs,proc10=proc10):
-        output10=proc10(inputs)
-        return(output10)
-def send_it11(inputs,proc11=proc11):
-        output11=proc11(inputs)
-        return(output11)
-def send_it12(inputs,proc12=proc12):
-        output12=proc12(inputs)
-        return(output12)
+    {"name": "Stable Diffusion 1.4","url": "CompVis/stable-diffusion-v1-4"},
+    {"name": "Stable Diffusion 1.5","url": "runwayml/stable-diffusion-v1-5"},
+    ]
+
+current_model = models[0]
+
+text_gen = gr.Interface.load("spaces/daspartho/prompt-extend") 
+
+models2 = []
+for model in models:
+    model_url = f"models/{model['url']}"
+    loaded_model = gr.Interface.load(model_url, live=True, preprocess=True)
+    models2.append(loaded_model)
+
+
+def text_it(inputs, text_gen=text_gen):
+    return text_gen(inputs)
+
+
+def set_model(current_model_index):
+    global current_model
+    current_model = models[current_model_index]
+    return gr.update(value=f"{current_model['name']}")
+
+
+def send_it(inputs, model_choice):
+    proc = models2[model_choice]
+    return proc(inputs)
 
 
 with gr.Blocks (css = 'main.css')  as myface:
@@ -137,92 +63,24 @@ with gr.Blocks (css = 'main.css')  as myface:
         
         with gr.Row():
         output1 = gr.Image(label="")
-       
-        with gr.Row():          
-        magic1 = gr.Textbox(label="2. GENERATE IMAGE HERE ! ", lines=2)
-
-        with gr.Row():
-            with gr.Column():
-                input_text=gr.Textbox(label="Short Prompt")
-                prompt=gr.Textbox(label="Text to Image")
-            with gr.Column():
-                see_prompts=gr.Button("Magic Prompt")
-                with gr.Row():
-                    run=gr.Button("Launch")
-                    clear_btn=gr.Button("Clear")
-
-        with gr.Row():
-                output1=gr.Image(label=(f"{model_1}"),visible=True)
-                output2=gr.Image(label=(f"{model_2}"),visible=True)
-                output3=gr.Image(label=(f"{model_3}"),visible=True)
-                output4=gr.Image(label=(f"{model_4}"),visible=True)
-        with gr.Row():
-                output5=gr.Image(label=(f"{model_5}"),visible=True)
-                output6=gr.Image(label=(f"{model_6}"),visible=True)
-                output7=gr.Image(label=(f"{model_7}"),visible=True)
-                output8=gr.Image(label=(f"{model_8}"),visible=True)
-        with gr.Row():
-                output9=gr.Image(label=(f"{model_9}"),visible=True)
-                output10=gr.Image(label=(f"{model_10}"),visible=True)
-                output11=gr.Image(label=(f"{model_11}"),visible=True)
-                output12=gr.Image(label=(f"{model_12}"),visible=True)
-        with gr.Row(visible=True):
-                start_box=gr.Number(interactive=False)
-                end_box=gr.Number(interactive=False)
-                tog_box=gr.Textbox(value=0,interactive=False)
-
-
-
-        def clear_it(val):
-            if int(val) != 0:
-                val = 0
-            else:
-                val = 0
-                pass
-            return val
-        def sesh_end(cnt,t_stamp):
-            to = t_stamp + 120
-            et = time.time()
-            if et > to and t_stamp != 0:
-                d=gr.update(value=0)
-                tog=gr.update(value=1)
-                #print(f'to: {to}  et: {et}')
-            else:
-                if cnt != 0:
-                    d=gr.update(value=et)
-                else:
-                    d=gr.update(value=0)
-                tog=gr.update(value=0)
-                #print (f'passing:  to: {to}  et: {et}')
-                pass
-            return d,tog
-        def sesh_start():
-            t = time.gmtime()
-            t_stamp = time.time()
-            current_time = time.strftime("%H:%M:%S", t)
-            return gr.update(value=t_stamp),gr.update(value=t_stamp),gr.update(value=0)
-        def clear_fn():
-            return None,None,None,None,None,None,None,None,None,None,None,None,None,None
-        
-        start_box.change(sesh_end,[start_box,end_box],[start_box,tog_box],show_progress=False)
-        run.click(sesh_start,None,[start_box,end_box,tog_box])
-        run1=run.click(send_it1, inputs=[prompt], outputs=[output1])
-        run2=run.click(send_it2, inputs=[prompt], outputs=[output2])
-        run3=run.click(send_it3, inputs=[prompt], outputs=[output3])
-        run4=run.click(send_it4, inputs=[prompt], outputs=[output4])
-        run5=run.click(send_it5, inputs=[prompt], outputs=[output5])
-        run6=run.click(send_it6, inputs=[prompt], outputs=[output6])
-        run7=run.click(send_it7, inputs=[prompt], outputs=[output7])
-        run8=run.click(send_it8, inputs=[prompt], outputs=[output8])
-        run9=run.click(send_it9, inputs=[prompt], outputs=[output9])
-        run10=run.click(send_it10, inputs=[prompt], outputs=[output10])
-        run11=run.click(send_it11, inputs=[prompt], outputs=[output11])
-        run12=run.click(send_it12, inputs=[prompt], outputs=[output12])
-        prompt1=see_prompts.click(get_prompts, inputs=[input_text], outputs=[prompt],cancels=[run,run1,run2,run3,run4,run5,run6,run7,run8,run9,run10,run11,run12])
-        tog_box.change(clear_it,tog_box,tog_box,cancels=[run1,run2,run3,run4,run5,run6,run7,run8,run9,run10,run11,run12,prompt1])
-        clear_btn.click(clear_fn, None,
-                        [input_text,prompt,output1,output2,output3,output4,output5,output6,output7,output8,output9,output10,output11,output12],
-                         cancels=[prompt1,run1,run2,run3,run4,run5,run6,run7,run8,run9,run10,run11,run12])
-
-myface.queue(concurrency_count=600,status_update_rate=1)
-myface.launch(enable_queue=True,inline=True,max_threads=600, show_api=False)
+        output2 = gr.Image(label="")
+        output3 = gr.Image(label="")
+    with gr.Row():          
+        magic1 = gr.Textbox(label="Generated Prompt", lines=2)
+        magic2 = gr.Textbox(label="Generated Prompt", lines=2)
+        magic3 = gr.Textbox(label="Generated Prompt", lines=2)
+   
+    model_name1.change(set_model, inputs=model_name1, outputs=[output1, output2, output3,])
+
+    run.click(send_it, inputs=[magic1, model_name1], outputs=[output1])
+    run.click(send_it, inputs=[magic2, model_name1], outputs=[output2])
+    run.click(send_it, inputs=[magic3, model_name1], outputs=[output3])
+   
+
+    see_prompts.click(text_it, inputs=[input_text], outputs=[magic1])
+    see_prompts.click(text_it, inputs=[input_text], outputs=[magic2])
+    see_prompts.click(text_it, inputs=[input_text], outputs=[magic3])
+   
+
+myface.queue(concurrency_count=200)
+myface.launch(inline=True, show_api=False, max_threads=400)