Commit
·
b492be8
1
Parent(s):
1eb08e4
Refactor Gradio interface by simplifying image input settings, streamlining background choice configuration, and enhancing output section layout for improved usability and clarity.
Browse files
app.py
CHANGED
@@ -215,29 +215,23 @@ with gr.Blocks() as demo:
|
|
215 |
with gr.Row():
|
216 |
image_input = gr.Image(
|
217 |
label="Image input",
|
218 |
-
image_mode="RGBA",
|
219 |
-
sources=["upload"],
|
220 |
type="pil"
|
221 |
)
|
222 |
processed_image = gr.Image(
|
223 |
label="Processed Image",
|
224 |
-
|
225 |
-
type="pil",
|
226 |
-
image_mode="RGB"
|
227 |
)
|
228 |
|
229 |
with gr.Row():
|
230 |
with gr.Column():
|
231 |
-
|
232 |
-
|
233 |
-
|
234 |
-
|
235 |
-
|
236 |
-
)
|
237 |
back_groud_color = gr.ColorPicker(
|
238 |
label="Background Color",
|
239 |
-
value="#7F7F7F"
|
240 |
-
interactive=False
|
241 |
)
|
242 |
foreground_ratio = gr.Slider(
|
243 |
label="Foreground Ratio",
|
@@ -271,34 +265,21 @@ with gr.Blocks() as demo:
|
|
271 |
|
272 |
gr.Examples(
|
273 |
examples=[[os.path.join("examples", i)] for i in os.listdir("examples")],
|
274 |
-
inputs=[image_input]
|
275 |
-
examples_per_page=20,
|
276 |
-
cache_examples=False
|
277 |
)
|
278 |
|
279 |
with gr.Column():
|
280 |
-
image_output = gr.Image(
|
281 |
-
|
282 |
-
|
283 |
-
)
|
284 |
-
xyz_output = gr.Image(
|
285 |
-
label="Output CCM image",
|
286 |
-
interactive=False
|
287 |
-
)
|
288 |
-
output_model = gr.Model3D(
|
289 |
-
label="Output GLB",
|
290 |
-
interactive=False
|
291 |
-
)
|
292 |
gr.Markdown("Note: Ensure that the input image is correctly pre-processed into a grey background, otherwise the results will be unpredictable.")
|
293 |
|
294 |
-
# Define the processing chain
|
295 |
def process_and_generate(image, bg_choice, fg_ratio, bg_color, seed_val, guidance, steps):
|
296 |
if image is None:
|
297 |
raise gr.Error("No image uploaded!")
|
298 |
processed = preprocess_image(image, bg_choice, fg_ratio, bg_color)
|
299 |
return gen_image(processed, seed_val, guidance, steps)
|
300 |
|
301 |
-
# Connect the button click event with API endpoint
|
302 |
text_button.click(
|
303 |
fn=process_and_generate,
|
304 |
inputs=[
|
@@ -319,4 +300,4 @@ with gr.Blocks() as demo:
|
|
319 |
)
|
320 |
|
321 |
if __name__ == "__main__":
|
322 |
-
demo.queue().launch(
|
|
|
215 |
with gr.Row():
|
216 |
image_input = gr.Image(
|
217 |
label="Image input",
|
|
|
|
|
218 |
type="pil"
|
219 |
)
|
220 |
processed_image = gr.Image(
|
221 |
label="Processed Image",
|
222 |
+
type="pil"
|
|
|
|
|
223 |
)
|
224 |
|
225 |
with gr.Row():
|
226 |
with gr.Column():
|
227 |
+
background_choice = gr.Radio(
|
228 |
+
choices=["Alpha as mask", "Auto Remove background"],
|
229 |
+
value="Auto Remove background",
|
230 |
+
label="Background choice"
|
231 |
+
)
|
|
|
232 |
back_groud_color = gr.ColorPicker(
|
233 |
label="Background Color",
|
234 |
+
value="#7F7F7F"
|
|
|
235 |
)
|
236 |
foreground_ratio = gr.Slider(
|
237 |
label="Foreground Ratio",
|
|
|
265 |
|
266 |
gr.Examples(
|
267 |
examples=[[os.path.join("examples", i)] for i in os.listdir("examples")],
|
268 |
+
inputs=[image_input]
|
|
|
|
|
269 |
)
|
270 |
|
271 |
with gr.Column():
|
272 |
+
image_output = gr.Image(label="Output RGB image")
|
273 |
+
xyz_output = gr.Image(label="Output CCM image")
|
274 |
+
output_model = gr.Model3D(label="Output GLB")
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
275 |
gr.Markdown("Note: Ensure that the input image is correctly pre-processed into a grey background, otherwise the results will be unpredictable.")
|
276 |
|
|
|
277 |
def process_and_generate(image, bg_choice, fg_ratio, bg_color, seed_val, guidance, steps):
|
278 |
if image is None:
|
279 |
raise gr.Error("No image uploaded!")
|
280 |
processed = preprocess_image(image, bg_choice, fg_ratio, bg_color)
|
281 |
return gen_image(processed, seed_val, guidance, steps)
|
282 |
|
|
|
283 |
text_button.click(
|
284 |
fn=process_and_generate,
|
285 |
inputs=[
|
|
|
300 |
)
|
301 |
|
302 |
if __name__ == "__main__":
|
303 |
+
demo.queue().launch()
|