Update gradio_app.py

gradio_app.py

@@ -1,12 +1,12 @@
-# Hunyuan 3D is licensed under the TENCENT HUNYUAN NON-COMMERCIAL LICENSE AGREEMENT
+# PolyGenix AI is licensed under the TENCENT HUNYUAN NON-COMMERCIAL LICENSE AGREEMENT
 # except for the third-party components listed below.
-# Hunyuan 3D does not impose any additional limitations beyond what is outlined
-# in the repsective licenses of these third-party components.
+# PolyGenix AI does not impose any additional limitations beyond what is outlined
+# in the respective licenses of these third-party components.
 # Users must comply with all terms and conditions of original licenses of these third-party
 # components and must ensure that the usage of the third party components adheres to
 # all relevant laws and regulations.
 
-# For avoidance of doubts, Hunyuan 3D means the large language models and
+# For avoidance of doubts, PolyGenix AI means the large language models and
 # their software and algorithms, including trained model weights, parameters (including
 # optimizer states), machine-learning model code, inference-enabling code, training-enabling code,
 # fine-tuning enabling code and other elements of the foregoing made publicly available
@@ -133,12 +133,12 @@ def build_model_viewer_html(save_folder, height=660, width=790, textured=False):
         f.write(template_html)
 
     rel_path = os.path.relpath(output_html_path, SAVE_DIR)
-    iframe_tag = f'<iframe src="/static/{rel_path}" height="{height}" width="100%" frameborder="0" style="border-radius: 8px; box-shadow: 0 4px 6px rgba(0, 0, 0, 0.1);"></iframe>'
+    iframe_tag = f'<iframe src="/static/{rel_path}" height="{height}" width="100%" frameborder="0" style="border-radius: 8px; box-shadow: 0 4px 6px rgba(128, 0, 128, 0.3);"></iframe>'
     print(
         f'Find html file {output_html_path}, {os.path.exists(output_html_path)}, relative HTML path is /static/{rel_path}')
 
     return f"""
-        <div style='height: {height}px; width: 100%; display: flex; justify-content: center; align-items: center; padding: 10px; background: #f9f9f9; border-radius: 8px; box-shadow: 0 2px 4px rgba(0, 0, 0, 0.1);'>
+        <div style='height: {height}px; width: 100%; display: flex; justify-content: center; align-items: center; padding: 10px; background: #3a3a3a; border-radius: 8px; box-shadow: 0 2px 4px rgba(0, 0, 0, 0.5);'>
         {iframe_tag}
         </div>
     """
