Add custom prompt field for user input

app.py

@@ -34,6 +34,7 @@ def enhance_image_simple(
     input_image: Image.Image,
     strength: float,
     guidance_scale: float,
+    custom_prompt: str,
     use_upscaler: bool,
     use_postprocess: bool,
     progress=gr.Progress()
@@ -53,6 +54,13 @@ def enhance_image_simple(
             logger.error(f"Model initialization failed: {e}", exc_info=True)
             raise ValueError(f"Failed to load AI models: {str(e)}")
         
+        # Build prompt
+        base_prompt = "professional architectural photography, highly detailed, 8k, photorealistic"
+        if custom_prompt and custom_prompt.strip():
+            final_prompt = f"{base_prompt}, {custom_prompt.strip()}"
+        else:
+            final_prompt = base_prompt
+        
         # Step 1: AI Enhancement
         progress(0.3, desc="Applying AI enhancement (this may take several minutes on CPU)...")
         try:
@@ -64,7 +72,7 @@ def enhance_image_simple(
             
             enhanced = pipeline_manager.enhance(
                 image=input_image,
-                prompt="professional architectural photography, highly detailed, 8k, photorealistic",
+                prompt=final_prompt,
                 negative_prompt="blurry, low quality, distorted, ugly, bad architecture",
                 strength=strength,
                 guidance_scale=guidance_scale,
@@ -113,9 +121,15 @@ iface = gr.Interface(
     fn=enhance_image_simple,
     inputs=[
         gr.Image(label="Input Image", type="pil"),
-        gr.Slider(0.1, 0.8, value=0.3, step=0.05, label="Strength"),
-        gr.Slider(1.0, 15.0, value=5.5, step=0.5, label="Guidance Scale"),
-        gr.Checkbox(label="Enable Upscaling", value=True),
+        gr.Slider(0.1, 0.8, value=0.3, step=0.05, label="Strength", info="Lower = more faithful to input, Higher = more creative"),
+        gr.Slider(1.0, 15.0, value=5.5, step=0.5, label="Guidance Scale", info="How closely to follow the prompt"),
+        gr.Textbox(
+            label="Additional Prompt (Optional)", 
+            placeholder="e.g., modern minimalist, glass facade, sunset lighting...",
+            lines=2,
+            info="Add custom details to enhance specific aspects"
+        ),
+        gr.Checkbox(label="Enable Upscaling (2x)", value=True),
         gr.Checkbox(label="Enable Post-Processing", value=True)
     ],
     outputs=gr.Image(label="Enhanced Image", type="pil"),