Update app.py

app.py

@@ -12,7 +12,31 @@ uid = uuid.uuid4()
 
 #token_self = os.environ['HF_TOKEN']
 
-
+models =[
+    "",
+    "CompVis/stable-diffusion-v1-4",
+    "runwayml/stable-diffusion-v1-5",
+    "prompthero/openjourney",
+    "stabilityai/stable-diffusion-2-1",
+    "stabilityai/stable-diffusion-2-1-base",
+    "SG161222/Realistic_Vision_V1.4",
+    "Linaqruf/anything-v3.0",
+    "eimiss/EimisAnimeDiffusion_1.0v",
+    "nitrosocke/Nitro-Diffusion",
+    "wavymulder/portraitplus",
+    "22h/vintedois-diffusion-v0-1",
+    "dreamlike-art/dreamlike-photoreal-2.0",
+    "dreamlike-art/dreamlike-diffusion-1.0",
+    "wavymulder/Analog-Diffusion",
+    "nitrosocke/redshift-diffusion",
+    "claudfuen/photorealistic-fuen-v1",
+    "prompthero/openjourney-v2",
+    "johnslegers/epic-diffusion",
+    "nitrosocke/Arcane-Diffusion",
+    "darkstorm2150/Protogen_x5.8_Official_Release",
+
+]
+current_model=models[5]
 
 def slide_fn(val):
     im =(f'tmpim22-{uid}.png') 
@@ -394,6 +418,18 @@ def up_c(im,c_slider):
     img1.save(f'tmpim22-{uid}.png') 
     out = slide_fn(c_slider)
     return out
+
+text_gen=gr.Interface.load("spaces/Omnibus/MagicPrompt-Stable-Diffusion_link",live=True, preprocess=True)
+im_gen=gr.Interface.load("spaces/Omnibus/Diffusionator-mod")
+def get_prompts(prompt_text):
+    gen=text_gen(prompt_text)
+    return gen   
+
+def gen_im(prompt,model_choice,count=1):
+    out = im_gen(prompt,model_choice,count)
+    return out
+    
+    
 with gr.Blocks() as build_app:
     gr.Markdown("""<center><h1>Interactive Twitter Card Builder</h1></center>""")
 
@@ -434,23 +470,28 @@ with gr.Blocks() as build_app:
                 with gr.Row():
                     def_im = gr.Button("Load Default Image")
                     gen_im = gr.Button("Generate Image")
-
+                with gr.Accordion("Generated Image Controls"):
+                    with gr.Row():
+                        prompt_input=gr.Textbox("Prompt")
+                        model_name1 = gr.Dropdown(show_label=False, choices=[m for m in models], type="index", value=current_model, interactive=True)
+                    with gr.Row():
+                        magic_btn=gr.Button("Magic Prompt")
+                        gen_btn=gr.Button("Generate Images")
+                    gen_im_out=gr.Gallery()
                 c_image=gr.Image(label="Cropped Image", type="filepath",interactive=False)
                 c_slider=gr.Slider(0,220,value=0,step=1,label="Adjust Vertical Crop")
                 c_upbtn=gr.Button("Update Crop")
                 with gr.Tab("Build"):
-                    with gr.Row():
-                        build_button = gr.Button("Build")
-                        gr.Column()
+                    build_button = gr.Button("Build")
                     output_html = gr.HTML("""""")
                 with gr.Tab("Download"):
-                    with gr.Row():
-                        dl_button = gr.Button("Download")
-                        gr.Column()
+                    dl_button = gr.Button("Download")
                     files_out = gr.Files()
         gr.Column(scale=1)
 
-        
+    magic_btn.click(get_prompts,prompt_input,prompt_input)
+    gen_btn.click(gen_im,prompt_input,gen_im_out)
+    
     t_image.change(up_c,[t_image,c_slider],c_image)
     c_upbtn.click(slide_fn,c_slider,c_image)
     build_button.click(build_space,[t_name,t_space,t_title,t_description,t_redirect,t_image,t_image_url,option_token],output_html)