Update app.py

app.py

@@ -43,11 +43,11 @@ def get_css(bg_color, width, height):
     .gradio-container {{
         background-color: {bg_color} !important;
         border: none !important;
-        width: {width} !important;  /* Set the width */
-        height: {height} !important; /* Set the height */
+        width: {width}px !important;  /* Set the width */
+        height: {height}px !important; /* Set the height */
         max-width: 100%; /* Ensure it does not exceed the container's width */
         max-height: 100%; /* Ensure it does not exceed the container's height */
-        overflow: scroll; /* Prevent internal scrolling */
+        overflow: hidden; /* Prevent internal scrolling */
     }}
     footer {{display: none !important;}} /* Hide footer */
     """
@@ -55,24 +55,32 @@ def get_css(bg_color, width, height):
 def update_status(img):
     return inference(img), gr.update(value="Processing complete!")
 
-def create_interface(request):
-    query_params = parse_qs(urlparse(request.url).query)
-    bg_color = query_params.get('bg_color', ['rgb(17, 24, 39)'])[0]
-    width = int(query_params.get('width', ['auto'])[0])
-    height = int(query_params.get('height', ['100%'])[0])
+def create_interface():
+    # Default values
+    bg_color = 'rgb(17, 24, 39)'
+    width = 'auto'
+    height = '100%'
 
-    css = get_css(bg_color, width, height)
-    
-    with gr.Blocks(css=css) as demo:
-        with gr.Column():
-            image_input = gr.Image(type="numpy", label="Upload Image", height=height, width=width)
-            process_button = gr.Button("Process Image")
-            status = gr.Markdown(value="")
+    # Function to update the interface based on the URL parameters
+    def update_interface(request: gr.Request):
+        nonlocal bg_color, width, height
+        query_params = parse_qs(urlparse(request.url).query)
+        bg_color = query_params.get('bg_color', [bg_color])[0]
+        width = int(query_params.get('width', [width])[0])
+        height = int(query_params.get('height', [height])[0])
+        css = get_css(bg_color, width, height)
+        return gr.Blocks(css=css)
 
-        process_button.click(update_status, inputs=image_input, outputs=[image_input, status])
+    with gr.Blocks() as demo:
+        gr.Markdown("Loading...")  # Temporary placeholder
+        
+    demo.load(update_interface)
 
     return demo
 
-demo = gr.Interface(create_interface, inputs=[], outputs=[])
+demo = create_interface()
+
+with gr.Blocks() as outer_demo:
+    outer_demo.mount(demo)
 
 demo.launch()