Update app.py

app.py

@@ -54,6 +54,14 @@ def casa_ai_run_tab3(dict=None):
 
     return results
 
+def casa_ai_run_tab_sketch(image=None, room_type=None, room_style=None):
+
+    text = f"{room_type}, {room_style}"
+    f = modal.Cls.lookup("casa-interior-hf-v6-sketch", "DesignModel")
+    result_image = f.inference.remote(image, text)
+    return result_image
+
+
 with gr.Blocks() as casa:
     title = "Casa-AI Demo"
     description = "A Gradio interface to use CasaAI for virtual staging"
@@ -102,4 +110,20 @@ with gr.Blocks() as casa:
         submit_btn = gr.Button("Find similar products!")
         submit_btn.click(casa_ai_run_tab3, inputs=inputs, outputs=outputs)
 
+
+    with gr.Tab("Re"):
+        with gr.Row():
+            with gr.Column():
+                inputs = [
+                            gr.Image(sources='upload', type="numpy", label="Upload"),
+                            gr.Dropdown(["Living Room", "Bedroom", "Kitchen"], label="Room Type", info="Select Room Type"),
+                            gr.Dropdown(["Modern", "Minimalist", "Scandinavian"], label="Style", info="Interior Style!"),
+                ]
+                    
+            with gr.Column():
+                outputs = [gr.Image(label="Image with new designed object")]
+                
+        submit_btn = gr.Button("Redesign!")
+        submit_btn.click(casa_ai_run_tab_sketch, inputs=inputs, outputs=outputs)
+
 casa.launch()