MrajaR commited on
Commit
54e2641
·
1 Parent(s): 01087aa

add description in gradio app

Browse files
Files changed (1) hide show
  1. app.py +50 -23
app.py CHANGED
@@ -511,11 +511,20 @@ class CreativeImageSuite:
511
 
512
  # Text-to-Image Section
513
  gr.Markdown("## Text to Image Feature")
 
 
 
 
 
 
 
 
 
514
  with gr.Row():
515
- prompt_input = gr.Textbox(label="Enter your prompt for image generation:")
516
- generate_button = gr.Button("Generate Image")
517
- output_image = gr.Image(label="Generated Image")
518
- download_button = gr.File(label="Download Generated Image", type="filepath")
519
 
520
  generate_button.click(self.inference.text_to_image,
521
  inputs=prompt_input,
@@ -523,11 +532,20 @@ class CreativeImageSuite:
523
 
524
  # Image-to-Image Section
525
  gr.Markdown("## Image to Image Feature")
 
 
 
 
 
 
 
 
 
526
  with gr.Row():
527
- input_image = gr.Image(label="Upload Image for Modification", type="pil")
528
- prompt_modification = gr.Textbox(label="Enter your prompt for modification:")
529
- modify_button = gr.Button("Modify Image")
530
- modified_output_image = gr.Image(label="Modified Image")
531
  download_modified_button = gr.File(label="Download Modified Image", type="filepath")
532
 
533
  modify_button.click(self.inference.text_image_to_image,
@@ -536,32 +554,41 @@ class CreativeImageSuite:
536
 
537
  # Background Removal and Resizing Section
538
  gr.Markdown("## Image Background Removal and Resizing with Optional Watermark")
 
 
 
 
 
 
 
 
 
539
  with gr.Row():
540
- input_files = gr.File(label="Upload Image or ZIP/RAR file", file_types=[".zip", ".rar", "image"], interactive=True)
541
- watermark = gr.File(label="Upload Watermark Image (Optional)", file_types=[".png"])
542
 
543
  with gr.Row():
544
- canvas_size = gr.Radio(choices=["Rox", "Columbia", "Zalora"], label="Canvas Size", value="Rox")
545
- output_format = gr.Radio(choices=["PNG", "JPG"], label="Output Format", value="JPG")
546
- num_workers = gr.Slider(minimum=1, maximum=16, step=1, label="Number of Workers", value=5)
547
 
548
  with gr.Row():
549
- bg_method = gr.Radio(choices=["bria", "rembg", None], label="Background Removal Method", value="bria")
550
- bg_choice = gr.Radio(choices=["transparent", "white", "custom"], label="Background Choice", value="white")
551
- custom_color = gr.ColorPicker(label="Custom Background Color", value="#ffffff", visible=False)
552
 
553
- process_button = gr.Button("Process Images")
554
  with gr.Row():
555
  gallery_processed = gr.Gallery(label="Processed Images")
556
  with gr.Row():
557
- image_original = gr.Image(label="Original Images", interactive=False)
558
- image_processed = gr.Image(label="Processed Images", interactive=False)
559
  with gr.Row():
560
- original_ratio = gr.Textbox(label="Original Ratio")
561
- processed_ratio = gr.Textbox(label="Processed Ratio")
562
  with gr.Row():
563
- output_zip = gr.File(label="Download Processed Images as ZIP")
564
- processing_time = gr.Textbox(label="Processing Time (seconds)")
565
 
566
  bg_choice.change(self.show_color_picker, inputs=bg_choice, outputs=custom_color)
567
  process_button.click(self.master_process, inputs=[input_files, bg_method, watermark, canvas_size, output_format, bg_choice, custom_color, num_workers],
 
511
 
512
  # Text-to-Image Section
513
  gr.Markdown("## Text to Image Feature")
514
+ gr.Markdown("""
515
+ *Create Visuals from Your Imagination*
516
+
517
+ This feature allows you to generate unique images from a simple text description. Just type your idea in the box, and our tool will bring it to life in seconds.
518
+ - Example Prompts:
519
+ - "Generate an image of a kitchenset with modern furniture."
520
+ - "Generate an image of a casual turtleneck shirt."
521
+ - "A futuristic laptop on a white background ."
522
+ """)
523
  with gr.Row():
524
+ prompt_input = gr.Textbox(label="Describe your vision to generate a unique image:")
525
+ generate_button = gr.Button("Create Image")
526
+ output_image = gr.Image(label="Your Generated Image")
527
+ download_button = gr.File(label="Download Image", type="filepath")
528
 
529
  generate_button.click(self.inference.text_to_image,
530
  inputs=prompt_input,
 
532
 
533
  # Image-to-Image Section
534
  gr.Markdown("## Image to Image Feature")
535
+ gr.Markdown("""
536
+ *Enhance or Transform Your Existing Images*
537
+
538
+ Upload an image and describe the changes you'd like to see. From subtle edits to artistic transformations, this feature helps you bring new life to your visuals.
539
+ - Example Edits:
540
+ - "Change the Color of the bag"
541
+ - "Add plastic to the suitcase."
542
+ - "mirror this photo of denim jacket."
543
+ """)
544
  with gr.Row():
545
+ input_image = gr.Image(label="Upload an image to modify:", type="pil")
546
+ prompt_modification = gr.Textbox(label="Describe the changes you want:")
547
+ modify_button = gr.Button("Apply Changes")
548
+ modified_output_image = gr.Image(label="Your Modified Image")
549
  download_modified_button = gr.File(label="Download Modified Image", type="filepath")
550
 
551
  modify_button.click(self.inference.text_image_to_image,
 
554
 
555
  # Background Removal and Resizing Section
556
  gr.Markdown("## Image Background Removal and Resizing with Optional Watermark")
557
+ gr.Markdown("""
558
+ *Perfect Your Images for Any Use Case*
559
+
560
+ Easily remove backgrounds, resize images to fit your needs, and even add watermarks to maintain originality or branding. This feature is ideal for e-commerce, social media, and design projects.
561
+ - Features:
562
+ - Supports batch processing of multiple images or ZIP/RAR files.
563
+ - Options for transparent, solid color, or custom backgrounds.
564
+ - Output in your choice of PNG or JPG format.
565
+ """)
566
  with gr.Row():
567
+ input_files = gr.File(label="Upload an image or a ZIP/RAR file for batch processing:", file_types=[".zip", ".rar", "image"], interactive=True)
568
+ watermark = gr.File(label="Upload an optional watermark (PNG only):", file_types=[".png"])
569
 
570
  with gr.Row():
571
+ canvas_size = gr.Radio(choices=["Rox", "Columbia", "Zalora"], label="Select the desired canvas size:", value="Rox")
572
+ output_format = gr.Radio(choices=["PNG", "JPG"], label="Choose the output format:", value="JPG")
573
+ num_workers = gr.Slider(minimum=1, maximum=16, step=1, label="Set the number of processing threads:", value=5)
574
 
575
  with gr.Row():
576
+ bg_method = gr.Radio(choices=["bria", "rembg", None], label="Choose a background removal method:", value="bria")
577
+ bg_choice = gr.Radio(choices=["transparent", "white", "custom"], label="Select a background style:", value="white")
578
+ custom_color = gr.ColorPicker(label="Pick a custom background color (if applicable):", value="#ffffff", visible=False)
579
 
580
+ process_button = gr.Button("Start Processing")
581
  with gr.Row():
582
  gallery_processed = gr.Gallery(label="Processed Images")
583
  with gr.Row():
584
+ image_original = gr.Image(label="Original Image Preview", interactive=False)
585
+ image_processed = gr.Image(label="Processed Image Preview", interactive=False)
586
  with gr.Row():
587
+ original_ratio = gr.Textbox(label="Aspect Ratio (Original)")
588
+ processed_ratio = gr.Textbox(label="Aspect Ratio (Processed)")
589
  with gr.Row():
590
+ output_zip = gr.File(label="Download All Processed Images (ZIP)")
591
+ processing_time = gr.Textbox(label="Total Processing Time (seconds)")
592
 
593
  bg_choice.change(self.show_color_picker, inputs=bg_choice, outputs=custom_color)
594
  process_button.click(self.master_process, inputs=[input_files, bg_method, watermark, canvas_size, output_format, bg_choice, custom_color, num_workers],