Update app.py

app.py

@@ -2,7 +2,7 @@ import torch
 import spaces
 from diffusers import StableDiffusionPipeline, DDIMScheduler, AutoencoderKL
 from transformers import AutoFeatureExtractor
-from controlnet.callable_functions import process_single_image_both_ways
+from controlnet.callable_functions import process_single_image_both_ways,make_stylecode,process_single_image
 
 from huggingface_hub import hf_hub_download
 import gradio as gr
@@ -21,7 +21,7 @@ device = "cuda"
 cv2.setNumThreads(1)
 
 @spaces.GPU(enable_queue=True)
-def generate_image(images, prompt, negative_prompt, scale, nfaa_negative_prompt, progress=gr.Progress(track_tqdm=True)):
+def generate_image(images, prompt, negative_prompt,  progress=gr.Progress(track_tqdm=True)):
 
     
     
@@ -38,6 +38,10 @@ def generate_image(images, prompt, negative_prompt, scale, nfaa_negative_prompt,
     
     yield image
 
+@spaces.GPU(enable_queue=True)
+def make_stylecode (images):
+    code = make_stylecode(ip_ckpt,images)
+    return code
 
 
 def swap_to_gallery(images):
@@ -65,15 +69,16 @@ with gr.Blocks(css=css) as demo:
                        placeholder="A photo of a [man/woman/person]...")
             
             negative_prompt = gr.Textbox(label="Negative Prompt", placeholder="low quality")
-            submit = gr.Button("Submit")
-            with gr.Accordion(open=False, label="Advanced Options"):
-                nfaa_negative_prompts = gr.Textbox(label="Appended Negative Prompts", info="Negative prompts to steer generations towards safe for all audiences outputs", value="naked, bikini, skimpy, scanty, bare skin, lingerie, swimsuit, exposed, see-through")    
-                scale = gr.Slider(label="Scale", value=0.8, step=0.1, minimum=0, maximum=2)
+            stylecode_gen = gr.Button("Make stylecode")
+            stylecode = gr.Textbox(label="stylecode")
+            submit = gr.Button("Use stylecode")
         with gr.Column():
             gallery = gr.Gallery(label="Generated Images")
+
+        stylecode_gen.click(fn=make_stylecode,inputs=[files],outputs=stylecode)
         
         submit.click(fn=generate_image,
-                    inputs=[files, prompt, negative_prompt, scale, nfaa_negative_prompts],
+                    inputs=[files, prompt, negative_prompt],
                     outputs=gallery)
     
     gr.Markdown("This demo includes extra features to mitigate the implicit bias of the model and prevent explicit usage of it to generate content with faces of people, including third parties, that is not safe for all audiences, including naked or semi-naked people.")