Update app.py

app.py

@@ -4,23 +4,20 @@ import torch
 from PIL import Image
 from transformers import AutoProcessor
 from longcat_image.models import LongCatImageTransformer2DModel
-from longcat_image.pipelines import LongCatImageEditPipeline, LongCatImagePipeline
+from longcat_image.pipelines import LongCatImagePipeline, LongCatImageEditPipeline
 import numpy as np
 
-# 1. Define the Custom Theme Class
-# Inherit from Base for simplicity and modify core parameters.
-class AppleSoft(gr.themes.Base):
+# 1. DEFINE THE CUSTOM THEME CLASS (Programmatic Theming, No Custom CSS)
+# Inherit from Base and use .set() to achieve the Apple aesthetic.
+class AppleStyleTheme(gr.themes.Base):
     def __init__(
         self,
-        # Use a deep Indigo/Blue for primary actions (like the iOS/macOS blue)
-        primary_hue=gr.themes.colors.indigo, 
+        # Core Parameters for Apple Aesthetic
+        primary_hue=gr.themes.colors.blue,
         secondary_hue=gr.themes.colors.gray,
         neutral_hue=gr.themes.colors.neutral,
-        # Increase radius for "card" look (iOS uses large rounded corners)
-        radius_size=gr.themes.sizes.radius_lg,
-        # Set font to use a system font stack (like the old custom CSS)
-        font=[gr.themes.GoogleFont("Source Sans Pro"), "system-ui", "sans-serif"],
-        font_mono=gr.themes.GoogleFont("IBM Plex Mono"),
+        radius_size=gr.themes.sizes.radius_lg, # Large radius for modern rounded look
+        font=["system-ui", "sans-serif"], # Use system fonts for native feel
         spacing_size=gr.themes.sizes.spacing_lg,
         text_size=gr.themes.sizes.text_md,
         **kwargs
@@ -31,29 +28,33 @@ class AppleSoft(gr.themes.Base):
             neutral_hue=neutral_hue,
             radius_size=radius_size,
             font=font,
-            font_mono=font_mono,
             spacing_size=spacing_size,
             text_size=text_size,
             **kwargs
         )
-        # 2. Use .set() to override specific CSS variables for the Apple aesthetic
+        # Use .set() to fine-tune variables beyond the core 8
         self.set(
-            # Remove main block shadow for a flatter, modern look
-            block_shadow="none",
-            block_label_padding="*spacing_sm *spacing_lg",
-            # Apply light shadow and clean background to input/panel elements (the "card" effect)
+            # General clean white/light gray background
+            body_background_fill="*neutral_50", 
             block_background_fill="white",
-            block_border_width="1px",
+            # Remove default shadows for a flatter, cleaner look
+            block_shadow="none",
+            # Add a subtle border/ring for 'card' definition
             block_border_color="*neutral_200",
-            # Ensure primary button looks clean and elevated
+            block_border_width="1px",
+            # Customize button look (Primary blue is defined by primary_hue=blue)
             button_primary_background_fill="*primary_500",
             button_primary_background_fill_hover="*primary_400",
             button_primary_border_color="*primary_500",
-            button_primary_shadow="0 2px 8px *primary_300", # Subtle shadow for depth
+            # Tabs (Segmented control look is largely handled by rounded corners and soft theme)
+            tabs_border_color="*neutral_200",
+            tab_border_color="*neutral_100",
+            tab_border_width="0",
         )
 
 # Instantiate the custom theme
-apple_theme = AppleSoft()
+apple_theme = AppleStyleTheme()
+# --- End Theme Definition ---
 
 # --- Model Loading (Kept for completeness) ---
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
@@ -61,7 +62,6 @@ device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 # Text-to-Image Model
 t2i_model_id = 'meituan-longcat/LongCat-Image'
 print(f"🔄 Loading Text-to-Image model from {t2i_model_id}...")
-# ... (rest of model loading remains the same) ...
 t2i_text_processor = AutoProcessor.from_pretrained(
     t2i_model_id,
     subfolder='tokenizer'
@@ -86,7 +86,6 @@ print(f"✅ Text-to-Image model loaded successfully")
 # Image Edit Model
 edit_model_id = 'meituan-longcat/LongCat-Image-Edit'
 print(f"🔄 Loading Image Edit model from {edit_model_id}...")
-# ... (rest of model loading remains the same) ...
 edit_text_processor = AutoProcessor.from_pretrained(
     edit_model_id,
     subfolder='tokenizer'
@@ -188,7 +187,7 @@ t2i_example_prompts = [
 ]
 
 # Build Gradio interface
-# Passed theme=apple_theme here and fill_width=True for responsiveness
+# Passed theme=apple_theme and fill_width=True
 with gr.Blocks(theme=apple_theme, fill_width=True) as demo:
     # Retaining the HTML header for branding/title
     gr.HTML("""
@@ -207,7 +206,7 @@ with gr.Blocks(theme=apple_theme, fill_width=True) as demo:
         with gr.TabItem("Edit Image", id=0):
             with gr.Row(): 
                 # Left Column (Inputs)
-                with gr.Column(scale=1, min_width=0, variant="panel"): # variant="panel" applies the card styling from theme
+                with gr.Column(scale=1, min_width=0, variant="panel"): 
                     gr.Markdown("### 🖼️ Input Image & Controls")
                     input_image = gr.Image(
                         label="Upload Image",
@@ -220,6 +219,7 @@ with gr.Blocks(theme=apple_theme, fill_width=True) as demo:
                         label="What would you like to change?",
                         placeholder="e.g., Add a mustache, Change to sunset, Make it vintage...",
                         lines=2,
+                        info="Describe the edit you want to make to the uploaded image.",
                         max_lines=3
                     )
 
@@ -229,7 +229,7 @@ with gr.Blocks(theme=apple_theme, fill_width=True) as demo:
                         value=42,
                         step=1,
                         label="Seed",
-                        visible=False
+                        visible=True # Make visible for control
                     )
 
                     edit_btn = gr.Button("Edit Image", variant="primary", size="lg") 
@@ -290,7 +290,7 @@ with gr.Blocks(theme=apple_theme, fill_width=True) as demo:
                         value=42,
                         step=1,
                         label="Seed",
-                        visible=False
+                        visible=True # Make visible for control
                     )
 
                     generate_btn = gr.Button("Generate Image", variant="primary", size="lg") 
@@ -339,10 +339,9 @@ with gr.Blocks(theme=apple_theme, fill_width=True) as demo:
         </div>
     """)
 
-# Launch the app with the custom, programmatic theme
+# Launch the app with the custom programmatic theme
 if __name__ == "__main__":
     demo.launch(
         mcp_server=True,
-        theme=apple_theme, 
-        # Custom CSS is removed to ensure theme integrity
+        theme=apple_theme,
     )