Update app.py

app.py

@@ -92,6 +92,7 @@ def imagebb(prompt, img, seed):
         except:
             print(r)
             bbtries += 1
+            print(bbtries)
             if bbtries < 3:
                 return imagebb(prompt, img, seed)
             else:
@@ -132,7 +133,7 @@ def imgur(prompt, img, seed):
     return link, loglink
     
 sdtries = 0
-def sd(prompt):
+def sd(prompt, negprompt):
     if prompt == "":
         return Image.open('error.png'), "Error! Please enter a prompt"
 
@@ -155,10 +156,11 @@ def sd(prompt):
     stability_api = client.StabilityInference(
         key = sdkey,
         verbose=True,
-        engine="stable-diffusion-v1-5",
+        engine="stable-diffusion-512-v2-1",
     )
     # convert string prompt to lowercase
     prompt = prompt.lower()
+    negprompt = negprompt.lower()
 
     # set random seed
     seed = random.randint(0, 9999999)
@@ -189,10 +191,12 @@ def sd(prompt):
                 # send prompt to dream API
                 # set random seed
                 answers = stability_api.generate(
-                    prompt=prompt,
-                    steps=15,
+                    #prompt=prompt,
+                    steps=10,
                     seed=seed,
-                )
+                    prompt = [generation.Prompt(text=prompt,parameters=generation.PromptParameters(weight=1)), 
+                    generation.Prompt(text=negprompt,parameters=generation.PromptParameters(weight=-1))],
+                    )
 
                 # convert seed to string
                 seed = str(seed)
@@ -230,19 +234,19 @@ def sd(prompt):
     
 
 
