Update app.py

app.py

@@ -185,19 +185,75 @@ def process_image(
         return None
 
 
+# Create Gradio interface
+# 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="clean, texture, high-resolution, 8k"),
-        gr.Textbox(label="Negative Prompt", value="blurry, dotted, noise, raster lines, unclear, lowres, over-smoothed"),
-        gr.Slider(minimum=1.0, maximum=20.0, value=2.5, label="Guidance Scale"),
-        gr.Slider(minimum=0.1, maximum=2.0, value=1.0, label="Conditioning Scale"),
-        gr.Slider(minimum=1, maximum=50, value=6, step=1, label="Number of Steps"),
-        gr.Number(label="Seed", value=None),
-        gr.Slider(minimum=1, maximum=4, value=4, step=1, label="Upscale Factor"),
-        gr.Radio(["none", "wavelet", "adain"], label="Color Fix Method", value="adain"),
+        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"]
+        ),
     ],
     outputs=gr.Image(label="Generated Image"),
     title="Controllable Conditional Super-Resolution",