Add GLM-Image Editor tab

app.py

@@ -227,6 +227,40 @@ def generate_3d_model(image: Image.Image, engine_choice: str = "Trellis (Texture
          debug_logs.append("💀 All engines failed.")
          return None, None, "\n".join(debug_logs)
 
+# --- GLM-IMAGE EDITING ---
+def edit_image_with_glm(image: Image.Image, prompt: str, strength: float = 0.8) -> Image.Image:
+    """Edit image using zai-org/GLM-Image via HF Inference"""
+    from huggingface_hub import InferenceClient
+    
+    if image is None:
+        raise gr.Error("Please upload an image first")
+    if not prompt:
+        raise gr.Error("Please enter a prompt")
+        
+    # Initialize client (uses HF_TOKEN env var automatically in Space)
+    client = InferenceClient("zai-org/GLM-Image")
+    
+    # Save temp image for API
+    with tempfile.NamedTemporaryFile(suffix=".png", delete=False) as f:
+        image.save(f.name)
+        temp_path = f.name
+        
+    try:
+        # GLM-Image specifically for Image-to-Image text editing
+        # We use the standard image_to_image task
+        output_image = client.image_to_image(
+            image=image,
+            prompt=prompt,
+            strength=strength, 
+            guidance_scale=7.5
+        )
+        return output_image
+    except Exception as e:
+        raise gr.Error(f"GLM Generation Failed: {str(e)}")
+    finally:
+        if os.path.exists(temp_path):
+            os.unlink(temp_path)
+
 # --- UI ---
 with gr.Blocks(title="DCR Vehicle Studio", theme=gr.themes.Monochrome()) as demo:
     gr.Markdown("# 🚗 DCR Vehicle Studio")
@@ -256,6 +290,26 @@ with gr.Blocks(title="DCR Vehicle Studio", theme=gr.themes.Monochrome()) as demo
                 outputs=[out_no_bg, out_with_bg, out_upscaled]
             )
         
+        with gr.Tab("✨ GLM Image Editor"):
+             gr.Markdown("### Experimental: Z.AI GLM-Image Editor")
+             gr.Markdown("Use `zai-org/GLM-Image` to modify your vehicle photos.")
+             
+             with gr.Row():
+                with gr.Column():
+                    glm_input = gr.Image(type="pil", label="Input Image")
+                    glm_prompt = gr.Textbox(label="Editing Prompt", placeholder="e.g. Change the car color to red, add snow on the ground")
+                    glm_strength = gr.Slider(minimum=0.1, maximum=1.0, value=0.8, label="Transformation Strength (0.1 = subtle, 1.0 = heavy)")
+                    glm_btn = gr.Button("✨ Re-Imagine", variant="primary")
+                
+                with gr.Column():
+                    glm_output = gr.Image(label="GLM Result")
+            
+             glm_btn.click(
+                fn=edit_image_with_glm,
+                inputs=[glm_input, glm_prompt, glm_strength],
+                outputs=[glm_output]
+             )
+
         with gr.Tab("🎮 3D Generation"):
             gr.Markdown("### Generate 3D Model")
             gr.Markdown("Choose your engine. **Trellis** provides textures but is experimental. **Hunyuan** is geometry only.")

requirements.txt

@@ -10,3 +10,4 @@ git+https://github.com/Tencent/Hunyuan3D-2.git
 diffusers
 accelerate
 trimesh
+huggingface_hub