Update app.py

app.py

@@ -35,6 +35,17 @@ import cv2
 import os
 import random
 
+js_func = """
+function refresh() {
+    const url = new URL(window.location);
+
+    if (url.searchParams.get('__theme') !== 'dark') {
+        url.searchParams.set('__theme', 'dark');
+        window.location.href = url.href;
+    }
+}
+"""
+
 MAX_SEED = np.iinfo(np.int32).max
 TMP_DIR = os.path.join(os.path.dirname(os.path.abspath(__file__)), 'tmp')
 os.makedirs(TMP_DIR, exist_ok=True)
@@ -68,6 +79,7 @@ def end_session(req: gr.Request):
 
 def preprocess_image(image: Image.Image, 
                      prompt: str,
+                    negative_prompt: str = "",
                     num_steps: int = 25,
                     guidance_scale: float = 5,
                     controlnet_conditioning_scale: float = 1.0,) -> Image.Image:
@@ -309,7 +321,7 @@ def split_image(image: Image.Image) -> List[Image.Image]:
     return [preprocess_image(image) for image in images]
 
 
-with gr.Blocks(delete_cache=(600, 600)) as demo:
+with gr.Blocks(delete_cache=(600, 600), js=js_func) as demo:
     gr.Markdown("""
     ## Image to 3D Asset with [TRELLIS](https://trellis3d.github.io/)
     * Upload an image and click "Generate" to create a 3D asset. If the image has alpha channel, it be used as the mask. Otherwise, we use `rembg` to remove the background.
@@ -324,7 +336,9 @@ with gr.Blocks(delete_cache=(600, 600)) as demo:
                 with gr.Tab(label="Single Image", id=0) as single_image_input_tab:
                     #image_prompt = gr.Image(label="Image Prompt", format="png", image_mode="RGBA", type="pil", height=300)
                     image_prompt = image = gr.ImageEditor(type="pil", image_mode="L", crop_size=(512, 512))
-                    prompt = gr.Textbox(label="Prompt")
+                    with gr.Row():
+                        prompt = gr.Textbox(label="Prompt")
+                        #negative_prompt = gr.Textbox(label="Negative prompt")
                 with gr.Tab(label="Multiple Images", id=1) as multiimage_input_tab:
                     multiimage_prompt = gr.Gallery(label="Image Prompt", format="png", type="pil", height=300, columns=3)
                     gr.Markdown("""
@@ -408,7 +422,7 @@ with gr.Blocks(delete_cache=(600, 600)) as demo:
     
     image_prompt.upload(
         preprocess_image,
-        inputs=[image_prompt, prompt],
+        inputs=[image_prompt, prompt, negative_prompt],
         outputs=[image_prompt],
     )
     multiimage_prompt.upload(