Update app.py

app.py

@@ -268,27 +268,78 @@ def gradio_interface(input_img):
         print(f"Error processing image: {str(e)}")
         return [None, None]
 
-# Define interface
-iface = gr.Interface(
-    fn=gradio_interface,
-    inputs=gr.Image(label="Input Image"),
-    outputs=[
-        gr.Image(label="Raw Depth Map"),
-        gr.Image(label="Normal Map")
+# Define default values
+DEFAULT_VALUES = {
+    "prompt": "clean, texture, high-resolution, 8k",
+    "negative_prompt": "blurry, dotted, noise, raster lines, unclear, lowres, over-smoothed",
+    "guidance_scale": 2.5,
+    "conditioning_scale": 1.0,
+    "num_steps": 6,
+    "seed": None,
+    "upscale_factor": 4,
+    "color_fix_method": "adain"
+}
+
+# Create Gradio interface
+demo = gr.Interface(
+    fn=process_image,
+    inputs=[
+        gr.Image(label="Input Image"),
+        gr.Textbox(
+            label="Prompt", 
+            value=DEFAULT_VALUES["prompt"],
+            default=DEFAULT_VALUES["prompt"]
+        ),
+        gr.Textbox(
+            label="Negative Prompt", 
+            value=DEFAULT_VALUES["negative_prompt"],
+            default=DEFAULT_VALUES["negative_prompt"]
+        ),
+        gr.Slider(
+            minimum=1.0, 
+            maximum=20.0, 
+            value=DEFAULT_VALUES["guidance_scale"],
+            default=DEFAULT_VALUES["guidance_scale"],
+            label="Guidance Scale"
+        ),
+        gr.Slider(
+            minimum=0.1, 
+            maximum=2.0, 
+            value=DEFAULT_VALUES["conditioning_scale"],
+            default=DEFAULT_VALUES["conditioning_scale"],
+            label="Conditioning Scale"
+        ),
+        gr.Slider(
+            minimum=1, 
+            maximum=50, 
+            value=DEFAULT_VALUES["num_steps"],
+            default=DEFAULT_VALUES["num_steps"],
+            step=1, 
+            label="Number of Steps"
+        ),
+        gr.Number(
+            label="Seed", 
+            value=DEFAULT_VALUES["seed"],
+            default=DEFAULT_VALUES["seed"]
+        ),
+        gr.Slider(
+            minimum=1, 
+            maximum=4, 
+            value=DEFAULT_VALUES["upscale_factor"],
+            default=DEFAULT_VALUES["upscale_factor"],
+            step=1, 
+            label="Upscale Factor"
+        ),
+        gr.Radio(
+            choices=["none", "wavelet", "adain"], 
+            label="Color Fix Method", 
+            value=DEFAULT_VALUES["color_fix_method"],
+            default=DEFAULT_VALUES["color_fix_method"]
+        ),
     ],
-    title="Depth and Normal Map Generation",
-    description="Upload an image to generate its depth map and normal map.",
-    examples=[
-        "PavingStones144_1K-PNG_Color.png",
-        "0269B55506557D8D_diffuse.png",
-        "Brick_Floor_tkqlcekcw_8K_surface_msAlbedo_baked.jpg",
-        "Brick_Painted_sb0hkjp0_4K_surface_msAlbedo_baked.jpg",
-        "Concrete_rlvlbep0_4K_surface_msAlbedo_baked.jpg",
-        "Grass_Dried_scmkvwp0_4K_surface_msAlbedo_baked.jpg",
-        "Stone_Tile_uc2jdbpg_8K_surface_msAlbedo_baked.jpg",
-        "Surface_Tiles_smgmjog_8K_surface_msAlbedo_baked.jpg",
-        "Panel.jpg"
-    ]
+    outputs=gr.Image(label="Generated Image"),
+    title="Controllable Conditional Super-Resolution",
+    description="Upload an image to enhance its resolution using CCSR."
 )
 
 # Launch the app