Update app.py

app.py

@@ -1,80 +1,39 @@
 import gradio as gr
-import torch
-from diffusers import StableDiffusionImg2ImgPipeline
+import os
+from huggingface_hub import InferenceClient
 from PIL import Image
+import tempfile
 
-MODEL_ID = "runwayml/stable-diffusion-v1-5"
-
-# ✅ FORCE CPU
-device = "cpu"
-dtype = torch.float32
-
-pipe = StableDiffusionImg2ImgPipeline.from_pretrained(
-    MODEL_ID,
-    torch_dtype=dtype
+# Create HF client
+client = InferenceClient(
+    model="stabilityai/stable-diffusion-xl-base-1.0",
+    token=os.environ["HF_TOKEN"]
 )
 
-pipe = pipe.to(device)
-
-# Disable safety checker safely
-pipe.safety_checker = lambda images, **kwargs: (images, [False] * len(images))
-
-
-def redesign_room(image, style, colors, lighting, strength):
-    prompt = f"""
-Redesign the interior of this room while keeping the same layout,
-walls, windows, doors, and camera angle.
+def redesign_room(image, prompt):
+    # Save uploaded image temporarily
+    with tempfile.NamedTemporaryFile(suffix=".png", delete=False) as tmp:
+        image.save(tmp.name)
+        image_path = tmp.name
 
-Interior style: {style}
-Color palette: {colors}
-Lighting: {lighting}
-
-Photorealistic interior design, realistic shadows,
-interior photography, high detail.
-"""
-
-    image = image.convert("RGB").resize((512, 512))
-
-    result = pipe(
+    # Call HF API (NO local inference)
+    output_image = client.image_to_image(
+        image=image_path,
         prompt=prompt,
-        image=image,
-        strength=strength,
-        guidance_scale=7.5,
-        num_inference_steps=25
-    ).images[0]
-
-    return result
-
-
-with gr.Blocks(title="Room Redesign AI (CPU)") as demo:
-    gr.Markdown("## 🏠 AI Room Redesign (CPU Mode)")
-
-    image_input = gr.Image(label="Upload Room Image", type="pil")
-
-    style = gr.Dropdown(
-        ["Modern", "Luxury", "Scandinavian", "Minimal", "Japanese"],
-        value="Modern",
-        label="Interior Style"
+        guidance_scale=7,
+        num_inference_steps=30
     )
 
-    colors = gr.Textbox(value="white, beige, wood", label="Color Palette")
-    lighting = gr.Textbox(value="warm ambient lighting", label="Lighting")
+    return output_image
 
-    strength = gr.Slider(
-        minimum=0.2,
-        maximum=0.6,
-        value=0.35,
-        step=0.05,
-        label="Redesign Strength (lower = preserve structure)"
-    )
-
-    output = gr.Image(label="Redesigned Room")
-
-    btn = gr.Button("Redesign Room")
-    btn.click(
-        redesign_room,
-        inputs=[image_input, style, colors, lighting, strength],
-        outputs=output
-    )
 
-demo.launch()
+gr.Interface(
+    fn=redesign_room,
+    inputs=[
+        gr.Image(type="pil", label="Upload Room Image"),
+        gr.Textbox(label="Design Prompt", placeholder="Modern Scandinavian interior, warm lighting...")
+    ],
+    outputs=gr.Image(label="Redesigned Room"),
+    title="AI Room Redesign (No Local Model)",
+    description="Upload a room image and redesign it using prompts"
+).launch()