Update app.py

app.py

@@ -23,13 +23,16 @@ with gr.Blocks() as casa:
     description = "A Gradio interface to use CasaAI for virtual staging"
 
     with gr.Tab("ReImagine"):
-        inputs = [
-                    gr.ImageEditor(sources='upload', brush=gr.Brush(colors=["#FFFFFF"]), elem_id="image_upload", type="pil", label="Upload", layers=False, eraser=True, transforms=[]),
-                    gr.Textbox(label="Prompt to design room")]
-        
-        outputs = [gr.Image(label="Generated room image")]
-        
-        submit_btn = gr.Button("Submit")
+        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.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)
 
         
     with gr.Tab("ReDesign"):
@@ -38,7 +41,7 @@ with gr.Blocks() as casa:
     with gr.Tab("Recommendation"):
         pass
 
-    submit_btn.click(casa_ai_run, inputs=inputs, outputs=outputs)
+    
 
 
 # demo = gr.Interface(casa_ai_run,