update gui

app.py

@@ -9,7 +9,7 @@ from app_texnet import create_demo as create_demo_texnet
 from model import Model
 from settings import ALLOW_CHANGING_BASE_MODEL, DEFAULT_MODEL_ID, SHOW_DUPLICATE_BUTTON
 
-DESCRIPTION = "# Material Authoring Demo v0.1. Under Construction"
+DESCRIPTION = "# Material Authoring Demo v0.2"
 
 if not torch.cuda.is_available():
     DESCRIPTION += "\n<p>Running on CPU 🥶 This demo does not work on CPU.</p> Check if the 'CUDA_VISIBLE_DEVICES' are set incorrectly in settings.py"
@@ -26,12 +26,8 @@ with gr.Blocks() as demo:
     )
 
     with gr.Tabs():
-        # with gr.Tab("Canny"):
-            # create_demo_canny(model.process_canny)
         with gr.Tab("Texnet+Matnet"):
             create_demo_texnet(model.process_texnet)
-        # with gr.Tab("Matnet"):
-            # create_demo_texnet(model.process_texnet)
 
     with gr.Accordion(label="Base model", open=False):
         with gr.Row():

app_texnet.py

@@ -92,7 +92,7 @@ def create_demo(process):
     with gr.Blocks() as demo:
         with gr.Row():
             with gr.Column():
-                gr.Markdown("Select from the examples")
+                gr.Markdown("## Select preset from the example list, and modify the prompt accordingly")
                 with gr.Row():
                     name = gr.Textbox(label="Name", interactive=False, visible=False)
                     representative = gr.Image(label="Geometry", interactive=False)
@@ -119,22 +119,18 @@ def create_demo(process):
                         value="longbody, lowres, bad anatomy, bad hands, missing fingers, extra digit, fewer digits, cropped, worst quality, low quality",
                     )
             with gr.Column():
-                # tex_fine = gr.Gallery(label="Output Texture", show_label=True, columns=2, object_fit="scale-down")
-                base_color = gr.Gallery(label="Base Color", show_label=True, columns=1, object_fit="scale-down")
-                normal = gr.Gallery(label="Normal Map", show_label=True, columns=1, object_fit="scale-down")
-                roughness = gr.Gallery(label="Roughness Map", show_label=True, columns=1, object_fit="scale-down")
-                metallic = gr.Gallery(label="Metallic Map", show_label=True, columns=1, object_fit="scale-down")
-                # base_color = gr.Image(label="Base Color", show_label=True, object_fit="scale-down")
-                # normal = gr.Image(label="Normal Map", show_label=True, object_fit="scale-down")
-                # roughness = gr.Image(label="Roughness Map", show_label=True, object_fit="scale-down")
-                # metallic = gr.Image(label="Metallic Map", show_label=True, object_fit="scale-down")
+                # 2x2 grid of images for the output textures
+                gr.Markdown("### Output BRDF")
+                with gr.Row():
+                    base_color = gr.Gallery(label="Base Color", show_label=True, columns=1, object_fit="scale-down")
+                    normal = gr.Gallery(label="Displacement Map", show_label=True, columns=1, object_fit="scale-down")
+                with gr.Row():
+                    roughness = gr.Gallery(label="Roughness Map", show_label=True, columns=1, object_fit="scale-down")
+                    metallic = gr.Gallery(label="Metallic Map", show_label=True, columns=1, object_fit="scale-down")
                 
+                gr.Markdown("### Download Packed Blender Files for 3D Visualization")
                 out_blender_path = gr.File(label="Generated Blender File", file_types=[".blend"])
 
-        # radio buttons let the user toggle which texture to view
-        # texture_choice = gr.Radio(["Input", "Coarse", "Fine"], label="Preview texture", value="Input")
-        # mesh_path_state = gr.State("examples/bunny/mesh.obj")
-
         inputs = [
             name,  # Name of the object
             representative,  # Geometry mesh