Update app.py

app.py

@@ -432,21 +432,10 @@ edit_tool = gr.Interface(
 
 
 # ============================================================
-# MAIN APP WITH TABS
+# MAIN UI
 # ============================================================
 
-custom_css = """
-.gradio-container { font-size: 18px !important; }
-h1 { font-size: 2.5rem !important; }
-h2 { font-size: 1.8rem !important; }
-h3 { font-size: 1.4rem !important; }
-label, .label-wrap { font-size: 1.1rem !important; }
-textarea, input { font-size: 1.1rem !important; }
-button { font-size: 1.1rem !important; }
-.markdown-text { font-size: 1.1rem !important; }
-"""
-
-with gr.Blocks(title="Text to 3D | MCP Server", css=custom_css) as demo:
+with gr.Blocks() as main_ui:
     
     current_image_state = gr.State(None)
     edit_count_state = gr.State(0)
@@ -460,65 +449,46 @@ with gr.Blocks(title="Text to 3D | MCP Server", css=custom_css) as demo:
     ⏱️ *Generation takes 1-2 minutes. First run may take longer as the model warms up.*
     """)
     
-    with gr.Tabs():
-        with gr.Tab("🎨 Interactive UI"):
-            gr.Markdown("## 1️⃣ Generate Initial 3D Model")
-            
-            with gr.Row():
-                with gr.Column(scale=2):
-                    prompt_input = gr.Textbox(label="Text Prompt", placeholder="A plane with eagle wings", lines=2)
-                with gr.Column(scale=1):
-                    generate_btn = gr.Button("🚀 Generate", variant="primary", size="lg")
-            
-            gr.Examples(
-                examples=["A plane with eagle wings", "A wooden chair", "A red sports car", "A ceramic coffee mug", "A robot dog"],
-                inputs=prompt_input
-            )
-            
-            gr.Markdown("## 2️⃣ Edit Your Model")
-            
-            with gr.Row():
-                with gr.Column(scale=2):
-                    edit_input = gr.Textbox(label="Edit Prompt", placeholder="Remove the wings", lines=2)
-                with gr.Column(scale=1):
-                    edit_btn = gr.Button("✏️ Apply Edit", variant="secondary", size="lg")
-                    edit_counter = gr.Markdown("*No edits yet*")
-            
-            gr.Examples(
-                examples=["Remove the wings", "Change color to blue", "Add racing stripes", "Make it larger", "Add wheels"],
-                inputs=edit_input
-            )
-            
-            gr.Markdown("## 📸 Images")
-            with gr.Row():
-                original_output = gr.Image(label="1. Original", type="filepath")
-                nobg_output = gr.Image(label="2. Transparent", type="filepath")
-                mask_output = gr.Image(label="3. Mask", type="filepath")
-            
-            gr.Markdown("## 🎮 3D Model")
-            model_output = gr.Model3D(label="Interactive 3D Model (drag to rotate)", clear_color=[0.1, 0.1, 0.1, 1.0])
-            
-            gr.Markdown("## 📥 Downloads")
-            with gr.Row():
-                glb_download = gr.File(label="GLB (mesh)")
-                ply_download = gr.File(label="PLY (splat)")
-        
-        with gr.Tab("🔧 Generate Tool (MCP)"):
-            gr.Markdown("### MCP Tool: `generate_3d`")
-            gr.Markdown("This tool is exposed to MCP clients like Claude Desktop and Cursor.")
-            mcp_prompt = gr.Textbox(label="Prompt", placeholder="A red sports car")
-            mcp_generate_btn = gr.Button("Generate", variant="primary")
-            mcp_generate_output = gr.Textbox(label="Result (JSON)", lines=10)
-            mcp_generate_btn.click(fn=generate_3d_model, inputs=[mcp_prompt], outputs=[mcp_generate_output], api_name="generate_3d")
-        
-        with gr.Tab("✏️ Edit Tool (MCP)"):
-            gr.Markdown("### MCP Tool: `edit_3d`")
-            gr.Markdown("This tool is exposed to MCP clients like Claude Desktop and Cursor.")
-            mcp_edit_prompt = gr.Textbox(label="Edit Prompt", placeholder="Remove the wings")
-            mcp_image_path = gr.Textbox(label="Transparent Image Path", placeholder="/path/to/transparent.png")
-            mcp_edit_btn = gr.Button("Edit", variant="primary")
-            mcp_edit_output = gr.Textbox(label="Result (JSON)", lines=10)
-            mcp_edit_btn.click(fn=edit_3d_model, inputs=[mcp_edit_prompt, mcp_image_path], outputs=[mcp_edit_output], api_name="edit_3d")
+    gr.Markdown("## 1️⃣ Generate Initial 3D Model")
+    
+    with gr.Row():
+        with gr.Column(scale=2):
+            prompt_input = gr.Textbox(label="Text Prompt", placeholder="A plane with eagle wings", lines=2)
+        with gr.Column(scale=1):
+            generate_btn = gr.Button("🚀 Generate", variant="primary", size="lg")
+    
+    gr.Examples(
+        examples=["A plane with eagle wings", "A wooden chair", "A red sports car", "A ceramic coffee mug", "A robot dog"],
+        inputs=prompt_input
+    )
+    
+    gr.Markdown("## 2️⃣ Edit Your Model")
+    
+    with gr.Row():
+        with gr.Column(scale=2):
+            edit_input = gr.Textbox(label="Edit Prompt", placeholder="Remove the wings", lines=2)
+        with gr.Column(scale=1):
+            edit_btn = gr.Button("✏️ Apply Edit", variant="secondary", size="lg")
+            edit_counter = gr.Markdown("*No edits yet*")
+    
+    gr.Examples(
+        examples=["Remove the wings", "Change color to blue", "Add racing stripes", "Make it larger", "Add wheels"],
+        inputs=edit_input
+    )
+    
+    gr.Markdown("## 📸 Images")
+    with gr.Row():
+        original_output = gr.Image(label="1. Original", type="filepath")
+        nobg_output = gr.Image(label="2. Transparent", type="filepath")
+        mask_output = gr.Image(label="3. Mask", type="filepath")
+    
+    gr.Markdown("## 🎮 3D Model")
+    model_output = gr.Model3D(label="Interactive 3D Model (drag to rotate)", clear_color=[0.1, 0.1, 0.1, 1.0])
+    
+    gr.Markdown("## 📥 Downloads")
+    with gr.Row():
+        glb_download = gr.File(label="GLB (mesh)")
+        ply_download = gr.File(label="PLY (splat)")
     
     gr.Markdown("""
     ---
@@ -547,5 +517,16 @@ with gr.Blocks(title="Text to 3D | MCP Server", css=custom_css) as demo:
         outputs=[nobg_output, mask_output, model_output, glb_download, ply_download, current_image_state, edit_count_state]
     ).then(fn=update_counter, inputs=[edit_count_state], outputs=[edit_counter])
 
+
+# ============================================================
+# COMBINE UI + MCP TOOLS
+# ============================================================
+
+demo = gr.TabbedInterface(
+    interface_list=[main_ui, generate_tool, edit_tool],
+    tab_names=["🎨 Interactive UI", "🔧 Generate Tool", "✏️ Edit Tool"],
+    title="Text to 3D | MCP Server"
+)
+
 if __name__ == "__main__":
     demo.launch(mcp_server=True)