Update app.py

app.py

@@ -48,10 +48,8 @@ def end_session(req: gr.Request):
 def preprocess_image(image: Image.Image) -> Tuple[str, Image.Image]:
     """
     Preprocess the input image.
-
     Args:
         image (Image.Image): The input image.
-
     Returns:
         str: uuid of the trial.
         Image.Image: The preprocessed image.
@@ -120,7 +118,6 @@ def image_to_3d(
 ) -> Tuple[dict, str]:
     """
     Convert an image to a 3D model.
-
     Args:
         image (Image.Image): The input image.
         seed (int): The random seed.
@@ -128,7 +125,6 @@ def image_to_3d(
         ss_sampling_steps (int): The number of sampling steps for sparse structure generation.
         slat_guidance_strength (float): The guidance strength for structured latent generation.
         slat_sampling_steps (int): The number of sampling steps for structured latent generation.
-
     Returns:
         dict: The information of the generated 3D model.
         str: The path to the video of the 3D model.
@@ -168,12 +164,10 @@ def extract_glb(
 ) -> Tuple[str, str]:
     """
     Extract a GLB file from the 3D model.
-
     Args:
         state (dict): The state of the generated 3D model.
         mesh_simplify (float): The mesh simplification factor.
         texture_size (int): The texture resolution.
-
     Returns:
         str: The path to the extracted GLB file.
     """
@@ -187,37 +181,37 @@ def extract_glb(
 
 
 with gr.Blocks(delete_cache=(600, 600)) as demo:
-    gr.Markdown("""""")
-    
+    gr.Markdown("""<h1 style='color: #4CAF50; text-align: center;'>Image to 3D Model Generator</h1>""")
+
     with gr.Row():
         with gr.Column():
-            image_prompt = gr.Image(label="Image Prompt", format="png", image_mode="RGBA", type="pil", height=300)
-            
-            with gr.Accordion(label="Generation Settings", open=False):
-                seed = gr.Slider(0, MAX_SEED, label="Seed", value=0, step=1)
-                randomize_seed = gr.Checkbox(label="Randomize Seed", value=True)
-                gr.Markdown("Stage 1: Sparse Structure Generation")
+            image_prompt = gr.Image(label="Image Prompt", format="png", image_mode="RGBA", type="pil", height=300, container_style={"border": "2px solid #4CAF50", "border-radius": "10px", "background-color": "#F9F9F9"})
+
+            with gr.Accordion(label="Generation Settings", open=False, container_style={"border": "1px solid #2196F3", "border-radius": "8px"}):
+                seed = gr.Slider(0, MAX_SEED, label="Seed", value=0, step=1, style={"color": "#2196F3"})
+                randomize_seed = gr.Checkbox(label="Randomize Seed", value=True, style={"color": "#2196F3"})
+                gr.Markdown("<strong>Stage 1: Sparse Structure Generation</strong>", style={"color": "#2196F3"})
                 with gr.Row():
                     ss_guidance_strength = gr.Slider(0.0, 10.0, label="Guidance Strength", value=7.5, step=0.1)
                     ss_sampling_steps = gr.Slider(1, 50, label="Sampling Steps", value=12, step=1)
-                gr.Markdown("Stage 2: Structured Latent Generation")
+                gr.Markdown("<strong>Stage 2: Structured Latent Generation</strong>", style={"color": "#2196F3"})
                 with gr.Row():
                     slat_guidance_strength = gr.Slider(0.0, 10.0, label="Guidance Strength", value=3.0, step=0.1)
                     slat_sampling_steps = gr.Slider(1, 50, label="Sampling Steps", value=12, step=1)
 
-            generate_btn = gr.Button("Generate")
-            
-            with gr.Accordion(label="GLB Extraction Settings", open=False):
+            generate_btn = gr.Button("Generate", style={"background-color": "#4CAF50", "color": "white", "border": "none", "border-radius": "8px"})
+
+            with gr.Accordion(label="GLB Extraction Settings", open=False, container_style={"border": "1px solid #2196F3", "border-radius": "8px"}):
                 mesh_simplify = gr.Slider(0.9, 0.98, label="Simplify", value=0.95, step=0.01)
                 texture_size = gr.Slider(512, 2048, label="Texture Size", value=1024, step=512)
-            
-            extract_glb_btn = gr.Button("Extract GLB", interactive=False)
+
+            extract_glb_btn = gr.Button("Extract GLB", interactive=False, style={"background-color": "#FF9800", "color": "white", "border-radius": "8px"})
 
         with gr.Column():
-            video_output = gr.Video(label="Generated 3D Asset", autoplay=True, loop=True, height=300)
-            model_output = LitModel3D(label="Extracted GLB", exposure=20.0, height=300)
-            download_glb = gr.DownloadButton(label="Download GLB", interactive=False)
-            
+            video_output = gr.Video(label="Generated 3D Asset", autoplay=True, loop=True, height=300, style={"border": "2px solid #4CAF50", "border-radius": "10px"})
+            model_output = LitModel3D(label="Extracted GLB", exposure=20.0, height=300, style={"border": "2px solid #4CAF50", "border-radius": "10px"})
+            download_glb = gr.DownloadButton(label="Download GLB", interactive=False, style={"background-color": "#4CAF50", "color": "white", "border": "none", "border-radius": "8px"})
+
     output_buf = gr.State()
 
     # Example images at the bottom of the page
@@ -237,7 +231,7 @@ with gr.Blocks(delete_cache=(600, 600)) as demo:
     # Handlers
     demo.load(start_session)
     demo.unload(end_session)
-    
+
     image_prompt.upload(
         preprocess_image,
         inputs=[image_prompt],