Update app.py

app.py

@@ -46,17 +46,40 @@ sys.path.append(MV_ADAPTER_CODE_DIR)
 sys.path.append(os.path.join(MV_ADAPTER_CODE_DIR, "scripts"))
 
 HEADER = """
-# 🔮 Image to 3D with [PolyGenixAI](https://www.anvilinteractive.com/polygenixai)
-## Cutting-Edge Open Source 3D Generation Powered by AnvilInteractive Solutions
-<p style="font-size: 1.1em;">By <a href="https://www.anvilinteractive.com/" style="color: #1E90FF; text-decoration: none; font-weight: bold;">AnvilInteractive Solutions</a></p>
-## 📋 Quick Start Guide:
-1. **Upload an image** (single object works best)
-2. Click **Generate Shape** to create the 3D mesh
-3. Click **Apply Texture** to add textures
-4. Use **Download GLB** to save your 3D model
-5. Adjust parameters under **Generation Settings** for fine-tuning
-Best results come from clean, well-lit images with clear subject isolation. Try it now!
-<p style="font-size: 0.9em; margin-top: 10px;">Texture generation enhanced by advanced multi-view technology from AnvilInteractive Solutions. Explore more at <a href="https://www.anvilinteractive.com/polygenixai" style="color: #1E90FF; text-decoration: none;">PolyGenixAI</a>.</p>
+# 🎨 PolyGenixAI: Transform Ideas into 3D Masterpieces
+## Unleash Your Creativity with AI-Powered 3D Generation by AnvilInteractive Solutions
+<p style="font-size: 1.1em;">By <a href="https://www.anvilinteractive.com/" style="color: #3B82F6; text-decoration: none; font-weight: bold;">AnvilInteractive Solutions</a></p>
+## 🚀 Get Started:
+1. **Upload an Image** (clear, single-object images work best)
+2. **Select Filters** to customize styles
+3. Click **Generate 3D Model** to create your mesh
+4. Click **Apply Texture** to enhance with realistic textures
+5. **Download GLB** to save your creation
+<p style="font-size: 0.9em; margin-top: 10px;">Powered by advanced AI and multi-view technology from AnvilInteractive Solutions. Join our community at <a href="https://www.anvilinteractive.com/polygenixai" style="color: #3B82F6; text-decoration: none;">PolyGenixAI</a> for tips and inspiration.</p>
+<style>
+  .gr-button-primary {
+    background-color: #3B82F6 !important;
+    color: white !important;
+    border-radius: 8px !important;
+    padding: 10px 20px !important;
+    font-weight: 600 !important;
+  }
+  .gr-button-secondary {
+    background-color: #E5E7EB !important;
+    color: #1F2937 !important;
+    border-radius: 8px !important;
+    padding: 10px 20px !important;
+  }
+  .gr-panel {
+    border-radius: 12px !important;
+    box-shadow: 0 4px 6px rgba(0,0,0,0.1) !important;
+    padding: 20px !important;
+  }
+  .gr-accordion {
+    background-color: #F9FAFB !important;
+    border-radius: 8px !important;
+  }
+</style>
 """
 
 # triposg
@@ -370,63 +393,75 @@ def run_texture(image: Image, mesh_path: str, seed: int, req: gr.Request):
     return textured_glb_path
 
 
-with gr.Blocks(title="PolyGenixAI") as demo:
+with gr.Blocks(title="PolyGenixAI", css="body { background-color: #F3F4F6; } .gr-panel { background-color: white; }") as demo:
     gr.Markdown(HEADER)
 
-    with gr.Row():
-        with gr.Column():
+    with gr.Tabs():
+        with gr.Tab("Create 3D Model"):
             with gr.Row():
