Update app.py

Try to add p5 sketch

app.py

@@ -8,6 +8,9 @@ from PIL import Image
 from diffusers import FlaxStableDiffusionControlNetPipeline, FlaxControlNetModel
 import cv2
 
+with open("test.html") as f:
+    lines = f.readlines()
+
 def create_key(seed=0):
     return jax.random.PRNGKey(seed)
 
@@ -49,37 +52,38 @@ def infer(prompts, negative_prompts, image):
     output_images = pipe.numpy_to_pil(np.asarray(output.reshape((num_samples,) + output.shape[-3:])))
     return output_images
 
-#gr.Interface(infer, inputs=["text", "text", "image"], outputs="gallery").launch()
-
-
-title = "Animal Pose Control Net"
-description = "This is a demo of Animal Pose ControlNet, which is a model trained on runwayml/stable-diffusion-v1-5 with new type of conditioning."
-
-#with gr.Blocks(theme=gr.themes.Default(font=[gr.themes.GoogleFont("Inconsolata"), "Arial", "sans-serif"])) as demo:
-    #gr.Markdown(
-      #  """
-    # Animal Pose Control Net
-    # This is a demo of Animal Pose Control Net, which is a model trained on runwayml/stable-diffusion-v1-5 with new type of conditioning.
-    #""")
-
-#theme = gr.themes.Default(primary_hue="green").set(
-#    button_primary_background_fill="*primary_200",
-#    button_primary_background_fill_hover="*primary_300",
-#)
-
-#gr.Interface(fn = infer, inputs = ["text"], outputs = "image",
-#            title = title, description = description, theme='gradio/soft').launch()
-
+mytheme = gr.themes.Default(primary_hue="slate")
 control_image = "https://huggingface.co/spaces/kfahn/Animal_Pose_Control_Net/blob/main/image_control.png"
 
-gr.Interface(fn = infer, inputs = ["text", "text", "image"], outputs = "gallery",
-            title = title, description = description, theme='gradio/soft',
-            #examples=[["a Labrador crossing the road", "low quality", control_image]]
-).launch()
+with gr.Blocks(theme = mytheme) as demo:  
+  gr.Markdown(
+      """
+      # Animal Pose Control Net
+      ## This is a demo of Animal Pose ControlNet, which is a model trained on runwayml/stable-diffusion-v1-5 with new type of conditioning.
+      """)
+  with gr.Column():
+    with gr.Row():
+      pos_prompts  = gr.Textbox(label="Prompt")
+      neg_prompts  = gr.Textbox(label="Negative Prompt")
+      image = gr.Image()
+
+  with gr.Column():
+    with gr.Row():
+      explain = gr.Textbox("Keypoint Tool:  Use mouse to move joints")
+    with gr.Row():
+      keypoint_tool = gr.HTML(lines)
 
 gr.Markdown(
-        """
+    """
     * [Dataset](https://huggingface.co/datasets/JFoz/dog-poses-controlnet-dataset)
     * [Diffusers model](), [Web UI model](https://huggingface.co/JFoz/dog-pose)
-    * [Training Report](https://wandb.ai/john-fozard/dog-cat-pose/runs/kmwcvae5))
+    * [Training Report](https://wandb.ai/john-fozard/dog-cat-pose/runs/kmwcvae5)
     """)
+
+btn = gr.Button("Run")
+  #btn.click(inputs = ["text", "text", "image"]) 
+
+btn.click(fn=infer, inputs = ["text", "text", "image"], outputs = "gallery",
+            examples=[["a Labrador crossing the road", "low quality", control_image]])   
+
+demo.launch()