Update app.py

app.py

@@ -271,11 +271,33 @@ def start_session(request: gr.Request):
     return request.session_hash
     
 def build_demo():
-    css = """
+            css = """
         #col-container {
             margin: 0 auto;
             max-width: 1560px;
         }
+        
+        .generate-btn {
+            background: linear-gradient(45deg, #22c55e, #16a34a) !important;
+            border: none !important;
+            color: white !important;
+            font-weight: bold !important;
+            transition: all 0.3s ease !important;
+        }
+        
+        .generate-btn:hover {
+            background: linear-gradient(45deg, #16a34a, #15803d) !important;
+            transform: translateY(-2px) !important;
+            box-shadow: 0 8px 25px rgba(34, 197, 94, 0.3) !important;
+        }
+        
+        .left-panel {
+            display: flex;
+            flex-direction: column;
+            justify-content: center;
+            height: 100%;
+            min-height: 500px;
+        }
         """
     theme = gr.themes.Ocean()
     
@@ -284,22 +306,9 @@ def build_demo():
         demo.load(start_session, outputs=[session_state])
 
         with gr.Column(elem_id="col-container"):
-            gr.HTML(
-                """
-                <div style="text-align: center;">
-                    <a href="https://github.com/wgsxm/PartCrafter" style="display: inline-block; vertical-align: middle; margin-left: 0.5em;">
-                        <img src="https://img.shields.io/badge/GitHub-Repo-blue" alt="GitHub Repo">
-                    </a>
-                </div>
-                <div style="text-align: center;">
-                    HF Space by :<a href="https://twitter.com/alexandernasa/" style="display: inline-block; vertical-align: middle; margin-left: 0.5em;">
-                        <img src="https://img.shields.io/twitter/url/https/twitter.com/cloudposse.svg?style=social&label=Follow Me" alt="GitHub Repo">
-                    </a>
-                </div>
-                """
-            )
+
             with gr.Row():
-                with gr.Column(scale=1):
+                with gr.Column(scale=1, elem_classes=["left-panel"]):
                     
                     input_image = gr.Image(type="filepath", label="Input Image", height=256)
                     num_parts = gr.Slider(1, MAX_NUM_PARTS, value=4, step=1, label="Number of Parts")
@@ -312,7 +321,7 @@ def build_demo():
                         flash_decoder = gr.Checkbox(value=False, label="Use Flash Decoder")
                         remove_bg = gr.Checkbox(value=True, label="Remove Background (RMBG)")
                     
-                    run_button = gr.Button("🚀 Generate 3D Parts")
+                    run_button = gr.Button("🚀 Generate 3D Parts", variant="primary", elem_classes=["generate-btn"])
 
                 with gr.Column(scale=2):
                     output_model = gr.Model3D(label="Merged 3D Object", height=512, interactive=False)