@@ -363,16 +363,16 @@ def shape_generation(
 
 
 def build_app():
-    title = 'PolyGenixAI: High Resolution Textured 3D Assets Generation'
+    title = 'PolyGenix AI: High Resolution Textured 3D Assets Generation'
     if MV_MODE:
-        title = 'Hunyuan3D-2mv: Image to 3D Generation with 1-4 Views'
+        title = 'PolyGenix AI-2mv: Image to 3D Generation with 1-4 Views'
     if 'mini' in args.subfolder:
-        title = 'Hunyuan3D-2mini: Strong 0.6B Image to Shape Generator'
+        title = 'PolyGenix AI-2mini: Strong 0.6B Image to Shape Generator'
     if TURBO_MODE:
         title = title.replace(':', '-Turbo: Fast ')
 
     title_html = f"""
-    <div style="font-size: 2.5em; font-weight: bold; text-align: center; margin-bottom: 15px; background: linear-gradient(90deg, #4a90e2, #9013fe); -webkit-background-clip: text; color: transparent; padding: 10px; border-radius: 8px;">
+    <div style="font-size: 2.5em; font-weight: bold; text-align: center; margin-bottom: 15px; background: linear-gradient(90deg, #2d2d2d, #800080); -webkit-background-clip: text; color: #ffffff; padding: 10px; border-radius: 8px; text-shadow: 0 2px 4px rgba(0, 0, 0, 0.5);">
     {title}
     </div>
     <div align="center">
@@ -383,78 +383,113 @@ def build_app():
     custom_css = """
     .app.svelte-wpkpf6.svelte-wpkpf6:not(.fill_width) {
         max-width: 1500px;
-        background-color: #f0f4f8;
+        background-color: #2d2d2d;
         padding: 20px;
         border-radius: 12px;
-        box-shadow: 0 4px 12px rgba(0, 0, 0, 0.1);
+        box-shadow: 0 4px 12px rgba(0, 0, 0, 0.5);
     }
     .mv-image button .wrap {
         font-size: 12px;
-        background-color: #ffffff;
+        background-color: #3a3a3a;
+        color: #ffffff;
         border-radius: 6px;
         padding: 5px;
         transition: all 0.3s ease;
     }
     .mv-image button:hover .wrap {
-        background-color: #e0e7ff;
-        color: #4a90e2;
+        background-color: #800080;
+        color: #ffffff;
     }
     .mv-image .icon-wrap {
         width: 22px;
-        background-color: #f0f4f8;
+        background-color: #3a3a3a;
         border-radius: 50%;
     }
     .gr-button {
-        background-color: #4a90e2;
-        color: white;
+        background-color: #800080;
+        color: #ffffff;
         border-radius: 8px;
         padding: 10px 20px;
         transition: all 0.3s ease;
     }
     .gr-button:hover {
-        background-color: #357abd;
+        background-color: #660066;
         transform: translateY(-2px);
     }
     .gr-slider, .gr-dropdown, .gr-checkbox, .gr-number {
-        border-color: #d1d5db;
+        border-color: #4a4a4a;
         border-radius: 8px;
-        background-color: #ffffff;
+        background-color: #3a3a3a;
+        color: #ffffff;
     }
     .gr-tab {
-        background-color: #e5e7eb;
+        background-color: #4a4a4a;
+        color: #ffffff;
         border-radius: 8px;
         padding: 5px;
     }
     .gr-tab:hover, .gr-tab:active {
-        background-color: #d1d5db;
+        background-color: #800080;
     }
     """
 
-    with gr.Blocks(theme=gr.themes.Soft(), title='Hunyuan-3D-2.0', analytics_enabled=False, css=custom_css) as demo:
+    with gr.Blocks(theme=gr.themes.Base(primary_hue="purple", secondary_hue="gray"), title='PolyGenix AI-2.0', analytics_enabled=False, css=custom_css) as demo:
         gr.HTML(title_html)
 
         with gr.Row():
             with gr.Column(scale=3):
-                with gr.Tabs(selected='tab_img_prompt') as tabs_prompt:
-                    with gr.Tab('Image Prompt', id='tab_img_prompt', visible=not MV_MODE) as tab_ip:
-                        image = gr.Image(label='Image', type='pil', image_mode='RGBA', height=290)
-
-                    with gr.Tab('Text Prompt', id='tab_txt_prompt', visible=HAS_T2I and not MV_MODE) as tab_tp:
-                        caption = gr.Textbox(label='Text Prompt',
-                                             placeholder='HunyuanDiT will be used to generate image.',
-                                             info='Example: A 3D model of a cute cat, white background')
-                    with gr.Tab('MultiView Prompt', visible=MV_MODE) as tab_mv:
-                        # gr.Label('Please upload at least one front image.')
-                        with gr.Row():
-                            mv_image_front = gr.Image(label='Front', type='pil', image_mode='RGBA', height=140,
-                                                      min_width=100, elem_classes='mv-image')
-                            mv_image_back = gr.Image(label='Back', type='pil', image_mode='RGBA', height=140,
-                                                     min_width=100, elem_classes='mv-image')
-                        with gr.Row():
-                            mv_image_left = gr.Image(label='Left', type='pil', image_mode='RGBA', height=140,
-                                                     min_width=100, elem_classes='mv-image')
-                            mv_image_right = gr.Image(label='Right', type='pil', image_mode='RGBA', height=140,
-                                                      min_width=100, elem_classes='mv-image')
+                with gr.Tabs(selected='tab_options') as tabs_options:
+                    with gr.Tab("Options", id='tab_options', visible=TURBO_MODE):
+                        gen_mode = gr.Radio(label='Generation Mode',
+                                            info='Recommendation: Turbo for most cases, Fast for very complex cases, Standard seldom use.',
+                                            choices=['Turbo', 'Fast', 'Standard'], value='Turbo')
+                        decode_mode = gr.Radio(label='Decoding Mode',
+                                               info='The resolution for exporting mesh from generated vectset',
+                                               choices=['Low', 'Standard', 'High'],
+                                               value='Standard')
+                    with gr.Tabs(selected='tab_advanced_options') as tabs_advanced:
+                        with gr.Tab('Advanced Options', id='tab_advanced_options'):
+                            with gr.Row():
+                                check_box_rembg = gr.Checkbox(value=True, label='Remove Background', min_width=100)
+                                randomize_seed = gr.Checkbox(label="Randomize seed", value=True, min_width=100)
+                            seed = gr.Slider(
+                                label="Seed",
+                                minimum=0,
+                                maximum=MAX_SEED,
+                                step=1,
+                                value=1234,
+                                min_width=100,
+                            )
+                            with gr.Row():
+                                num_steps = gr.Slider(maximum=100,
+                                                      minimum=1,
+                                                      value=5 if 'turbo' in args.subfolder else 30,
+                                                      step=1, label='Inference Steps')
+                                octree_resolution = gr.Slider(maximum=512, minimum=16, value=256, label='Octree Resolution')
+                            with gr.Row():
+                                cfg_scale = gr.Number(value=5.0, label='Guidance Scale', min_width=100)
+                                num_chunks = gr.Slider(maximum=5000000, minimum=1000, value=8000,
+                                                       label='Number of Chunks', min_width=100)
+                    with gr.Tabs(selected='tab_img_prompt') as tabs_prompt:
+                        with gr.Tab('Image Prompt', id='tab_img_prompt', visible=not MV_MODE) as tab_ip:
+                            image = gr.Image(label='Image', type='pil', image_mode='RGBA', height=290)
+
+                        with gr.Tab('Text Prompt', id='tab_txt_prompt', visible=HAS_T2I and not MV_MODE) as tab_tp:
+                            caption = gr.Textbox(label='Text Prompt',
+                                                 placeholder='PolyGenix AI DiT will be used to generate image.',
+                                                 info='Example: A 3D model of a cute cat, white background')
+                        with gr.Tab('MultiView Prompt', visible=MV_MODE) as tab_mv:
+                            # gr.Label('Please upload at least one front image.')
+                            with gr.Row():
+                                mv_image_front = gr.Image(label='Front', type='pil', image_mode='RGBA', height=140,
+                                                          min_width=100, elem_classes='mv-image')
+                                mv_image_back = gr.Image(label='Back', type='pil', image_mode='RGBA', height=140,
+                                                         min_width=100, elem_classes='mv-image')
+                            with gr.Row():
+                                mv_image_left = gr.Image(label='Left', type='pil', image_mode='RGBA', height=140,
+                                                         min_width=100, elem_classes='mv-image')
+                                mv_image_right = gr.Image(label='Right', type='pil', image_mode='RGBA', height=140,
+                                                          min_width=100, elem_classes='mv-image')
 
                 with gr.Row():
                     btn = gr.Button(value='Gen Shape', variant='primary', min_width=100)
@@ -467,37 +502,7 @@ def build_app():
                     file_out = gr.File(label="File", visible=False)
                     file_out2 = gr.File(label="File", visible=False)
 
-                with gr.Tabs(selected='tab_options' if TURBO_MODE else 'tab_export'):
-                    with gr.Tab("Options", id='tab_options', visible=TURBO_MODE):
-                        gen_mode = gr.Radio(label='Generation Mode',
-                                            info='Recommendation: Turbo for most cases, Fast for very complex cases, Standard seldom use.',
-                                            choices=['Turbo', 'Fast', 'Standard'], value='Turbo')
-                        decode_mode = gr.Radio(label='Decoding Mode',
-                                               info='The resolution for exporting mesh from generated vectset',
-                                               choices=['Low', 'Standard', 'High'],
-                                               value='Standard')
-                    with gr.Tab('Advanced Options', id='tab_advanced_options'):
-                        with gr.Row():
-                            check_box_rembg = gr.Checkbox(value=True, label='Remove Background', min_width=100)
-                            randomize_seed = gr.Checkbox(label="Randomize seed", value=True, min_width=100)
-                        seed = gr.Slider(
-                            label="Seed",
-                            minimum=0,
-                            maximum=MAX_SEED,
-                            step=1,
-                            value=1234,
-                            min_width=100,
-                        )
-                        with gr.Row():
-                            num_steps = gr.Slider(maximum=100,
-                                                  minimum=1,
-                                                  value=5 if 'turbo' in args.subfolder else 30,
-                                                  step=1, label='Inference Steps')
-                            octree_resolution = gr.Slider(maximum=512, minimum=16, value=256, label='Octree Resolution')
-                        with gr.Row():
-                            cfg_scale = gr.Number(value=5.0, label='Guidance Scale', min_width=100)
-                            num_chunks = gr.Slider(maximum=5000000, minimum=1000, value=8000,
-                                                   label='Number of Chunks', min_width=100)
+                with gr.Tabs(selected='tab_export'):
                     with gr.Tab("Export", id='tab_export'):
                         with gr.Row():
                             file_type = gr.Dropdown(label='File Type', choices=SUPPORTED_FORMATS,
@@ -512,7 +517,7 @@ def build_app():
                             file_export = gr.DownloadButton(label="Download", variant='primary',
                                                             interactive=False, min_width=100)
 
-            with gr.Column(scale=6):
+            with gr.Column(scale=7):
                 with gr.Tabs(selected='gen_mesh_panel') as tabs_output:
                     with gr.Tab('Generated Mesh', id='gen_mesh_panel'):
                         html_gen_mesh = gr.HTML(HTML_OUTPUT_PLACEHOLDER, label='Output')
@@ -539,22 +544,22 @@ def build_app():
                                         label=None, examples_per_page=6)
 
         gr.HTML(f"""
-        <div align="center" style="background-color: #e0e7ff; padding: 10px; border-radius: 8px; margin-top: 10px;">
-        Activated Model - Shape Generation ({args.model_path}/{args.subfolder}) ; Texture Generation ({'Hunyuan3D-2' if HAS_TEXTUREGEN else 'Unavailable'})
+        <div align="center" style="background-color: #3a3a3a; padding: 10px; border-radius: 8px; margin-top: 10px; color: #ffffff;">
+        Activated Model - Shape Generation ({args.model_path}/{args.subfolder}) ; Texture Generation ({'PolyGenix AI-2' if HAS_TEXTUREGEN else 'Unavailable'})
         </div>
         """)
         if not HAS_TEXTUREGEN:
             gr.HTML("""
-            <div style="margin-top: 5px; background-color: #fff3f3; padding: 10px; border-radius: 8px; border: 1px solid #fee2e2;" align="center">
-                <b style="color: #dc2626;">Warning: </b>
+            <div style="margin-top: 5px; background-color: #4a2d2d; padding: 10px; border-radius: 8px; border: 1px solid #800000; color: #ffffff;" align="center">
+                <b style="color: #ff4040;">Warning: </b>
                 Texture synthesis is disable due to missing requirements,
-                 please install requirements following <a href="https://github.com/Tencent/Hunyuan3D-2?tab=readme-ov-file#install-requirements">README.md</a>to activate it.
+                 please install requirements following <a href="https://github.com/Tencent/Hunyuan3D-2?tab=readme-ov-file#install-requirements" style="color: #800080;">README.md</a> to activate it.
             </div>
             """)
         if not args.enable_t23d:
             gr.HTML("""
-            <div style="margin-top: 5px; background-color: #fff3f3; padding: 10px; border-radius: 8px; border: 1px solid #fee2e2;" align="center">
-                <b style="color: #dc2626;">Warning: </b>
+            <div style="margin-top: 5px; background-color: #4a2d2d; padding: 10px; border-radius: 8px; border: 1px solid #800000; color: #ffffff;" align="center">
+                <b style="color: #ff4040;">Warning: </b>
                 Text to 3D is disable. To activate it, please run `python gradio_app.py --enable_t23d`.
             </div>
             """)
@@ -686,9 +691,9 @@ if __name__ == '__main__':
     import argparse
 
     parser = argparse.ArgumentParser()
-    parser.add_argument("--model_path", type=str, default='tencent/Hunyuan3D-2')
-    parser.add_argument("--subfolder", type=str, default='hunyuan3d-dit-v2-0')
-    parser.add_argument("--texgen_model_path", type=str, default='tencent/Hunyuan3D-2')
+    parser.add_argument("--model_path", type=str, default='tencent/PolyGenix AI-2')
+    parser.add_argument("--subfolder", type=str, default='polygenix-dit-v2-0')
+    parser.add_argument("--texgen_model_path", type=str, default='tencent/PolyGenix AI-2')
     parser.add_argument('--port', type=int, default=7860)
     parser.add_argument('--host', type=str, default='0.0.0.0')
     parser.add_argument('--device', type=str, default='cuda')
@@ -714,17 +719,17 @@ if __name__ == '__main__':
     HTML_HEIGHT = 690 if MV_MODE else 650
     HTML_WIDTH = 500
     HTML_OUTPUT_PLACEHOLDER = f"""
-    <div style='height: {650}px; width: 100%; border-radius: 8px; border-color: #e5e7eb; border-style: solid; border-width: 1px; display: flex; justify-content: center; align-items: center; background-color: #ffffff; box-shadow: 0 4px 6px rgba(0, 0, 0, 0.1);'>
-      <div style='text-align: center; font-size: 18px; color: #4a5568;'>
-        <p style="color: #718096;">Welcome to Hunyuan3D!</p>
-        <p style="color: #718096;">No mesh here.</p>
+    <div style='height: {650}px; width: 100%; border-radius: 8px; border-color: #4a4a4a; border-style: solid; border-width: 1px; display: flex; justify-content: center; align-items: center; background-color: #3a3a3a; box-shadow: 0 4px 6px rgba(0, 0, 0, 0.5);'>
+      <div style='text-align: center; font-size: 18px; color: #ffffff;'>
+        <p style="color: #cccccc;">Welcome to PolyGenix AI!</p>
+        <p style="color: #cccccc;">No mesh here.</p>
       </div>
     </div>
     """
 
     INPUT_MESH_HTML = """
     <div style='height: 490px; width: 100%; border-radius: 8px; 
-    border-color: #e5e7eb; order-style: solid; border-width: 1px;'>
+    border-color: #4a4a4a; order-style: solid; border-width: 1px;'>
     </div>
     """
     example_is = get_example_img_list()
@@ -736,9 +741,9 @@ if __name__ == '__main__':
     HAS_TEXTUREGEN = False
     if not args.disable_tex:
         try:
-            from hy3dgen.texgen import Hunyuan3DPaintPipeline
+            from hy3dgen.texgen import PolyGenix AIPaintPipeline
 
-            texgen_worker = Hunyuan3DPaintPipeline.from_pretrained(args.texgen_model_path)
+            texgen_worker = PolyGenix AIPaintPipeline.from_pretrained(args.texgen_model_path)
             if args.low_vram_mode:
                 texgen_worker.enable_model_cpu_offload()
             # Not help much, ignore for now.
@@ -756,18 +761,18 @@ if __name__ == '__main__':
 
     HAS_T2I = True
     if args.enable_t23d:
-        from hy3dgen.text2image import HunyuanDiTPipeline
+        from hy3dgen.text2image import PolyGenix AIDiTPipeline
 
-        t2i_worker = HunyuanDiTPipeline('Tencent-Hunyuan/HunyuanDiT-v1.1-Diffusers-Distilled')
+        t2i_worker = PolyGenix AIDiTPipeline('Tencent-Hunyuan/PolyGenix AIDiT-v1.1-Diffusers-Distilled')
         HAS_T2I = True
 
     from hy3dgen.shapegen import FaceReducer, FloaterRemover, DegenerateFaceRemover, MeshSimplifier, \
-        Hunyuan3DDiTFlowMatchingPipeline
+        PolyGenix AIDiTFlowMatchingPipeline
     from hy3dgen.shapegen.pipelines import export_to_trimesh
     from hy3dgen.rembg import BackgroundRemover
 
     rmbg_worker = BackgroundRemover()
-    i23d_worker = Hunyuan3DDiTFlowMatchingPipeline.from_pretrained(
+    i23d_worker = PolyGenix AIDiTFlowMatchingPipeline.from_pretrained(
         args.model_path,
         subfolder=args.subfolder,
         use_safetensors=True,