-examples = ["Futuristic utopian city in Coruscant. god beams. white buildings. street view. golden sunset. colorful, cityscape, green trees, future, urban, megapolis, breathtaking, Imaginative, utopia, Wallpaper, beautiful design, scifi, high detail, global illumination, ArtStation, art by Sebastian Wagner, Stephan Martiniere, Leon Tukker, Vitaly-Sokol",
-"A dream of a distant galaxy, by Caspar David Friedrich, matte painting trending on artstation HQ",
-"1990s movie, orbit space new york city street with many pedestrians as a loading screen, cinestill 800t 18mm, heavy grainy picture, very detailed, high quality, 4k panoramic, dramatic lightning, streetlight at night, rain, mud, foggy, anchray flags",
-"Portrait of a woman by Greg Rutkowski, she is about 30 years old, pretty, blond hair with two strans around her face, slavic features, melancholic gaze, pretty aquiline nose, she is wearing a blue utilitarian jumpsuit, highly detailed portrait, digital painting, artstation, concept art, smooth, sharp foccus ilustration, Artstation HQ.",
-"A small cabin on top of a snowy mountain in the style of Disney, artstation",
-"futuristic space landscape by Jessica Rossier and HR Giger",
-"Tiny cute isometric living room in a cutaway box, soft smooth lighting, soft colors, purple and blue color scheme, soft colors, 100mm lens, 3d blender render",
-"techno mining tailing in chuquicamata bioremediation white futuristic horizontal architecture, epic, cinematic, hyperealistic, high detailed, corona render, hdr, ray tracing",
-"A highly detailed matte painting of Orgrimmar the warrior city by Studio Ghibli, Makoto Shinkai, by Artgerm, by beeple, volumetric lighting, octane render, 4K resolution, trending on artstation, masterpiece",
-"Cute close portrait dainty beautiful futurstic white robotic girl, portraits, white cyberpunk inflatable formfitting tops, highly detailed sharp big eyes, white fringeshort hair, transparent intricate details, professional 3d visualisation in pastel colours, by wlop, intricate linework, trending on artstation, unreal engine 5 highly rendered, epic composition, by guweiz, shal. e, laica chrose, final fantasy",]
+examples = [["Futuristic utopian city in Coruscant. god beams. white buildings. street view. golden sunset. colorful, cityscape, green trees, future, urban, megapolis, breathtaking, Imaginative, utopia, Wallpaper, beautiful design, scifi, high detail, global illumination, ArtStation, art by Sebastian Wagner, Stephan Martiniere, Leon Tukker, Vitaly-Sokol","low quality"],
+["A dream of a distant galaxy, by Caspar David Friedrich, matte painting trending on artstation HQ","low quality"],
+["1990s movie, orbit space new york city street with many pedestrians as a loading screen, cinestill 800t 18mm, heavy grainy picture, very detailed, high quality, 4k panoramic, dramatic lightning, streetlight at night, rain, mud, foggy, anchray flags","low quality"],
+["Portrait of a woman by Greg Rutkowski, she is about 30 years old, pretty, blond hair with two strans around her face, slavic features, melancholic gaze, pretty aquiline nose, she is wearing a blue utilitarian jumpsuit, highly detailed portrait, digital painting, artstation, concept art, smooth, sharp foccus ilustration, Artstation HQ.","low quality"],
+["A small cabin on top of a snowy mountain in the style of Disney, artstation","low quality"],
+["futuristic space landscape by Jessica Rossier and HR Giger","low quality"],
+["Tiny cute isometric living room in a cutaway box, soft smooth lighting, soft colors, purple and blue color scheme, soft colors, 100mm lens, 3d blender render","low quality"],
+["techno mining tailing in chuquicamata bioremediation white futuristic horizontal architecture, epic, cinematic, hyperealistic, high detailed, corona render, hdr, ray tracing","low quality"],
+["A highly detailed matte painting of Orgrimmar the warrior city by Studio Ghibli, Makoto Shinkai, by Artgerm, by beeple, volumetric lighting, octane render, 4K resolution, trending on artstation, masterpiece","low quality"],
+["Cute close portrait dainty beautiful futurstic white robotic girl, portraits, white cyberpunk inflatable formfitting tops, highly detailed sharp big eyes, white fringeshort hair, transparent intricate details, professional 3d visualisation in pastel colours, by wlop, intricate linework, trending on artstation, unreal engine 5 highly rendered, epic composition, by guweiz, shal. e, laica chrose, final fantasy","low quality"]]
 
 with gr.Blocks(css = """
-*, body, #name, #ex, #article, .border-gray-200, .gr-input, .border-gray-100 { background: #0b0f19; color: white; border-color: #4c4c4c; } 
+*, body, #name, #negprompt, #ex, #article, .border-gray-200, .gr-input, .border-gray-100 { background: #0b0f19; color: white; border-color: #4c4c4c; } 
 .gr-samples-gallery .gr-sample-textbox, #greet_btn, #save_btn, .py-1, .gr-samples-gallery:hover .gr-sample-textbox:hover, .gr-text-input, #output, .wrap.svelte-1c38quw, .h-wrap { background: #1f2937; color: white; border-color: #4c4c4c;}
 .py-6 {padding-top: 0;}
 """
@@ -250,10 +254,10 @@ with gr.Blocks(css = """
     download = gr.HTML(elem_id="download")
     output = gr.Image(label="Image Generation", elem_id="output")
     name = gr.Textbox(label="Prompt", placeholder="Describe the image you want to generate. Longer and more detailed prompts work better.", elem_id="name") 
+    negprompt = gr.Textbox(label="Negative Prompt", placeholder="Describe the image you want to avoid. Longer and more detailed prompts work better.", elem_id="negprompt")
     greet_btn = gr.Button("Generate Image", elem_id="greet_btn")
     bin = gr.HTML("", visible=False)
-    ex = gr.Examples(examples=examples, fn=sd, inputs=name, outputs=[output, download], cache_examples=True)
-    ex.dataset.headers = [""]
+    ex = gr.Examples(examples=examples, fn=sd, inputs=[name, negprompt], outputs=[output, download], cache_examples=True)
     greet_btn.click(
             None,
             [],
@@ -269,7 +273,7 @@ with gr.Blocks(css = """
                 Cookies.set('creations', parseInt(Cookies.get('creations')) + 1, { expires: 250 / 24 / 60 / 60 }); }
                 console.log(Cookies.get('creations'));
             }""")
-    greet_btn.click(fn=sd, inputs=name, outputs=[output, download])
+    greet_btn.click(fn=sd, inputs=[name, negprompt], outputs=[output, download])
 
 
 demo.queue(concurrency_count=3, max_size=20).launch(max_threads=150)