Update app.py

app.py

@@ -56,6 +56,18 @@ def casa_ai_run_tab3(dict=None):
 
 def casa_ai_run_tab_sketch(image=None, room_type=None, room_style=None):
 
+    if image is None: 
+        print('Please provide a sketch or ketchup image')
+        return None
+
+    if room_type is None: 
+        print('Please select a room type')
+        return None
+
+    if room_style is None: 
+        print('Please select a room style')
+        return None
+
     text = f"{room_type}, {room_style}"
     f = modal.Cls.lookup("casa-interior-hf-v6-sketch", "DesignModel")
     result_image = f.inference.remote(image, text)
@@ -85,6 +97,26 @@ with gr.Blocks() as casa:
                               ['example_images/image_2.jpg', 'Modern bedroom art deco style']], 
                     inputs=inputs, outputs=outputs, fn=casa_ai_run_tab1, cache_examples=True)
 
+    
+    with gr.Tab("Sketch Transform"):
+        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 of sketch transformed")]
+                
+        submit_btn = gr.Button("Transform!")
+        submit_btn.click(casa_ai_run_tab_sketch, inputs=inputs, outputs=outputs)
+        gr.Examples(examples=[['example_images/sketch01.jpeg 'Living Room',  'Modern'], 
+                              ['example_images/sketch02.jpeg', 'Bedroom', 'Minimalist'], 
+                              ['example_images/sketchup.jpeg', 'Kitchen', 'Modern']], 
+                    inputs=inputs, outputs=outputs, fn=casa_ai_run_tab_sketch, cache_examples=True)
+
         
     with gr.Tab("Redesign"):
         with gr.Row():
@@ -111,19 +143,4 @@ with gr.Blocks() as casa:
         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()