Bug fixes

app.py

@@ -1,39 +1,27 @@
 import gradio as gr
 from utils import *
 
-styles_mapping = {
-    "Illustration Style": '<illustration-style>', "Line Art":'<line-art>',
-    "Hitokomoru Style":'<hitokomoru-style-nao>', "Marc Allante": '<Marc_Allante>',
-    "Midjourney":'<midjourney-style>', "Hanfu Anime": '<hanfu-anime-style>',
-    "Birb Style": '<birb-style>'
-}
-with gr.Blocks() as interface:
+with gr.Blocks() as gr_interface:
     #gr.HTML(value=HTML_TEMPLATE, show_label=False)
     with gr.Row():
         text_input = gr.Textbox(
             label="Enter your prompt",
-            placeholder="Cats fighting on the road.....",
+            placeholder="Cats fighting on the road ",
         )
         concept_dropdown = gr.Dropdown(
             label="Select a Concept",
-            choices=["Illustration Style", "Line Art", "Hitokomoru Style", "Marc Allante", "Midjourney", "Hanfu Anime", "Birb Style"],
-            value='Marc Allante'
+            choices=["Illustration Style", "Line Art", "Hitokomoru Style",
+                     "Marc Allante", "Midjourney", "Hanfu Anime", "Birb Style"],
+            value="Illustration Style"
         )
 
         method_dropdown = gr.Dropdown(
             label="Select Guidance Type",
             choices=["edge", "contrast", "sharpness", "blue", "brightness"],
-            value='Contrast'
+            value="contrast"
         )
 
-        seed_slider = gr.Slider(
-            label="Random Seed",
-            minimum=0,
-            maximum=2000,
-            step=1,
-            value=42
-        )
-        inputs = [text_input, concept_dropdown, method_dropdown, seed_slider]
+        inputs = [text_input, concept_dropdown, method_dropdown]
 
     with gr.Row():
         outputs = gr.Gallery(
@@ -41,13 +29,13 @@ with gr.Blocks() as interface:
             columns=[2], rows=[1], object_fit="contain"
         )
 
-    with gr.Row():
-        button = gr.Button("Generate Image")
-        button.click(show_image, inputs=inputs, outputs=outputs)
-
     with gr.Row():
         gr.Examples(examples=get_examples(), inputs=inputs, outputs=outputs, fn=show_image, cache_examples=True)
 
+title = "Generative Art - Stable Diffusion with Styles and Custom Loss"
+demo = gr.Interface(fn=show_image, inputs=inputs, outputs=outputs, examples=get_examples(), title=title)
+
+
 
 if __name__ == "__main__":
-    interface.launch(enable_queue=True)
+    demo.launch(enable_queue=True, debug=True)

utils.py

@@ -126,7 +126,11 @@ def latents_to_pil(latents):
 
 def show_image(prompt, concept, guidance_type, seed):
 
-  prompt = f"{prompt} in the style of {styles_mapping[concept]}"
+  for idx, sd in enumerate(styles_mapping.keys()):
+    if(sd == concept):
+      break
+  seed =  seed_list[idx]
+  prompt = f"{prompt} in the style of {styles_mapping[sd]}"
   styled_image_without_loss = latents_to_pil(generate_image(seed, prompt, guidance_type, loss_flag=False))
   styled_image_with_loss = latents_to_pil(generate_image(seed, prompt, guidance_type, loss_flag=True))
   return([styled_image_without_loss, styled_image_with_loss])