Update app.py

app.py

@@ -8,10 +8,8 @@ from PIL import Image
 from diffusers import StableDiffusionXLPipeline
 import open_clip
 from huggingface_hub import hf_hub_download
-from IP_Adapter.ip_adapter import IPAdapterXL
-from perform_swap import compute_dataset_embeds_svd, get_modified_images_embeds_composition
-import tempfile
-import uuid
+from IP_Composer.IP_Adapter.ip_adapter import IPAdapterXL
+from IP_Composer.perform_swap import compute_dataset_embeds_svd, get_modified_images_embeds_composition
 
 device = "cuda" if torch.cuda.is_available() else "cpu"
 
@@ -41,14 +39,6 @@ def get_image_embeds(pil_image, model=clip_model, preproc=preprocess, dev=device
         embeds = model.encode_image(image.to(dev))
     return embeds.cpu().detach().numpy()
 
-def save_temp_image(image):
-    """Save a PIL image to a temporary file and return the path"""
-    temp_dir = tempfile.gettempdir()
-    filename = f"{uuid.uuid4()}.png"
-    filepath = os.path.join(temp_dir, filename)
-    image.save(filepath)
-    return filepath
-
 def process_images(
     base_image, 
     concept_image1, concept_desc1,
@@ -159,8 +149,8 @@ def process_and_display(
     return modified_images
 
 with gr.Blocks(title="Image Concept Composition") as demo:
-        gr.Markdown("# Image Concept Composition")
-        gr.Markdown("Upload a base image and 1-3 concept images to create new images that combine these concepts.")
+        gr.Markdown("# IP Composer")
+        gr.Markdown("")
         
         with gr.Row():
             with gr.Column():
@@ -193,11 +183,10 @@ with gr.Blocks(title="Image Concept Composition") as demo:
                     scale = gr.Slider(minimum=0.1, maximum=2.0, value=1.0, step=0.1, label="Scale")
                     seed = gr.Number(value=420, label="Seed", precision=0)
                 
-                submit_btn = gr.Button("Generate Image")
+                submit_btn = gr.Button("Generate")
             
             with gr.Column():
-                gallery = gr.Gallery(label="Generated Image", show_label=True)
-                status = gr.Markdown("Upload images and click Generate")
+                output_image = gr.Image(label="composed output", show_label=True)
         
         submit_btn.click(
             fn=process_and_display,
@@ -209,7 +198,7 @@ with gr.Blocks(title="Image Concept Composition") as demo:
                 rank1, rank2, rank3, 
                 prompt, scale, seed
             ],
-            outputs=[gallery, status]
+            outputs=[output_image]
         )