update desc, layout

app.py

@@ -37,28 +37,30 @@ be to the input. This pipeline requires a value of at least `1`. It's possible y
     * Cropping the image so the face takes up a larger portion of the frame.
 """
 
-def chat(image_in, message, history, progress=gr.Progress(track_tqdm=True)):
+def chat(image_in, in_steps, in_guidance_scale, in_img_guidance_scale, prompt, history, progress=gr.Progress(track_tqdm=True)):
     progress(0, desc="Starting...")
     global counter 
     #global seed 
     #img_nm = f"./edited_image_{seed}.png"
+    #print(f"seed is:{seed}")
+    #print(f"image name is:{img_nm}")
+
     counter += 1
-    #print(f"seed is : {seed}")
-    #print(f"image_in name is :{img_nm}")
-    
     #if message == "revert": --to add revert functionality later
     if counter > 1:
       # Open the image
       image_in = Image.open("edited_image.png") #(img_nm) 
-    prompt = message #eg - "turn him into cyborg"
-    edited_image = pipe(prompt, image=image_in, num_inference_steps=20, image_guidance_scale=1).images[0]
-    edited_image.save("edited_image.png") # (img_nm) #("./edited_image.png")
+    #prompt = message #eg - "turn him into cyborg"
+    #edited_image = pipe(prompt, image=image_in, num_inference_steps=20, image_guidance_scale=1).images[0]
+    edited_image = pipe(prompt, image=image_in, num_inference_steps=int(in_steps), guidance_scale=float(in_guidance_scale), image_guidance_scale=float(in_img_guidance_scale)).images[0]
+    edited_image.save("edited_image.png") #("/tmp/edited_image.png") #(img_nm)
     history = history or []
-    add_text_list = ["There you go ", "Enjoy your image! ", "Nice work! Wonder what you gonna do next! ", "Way to go! ", "Does this work for you? ", "Something like this? "]
     #Resizing the image for better display
+    add_text_list = ["There you go", "Enjoy your image!", "Nice work! Wonder what you gonna do next!", "Way to go!", "Does this work for you?", "Something like this?"]
     #response = random.choice(add_text_list) + '<img src="/file=' + img_nm[2:] + '" style="width: 200px; height: 200px;">'
-    response =  random.choice(add_text_list) + '<img src="/file=edited_image.png" style="width: 200px; height: 200px;">'
-    history.append((message, response))
+    #response = random.choice(add_text_list) + '<img src="/file=edited_image.png" style="width: 350px; height: 350px;">'
+    response = random.choice(add_text_list) + '<img src="/file=edited_image.png">' # style="width: 350px; height: 350px;">'
+    history.append((prompt, response))
     return history, history
 
 with gr.Blocks() as demo:
@@ -73,9 +75,14 @@ with gr.Blocks() as demo:
         text_in = gr.Textbox()
         state_in = gr.State()
         b1 = gr.Button('Edit the image!')
+        with gr.Accordion("Advance settings for Training and Inference", open=False):
+          gr.Markdown("Advance settings for - Number of Inference steps, Guidanace scale, and Image guidance scale.")
+          in_steps = gr.Number(label="Enter the number of Inference steps", value = 20)
+          in_guidance_scale = gr.Slider(1,10, step=0.5, label="Set Guidance scale", value=7.5)
+          in_img_guidance_scale = gr.Slider(1,10, step=0.5, label="Set Image Guidance scale", value=1.5)
       chatbot = gr.Chatbot() 
-    b1.click(chat,[image_in, text_in, state_in], [chatbot, state_in])
+    b1.click(chat,[image_in, in_steps, in_guidance_scale, in_img_guidance_scale,  text_in, state_in], [chatbot, state_in]) #, queue=True)
     gr.Markdown(help_text)
     
 demo.queue(concurrency_count=10) 
-demo.launch(debug=True, width="80%", height=1500)
+demo.launch(debug=True, width="80%", height=2000)