Upload app.py

app.py

@@ -386,15 +386,15 @@ def outpaint_image(image, direction, extend_percent, custom_prompt, progress=gr.
 
 with gr.Blocks(
                 title="CanvasAI",
-                theme= gr.themes.Glass(),
-                js="""
-                function() {
-                    document.body.classList.add('dark');
-                }
-                """) as demo:
-
-    gr.Markdown("""# CanvasAI
-    ### AI-powered image enhancement, colorization, and outpainting""")
+                theme= gr.themes.Soft(
+                    primary_hue=gr.themes.colors.violet,
+                    secondary_hue=gr.themes.colors.purple,
+                    neutral_hue=gr.themes.colors.slate,
+                    radius_size=gr.themes.sizes.radius_sm,
+                )) as demo:
+
+    gr.Markdown(""" # CanvasAI
+    * AI-powered image enhancement, colorization, and outpainting *""")
     gr.Markdown(
         "**Enhance** with Real-ESRGAN  |  "
         "**Colorize** B&W photos with DDColor  |  "
@@ -405,7 +405,7 @@ with gr.Blocks(
     with gr.Tabs():
         #---------Enhance----------------
         with gr.Tab(" ✨ Enhance"):
-            gr.Markdown("### Upscale and sharpen any image 2x or 4x using Real-ESRGAN.")
+            gr.Markdown("## Upscale and sharpen any image 2x or 4x using Real-ESRGAN.")
             with gr.Row(equal_height= True):
                 with gr.Column(scale= 1):
                     enh_input = gr.Image(label="Upload Image", type="pil", height= 350)
@@ -417,14 +417,14 @@ with gr.Blocks(
                     )
                     enh_btn = gr.Button("Enhance", variant="primary", size="lg")
                 with gr.Column(scale=1):
-                    enh_output = gr.Image(label="Result", type="pil", interactive=False, height=350, show_download_button=True,)
+                    enh_output = gr.Image(label="Result", type="pil", interactive=False, height=350,)
                     enh_info   = gr.Textbox(label="Size Info", interactive=False, container=True,)
 
             enh_btn.click(fn=enhance_image, inputs=[enh_input, enh_scale], outputs=[enh_output, enh_info])
         #--------Colourize---------------------------
         with gr.Tab("Colorize"):
             gr.Markdown("""
-                ### B&W Photo Colorization
+                ## B&W Photo Colorization
                 DDColor adds natural, realistic colors while preserving the original structure."""
             )
             with gr.Row(equal_height= True):
@@ -433,18 +433,16 @@ with gr.Blocks(
                     col_strength = gr.Slider(
                         minimum=0.5, maximum=1.0, value=0.9, step=0.05,
                         label="Color Strength",
-                        # BUG 10 FIX: maximum changed from 0.7 to 1.0
                         info="0.5 = subtle tint  ·  1.0 = full vivid color"
                     )
                     col_btn = gr.Button("Colorize", variant="primary", size="lg")
                 with gr.Column():
-                    col_output = gr.Image(label="Colorized Result", type="pil", interactive=False, height=350,
-                        show_download_button=True,)
+                    col_output = gr.Image(label="Colorized Result", type="pil", interactive=False, height=350,)
 
             col_btn.click(fn=colour_image, inputs=[col_input, col_strength], outputs=[col_output])
         #----------Outpaint-----------------------------
         with gr.Tab("Outpaint"):
-            gr.Markdown("""### Extend Any Scene
+            gr.Markdown("""## Extend Any Scene
             Upload an image and extend it in any direction. 
             BLIP reads the scene automatically — no prompt needed."""
             )
@@ -468,7 +466,7 @@ with gr.Blocks(
                         )
                     out_btn = gr.Button("Outpaint", variant="primary", size="lg")
                 with gr.Column(scale=1):
-                    out_output  = gr.Image(label="Result", type="pil", interactive=False, height= 350, show_download_button= True)
+                    out_output  = gr.Image(label="Result", type="pil", interactive=False, height= 350,)
                     out_caption = gr.Textbox(label="Prompt Used", interactive=False, lines=4)
 
             out_btn.click(