Update app.py

app.py

@@ -1,91 +1,4 @@
-import modal
-import gradio as gr 
-import numpy as np 
-from io import BytesIO
-import requests
+from subprocess import call
+import os
 
-f = modal.Cls.lookup("casa-interior-hf-v3", "DesignModel")
-
-import requests
-from io import BytesIO
-from google.cloud import vision
-from google.oauth2 import service_account
-import PIL 
-
-
-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 text is None: 
-        print('Please provide a text prompt')
-        return 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 np.sum(np.array(mask)) == 0: 
-        mask = None 
-        
-    if mask is None: 
-        print('Please provide a mask over the object you want to generate again.')
-        
-    if image is None and text is None: 
-        print('Please provide context in form of image, text')
-        return None
-    
-    result_image = f.inference.remote("tab2", image, text, mask)
-    return result_image
-
-
-with gr.Blocks() as casa:
-    title = "Casa-AI Demo"
-    description = "A Gradio interface to use CasaAI for virtual staging"
-
-    with gr.Tab("Reimagine"):
-        with gr.Row():
-            with gr.Column():
-                inputs = [
-                            gr.Image(sources='upload', type="pil", label="Upload"), 
-                            gr.Textbox(label="Room description.")
-                        ]
-            with gr.Column():
-                outputs = [gr.Image(label="Generated room image")]
-
-        
-        submit_btn = gr.Button("Generate!")
-        submit_btn.click(casa_ai_run_tab1, inputs=inputs, outputs=outputs)
-
-        
-    with gr.Tab("Redesign"):
-        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="Description for redesigning masked object")]
-            with gr.Column():
-                outputs = [gr.Image(label="Image with new designed object")]
-                
-        submit_btn = gr.Button("Redesign!")
-        submit_btn.click(casa_ai_run_tab2, inputs=inputs, outputs=outputs)
-
-    with gr.Tab("Recommendation"):
-        with gr.Row():
-            with gr.Column():
-                inputs = [
-                            gr.ImageEditor(sources='upload', elem_id="image_upload", type="numpy", label="Upload", layers=False, eraser=False, brush=False, transforms=['crop']),
-                            ]
-            with gr.Column():
-                outputs = [gr.Gallery(label="Similar products")]
-        pass
-        # submit_btn = gr.Button("Find similar products!")
-        # submit_btn.click(casa_ai_run_tab3, inputs=inputs, outputs=outputs)
-
-casa.launch()
+call(["python", os.getenv('APP_RUN_CASA')])