Update app.py

app.py

@@ -4,19 +4,38 @@ import gradio as gr
 
 f = modal.Cls.lookup("casa-interior-hf-v2", "DesignModel")
 
+def casa_ai_run_tab1(image=None, prompt=None): 
+    
+    if image is None and text is None: 
+        print('Please provide context in form of image, text')
+        return None
 
-def casa_ai_run(dict=None, text=None):
+    result_image = f.inference.remote("tab1", image, text)
+    return result_image
 
+def casa_ai_run_tab2(dict=None, text=None):
+    
     image = dict["background"].convert("RGB")
     mask = dict["layers"][0].convert('L')
     
     if image is None and text is None: 
-        print('Please provide context in form of image, audio, text')
+        print('Please provide context in form of image, text')
         return None
     
-    result_image = f.inference.remote(image, text, mask)
+    result_image = f.inference.remote("tab2", image, text, mask)
     return result_image
 
+def casa_ai_run_tab3(dict=None, text=None):
+    
+    image = dict["background"].convert("RGB")
+    mask = dict["layers"][0].convert('L')
+    
+    if image is None and text is None: 
+        print('Please provide context in form of image, text')
+        return None
+    
+    result_image = f.inference.remote("tab3", image, text, mask)
+    return result_image
 
 with gr.Blocks() as casa:
     title = "Casa-AI Demo"
@@ -26,13 +45,13 @@ with gr.Blocks() as casa:
         with gr.Row():
             with gr.Column():
                 inputs = [
-                            gr.ImageEditor(sources='upload', brush=gr.Brush(colors=["#FFFFFF"]), elem_id="image_upload", type="pil", label="Upload", layers=False, eraser=True, transforms=[]),
+                            gr.Image(sources='upload', type="pil", label="Upload"),
                             gr.Textbox(label="Prompt to design room")]
             with gr.Column():
                 outputs = [gr.Image(label="Generated room image")]
                 
         submit_btn = gr.Button("Submit")
-        submit_btn.click(casa_ai_run, inputs=inputs, outputs=outputs)
+        submit_btn.click(casa_ai_run_tab1, inputs=inputs, outputs=outputs)
 
         
     with gr.Tab("ReDesign"):