Update app.py

app.py

@@ -5,7 +5,7 @@ from PIL import Image
 from gradio_client import Client
 import numpy as np
 
-DESCRIPTION = "# SDXL Pixelart"
+DESCRIPTION = "# SDXL Texture"
 
 MAX_SEED = np.iinfo(np.int32).max
 MAX_IMAGE_SIZE = int(os.getenv("MAX_IMAGE_SIZE", "1024"))
@@ -48,7 +48,7 @@ def generate(
 ):
     client = Client("hysts/SDXL")
     result = client.predict(
-        prompt="((pixelart)), ((retro illustration)), bit games, 8-bit illustration, pixelated, "+prompt,
+        prompt="((Seamless texture)), versatile pattern, high resolution, detailed design, subtle patterns, non-repetitive, smooth edges, square, "+prompt,
         negative_prompt=negative_prompt,
         prompt_2=prompt_2,
         negative_prompt_2=negative_prompt_2,
@@ -68,20 +68,34 @@ def generate(
     image = pixelate(result, pixel_size)
     return image
 
+def generate_normal_map(image):
+    # Convert image to grayscale
+    grayscale = image.convert("L")
+    grayscale_np = np.array(grayscale)
+    
+    # Compute gradients
+    grad_x, grad_y = np.gradient(grayscale_np.astype(float))
+    
+    # Normalize gradients
+    grad_x = (grad_x - grad_x.min()) / (grad_x.max() - grad_x.min())
+    grad_y = (grad_y - grad_y.min()) / (grad_y.max() - grad_y.min())
+    
+    # Create normal map
+    normal_map = np.dstack((grad_x, grad_y, np.ones_like(grad_x)))
+    normal_map = (normal_map * 255).astype(np.uint8)
+    
+    return Image.fromarray(normal_map)
+
+def compose(image):
+    normal_map = generate_normal_map(image)
+    return normal_map
 
 examples = [
-    "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k",
-    "An astronaut riding a green horse",
-    "City of Tokyo at night",
+    "A texture of wooden planks, grey wood, high contrast",
 ]
 
 with gr.Blocks(css="style.css") as demo:
     gr.Markdown(DESCRIPTION)
-    gr.DuplicateButton(
-        value="Duplicate Space for private use",
-        elem_id="duplicate-button",
-        visible=os.getenv("SHOW_DUPLICATE_BUTTON") == "1",
-    )
     with gr.Group():
         with gr.Row():
             prompt = gr.Text(
@@ -92,7 +106,8 @@ with gr.Blocks(css="style.css") as demo:
                 container=False,
             )
             run_button = gr.Button("Run", scale=0)
-        result = gr.Image(label="Result", show_label=False)
+        result_image = gr.Image(label="Texture", show_label=True)
+        result_normal = gr.Image(label="Normal", show_label=True)
     with gr.Accordion("Advanced options", open=False):
         with gr.Row():
             use_negative_prompt = gr.Checkbox(label="Use negative prompt", value=True)
@@ -103,7 +118,7 @@ with gr.Blocks(css="style.css") as demo:
             max_lines=1,
             placeholder="Enter a negative prompt",
             visible=True,
-            value="(deformed eyes, nose, ears, nose), bad anatomy, ugly",
+            value="anatomy, text, logos, faces, animals, recognizable objects, cube, sphere, human, hands",
         )
         prompt_2 = gr.Text(
             label="Prompt 2",
@@ -183,7 +198,7 @@ with gr.Blocks(css="style.css") as demo:
     gr.Examples(
         examples=examples,
         inputs=prompt,
-        outputs=result,
+        outputs=result_image,
         fn=generate,
     )
 
@@ -249,8 +264,13 @@ with gr.Blocks(css="style.css") as demo:
             apply_refiner,
             pixel_size
         ],
-        outputs=result,
-        api_name="run",
+        outputs=result_image,
+        api_name="generate",
+    ).then(
+        fn=compose,
+        inputs=[result_image],
+        outputs=[result_normal],
+        api_name="compose",
     )
 
 if __name__ == "__main__":