Update app.py

app.py

@@ -4,19 +4,32 @@ import numpy as np
 
 f = modal.Cls.lookup("casa-interior-hf-v3", "DesignModel")
 
-def casa_ai_run_tab1(image=None, room_type=None, interior_style=None, objects=None): 
+# def casa_ai_run_tab1(image=None, room_type=None, interior_style=None, objects=None): 
+    
+#     if image is None: 
+#         print('Please provide image of empty room to design')
+#         return None
+
+#     if (room_type is None or interior_style is None or objects is None): 
+#         print('Please select all three (Room type, Interior Style and Objects')
+#         return None
+
+#     objects_str = ", ".join(objects)
+#     text = f"A {room_type} in style of {interior_style} with {objects_str}."
+    
+#     result_image = f.inference.remote("tab1", image, text)
+#     return result_image
+
+def casa_ai_run_tab1(image=None, text=None): 
     
     if image is None: 
         print('Please provide image of empty room to design')
         return None
 
-    if (room_type is None or interior_style is None or objects is None): 
-        print('Please select all three (Room type, Interior Style and Objects')
+    if text is None: 
+        print('Please provide a text prompt')
         return None
 
-    objects_str = ", ".join(objects)
-    text = f"A {room_type} in style of {interior_style} with {objects_str}."
-    
     result_image = f.inference.remote("tab1", image, text)
     return result_image
 
@@ -59,21 +72,22 @@ with gr.Blocks() as casa:
             with gr.Column():
                 inputs = [
                             gr.Image(sources='upload', type="pil", label="Upload"), 
-                            gr.Dropdown(
-                                        label="Room type",
-                                        choices=["Living Room", "Bedroom", "Home Office", "Dining Room", "Kitchen", "Bathroom"],
-                                        multiselect=False
-                                    ), 
-                            gr.Dropdown(
-                                        label="Interior Style",
-                                        choices=["Modern", "Traditional", "Contemporary", "Minimalist", "Rustic", "Industrial", "Scandinavian"],
-                                        multiselect=False
-                                    ), 
-                            gr.Dropdown(
-                                        label="Objects",
-                                        choices=["Sofa", "Bed", "Table", "Chair", "Fridge", "Dinning table", "Coffee table"],
-                                        multiselect=True
-                                    ),
+                            gr.Textbox(label="Prompt for redesigning masked object")]
+                            # gr.Dropdown(
+                            #             label="Room type",
+                            #             choices=["Living Room", "Bedroom", "Home Office", "Dining Room", "Kitchen", "Bathroom"],
+                            #             multiselect=False
+                            #         ), 
+                            # gr.Dropdown(
+                            #             label="Interior Style",
+                            #             choices=["Modern", "Traditional", "Contemporary", "Minimalist", "Rustic", "Industrial", "Scandinavian"],
+                            #             multiselect=False
+                            #         ), 
+                            # gr.Dropdown(
+                            #             label="Objects",
+                            #             choices=["Sofa", "Bed", "Table", "Chair", "Fridge", "Dinning table", "Coffee table"],
+                            #             multiselect=True
+                            #         ),
                         ]
             with gr.Column():
                 outputs = [gr.Image(label="Generated room image")]