Create app_test.py

app_test.py

@@ -0,0 +1,25 @@
+def casa_ai_run(image=None, text=None):
+
+    if image is None and text is None: 
+        print('Please provide context in form of image, audio, text')
+        return None
+    
+    result_image = controlnetdesign.generate_design(image, text)
+    return result_image
+
+import gradio as gr 
+
+title = "CasaAI Demo"
+description = "A simple Gradio interface to use CasaAI for virtual staging"
+
+demo = gr.Interface(casa_ai_run,
+                    inputs = [gr.Image(label="Empty room image", type='pil'),
+                              gr.Textbox(label="Prompt to design room"), 
+                              ],
+                    outputs = [
+                              gr.Image(label="Generated room image"),
+                              ],
+                    title = title,
+                    description = description,
+                    )
+demo.launch()