-                image_prompts = gr.Image(label="Input Image", type="filepath")
-                seg_image = gr.Image(
-                    label="Segmentation Result", type="pil", format="png", interactive=False
-                )
-
-            with gr.Accordion("Generation Settings", open=True):
-                seed = gr.Slider(
-                    label="Seed",
-                    minimum=0,
-                    maximum=MAX_SEED,
-                    step=0,
-                    value=0
-                )
-                randomize_seed = gr.Checkbox(label="Randomize seed", value=True)
-                num_inference_steps = gr.Slider(
-                    label="Number of inference steps",
-                    minimum=8,
-                    maximum=50,
-                    step=1,
-                    value=50,
-                )
-                guidance_scale = gr.Slider(
-                    label="CFG scale",
-                    minimum=0.0,
-                    maximum=20.0,
-                    step=0.1,
-                    value=7.0,
-                )
-
-                with gr.Row():
-                    reduce_face = gr.Checkbox(label="Simplify Mesh", value=True)
-                    target_face_num = gr.Slider(maximum=1000000, minimum=10000, value=DEFAULT_FACE_NUMBER, label="Target Face Number")
-
-                gen_button = gr.Button("Generate Shape", variant="primary")
-                gen_texture_button = gr.Button("Apply Texture", interactive=False)
-
-        with gr.Column():
-            model_output = gr.Model3D(label="PolyGenixAI GLB", interactive=False)
-            textured_model_output = gr.Model3D(label="PolyGenixAI Textured GLB", interactive=False)
-
-    with gr.Row():
-        examples = gr.Examples(
-            examples=[
-                f"{TRIPOSG_CODE_DIR}/assets/example_data/{image}"
-                for image in os.listdir(f"{TRIPOSG_CODE_DIR}/assets/example_data")
-            ],
-            fn=run_full,
-            inputs=[image_prompts],
-            outputs=[seg_image, model_output, textured_model_output],
-            cache_examples=True,
-        )
+                with gr.Column(scale=1):
+                    image_prompts = gr.Image(label="Upload Image", type="filepath", height=300)
+                    seg_image = gr.Image(label="Preview Segmentation", type="pil", format="png", interactive=False, height=300)
+                    with gr.Accordion("Style & Settings", open=True):
+                        style_filter = gr.Dropdown(
+                            choices=["None", "Realistic", "Fantasy", "Cartoon", "Sci-Fi", "Vintage"],
+                            label="Style Filter",
+                            value="None",
+                            info="Select a style to enhance your 3D model (optional)"
+                        )
+                        seed = gr.Slider(
+                            label="Seed",
+                            minimum=0,
+                            maximum=MAX_SEED,
+                            step=1,
+                            value=0
+                        )
+                        randomize_seed = gr.Checkbox(label="Randomize Seed", value=True)
+                        num_inference_steps = gr.Slider(
+                            label="Inference Steps",
+                            minimum=8,
+                            maximum=50,
+                            step=1,
+                            value=50,
+                            info="Higher steps improve quality but take longer"
+                        )
+                        guidance_scale = gr.Slider(
+                            label="Guidance Scale",
+                            minimum=0.0,
+                            maximum=20.0,
+                            step=0.1,
+                            value=7.0,
+                            info="Controls how closely the model follows the input"
+                        )
+                        reduce_face = gr.Checkbox(label="Simplify Mesh", value=True)
+                        target_face_num = gr.Slider(
+                            maximum=1000000,
+                            minimum=10000,
+                            value=DEFAULT_FACE_NUMBER,
+                            label="Target Face Number",
+                            info="Adjust mesh complexity"
+                        )
+                        gen_button = gr.Button("Generate 3D Model", variant="primary")
+                        gen_texture_button = gr.Button("Apply Texture", variant="secondary", interactive=False)
+
+                with gr.Column(scale=1):
+                    model_output = gr.Model3D(label="3D Model Preview", interactive=False, height=400)
+                    textured_model_output = gr.Model3D(label="Textured 3D Model", interactive=False, height=400)
+                    download_button = gr.Button("Download GLB", variant="secondary")
+
+        with gr.Tab("Gallery & Community"):
+            gr.Markdown("### Explore Creations")
+            gr.Examples(
+                examples=[
+                    f"{TRIPOSG_CODE_DIR}/assets/example_data/{image}"
+                    for image in os.listdir(f"{TRIPOSG_CODE_DIR}/assets/example_data")
+                ],
+                fn=run_full,
+                inputs=[image_prompts],
+                outputs=[seg_image, model_output, textured_model_output],
+                cache_examples=True,
+            )
+            gr.Markdown("Join our [PolyGenixAI Community](https://www.anvilinteractive.com/community) to share your creations and get inspired!")
 
     gen_button.click(
         run_segmentation,