Thumbnail update #1

app.py

@@ -38,6 +38,7 @@ from utils.misc import (get_filename, pause, convert_ratio_to_dimensions) #insta
 from utils.image_utils import (
     change_color,
     open_image,
+    build_prerendered_images_by_quality,
     upscale_image,
     lerp_imagemath,
     shrink_and_paste_on_blank,
@@ -48,7 +49,8 @@ from utils.image_utils import (
     crop_and_resize_image,
     convert_to_rgba_png,
     resize_image_with_aspect_ratio,
-    build_prerendered_images_by_quality
+    build_prerendered_images_by_quality,
+    get_image_from_dict
 )
 
 from utils.hex_grid import (
@@ -83,14 +85,13 @@ PIPELINE_CLASSES = {
     "FluxControlPipeline": FluxControlPipeline
 }
 
-import spaces
-
 from utils.version_info import (
     versions_html,
     #initialize_cuda,
     #release_torch_resources,
     #get_torch_info
 )
+import spaces
 
 input_image_palette = []
 current_prerendered_image = gr.State("./images/images/Beeuty-1.png")
@@ -304,7 +305,7 @@ class Condition(object):
 # def generate_image(pipe, generate_params, progress=gr.Progress(track_tqdm=True)):
 #     return pipe(**generate_params)
 
-#@spaces.GPU(duration=140, progress=gr.Progress(track_tqdm=True))
+@spaces.GPU(duration=140, progress=gr.Progress(track_tqdm=True))
 def generate_image_lowmem(
     text,
     neg_prompt=None,
@@ -603,7 +604,7 @@ def generate_ai_image_local (
         #gc.collect()
         return None
 
-#@spaces.GPU(duration=140,progress=gr.Progress(track_tqdm=True))
+@spaces.GPU(duration=140,progress=gr.Progress(track_tqdm=True))
 def generate_input_image_click(map_option, prompt_textbox_value, negative_prompt_textbox_value, model_textbox_value, randomize_seed=True, seed=None, use_conditioned_image=False, strength=0.5, image_format="16:9", scale_factor=(8/3), progress=gr.Progress(track_tqdm=True)):
     if randomize_seed:
         seed = random.randint(0, constants.MAX_SEED)
@@ -789,6 +790,7 @@ def create_3d_obj(rgb_image, raw_depth, image_path, depth=10, z_scale=200):
     o3d.io.write_triangle_mesh(gltf_path, mesh_crop, write_triangle_uvs=True)
     return gltf_path
 
+@spaces.GPU()
 def depth_process_image(image_path, resized_width=800, z_scale=208):
     """
     Processes the input image to generate a depth map and a 3D mesh reconstruction.
@@ -863,7 +865,7 @@ def generate_depth_button_click(depth_image_source, resize_width, z_scale, input
 
     return generate_depth_and_3d(image_path, resize_width, z_scale)
 
-
+@spaces.GPU()
 def getVersions():
     return versions_html()
 
@@ -1034,7 +1036,7 @@ with gr.Blocks(css_paths="style_20250128.css", title=title, theme='Surn/beeuty')
                     with gr.Row():
                         with gr.Column():
                             map_options = gr.Dropdown(
-                            label="Map Options",
+                            label="Map Options*",
                             choices=list(constants.PROMPTS.keys()),
                             value="Alien Landscape",
                             elem_classes="solid",
@@ -1051,9 +1053,9 @@ with gr.Blocks(css_paths="style_20250128.css", title=title, theme='Surn/beeuty')
                                 maximum=constants.MAX_SEED,
                                 step=1,
                                 value=0,
-                                scale=0
+                                scale=0, randomize=True, elem_id="rnd_seed"
                             )
-                            randomize_seed = gr.Checkbox(label="Randomize seed", value=True, scale=0, interactive=True)
+                            randomize_seed = gr.Checkbox(label="Randomize seed", value=False, scale=0, interactive=True)
                     prompt_textbox = gr.Textbox(
                         label="Prompt",
                         visible=False,
@@ -1084,20 +1086,20 @@ with gr.Blocks(css_paths="style_20250128.css", title=title, theme='Surn/beeuty')
                             "Generate AI Image",
                             elem_id="generate_input_image",
                             elem_classes="solid"
-                    )
+                        )
                 with gr.Column(scale=2):
                         with gr.Accordion("Template Image Styles", open = False):
                             with gr.Row():
                                 with gr.Column(scale=2):
                                     # Gallery from PRE_RENDERED_IMAGES GOES HERE
-                                    prerendered_image_gallery = gr.Gallery(label="Image Gallery", show_label=True, value=build_prerendered_images_by_quality(1), elem_id="gallery", elem_classes="solid", type="filepath", columns=[3], rows=[3], preview=False ,object_fit="contain", height="auto", format="png",allow_preview=False)
+                                    prerendered_image_gallery = gr.Gallery(label="Image Gallery", show_label=True, value=build_prerendered_images_by_quality(3,'thumbnail'), elem_id="gallery", elem_classes="solid", type="filepath", columns=[3], rows=[3], preview=False ,object_fit="contain", height="auto", format="png",allow_preview=False)
                                 with gr.Column():
-                                    image_guidance_stength = gr.Slider(label="Image Guidance Strength (prompt percentage)", minimum=0, maximum=1.0, value=0.8, step=0.01, interactive=True)                                
+                                    image_guidance_stength = gr.Slider(label="Image Guidance Strength (prompt percentage)", minimum=0, maximum=1.0, value=0.85, step=0.01, interactive=True)                                
                                     replace_input_image_button = gr.Button(
                                         "Replace Input Image",
                                         elem_id="prerendered_replace_input_image_button",
                                         elem_classes="solid"
-                                    )                                
+                                    )
                                     generate_input_image_from_gallery = gr.Button(
                                         "Generate AI Image from Gallery",
                                         elem_id="generate_input_image_from_gallery",
@@ -1295,10 +1297,8 @@ with gr.Blocks(css_paths="style_20250128.css", title=title, theme='Surn/beeuty')
         scroll_to_output=True
     )
 
-    hexaGrid.queue(default_concurrency_limit=1,max_size=12,api_open=False)
-    hexaGrid.launch(allowed_paths=["assets","/","./assets","images","./images", "./images/prerendered"], favicon_path="./assets/favicon.ico", max_file_size="10mb")
-
 if __name__ == "__main__":
+    constants.load_env_vars(constants.dotenv_path)
     logging.basicConfig(
         format="[%(levelname)s] %(asctime)s %(message)s", level=logging.INFO
     )
@@ -1313,4 +1313,6 @@ if __name__ == "__main__":
     # setup_runtime_env()
     #main(os.getenv("DEBUG") == "1")
     #main()
+    hexaGrid.queue(default_concurrency_limit=1,max_size=12,api_open=False)
+    hexaGrid.launch(allowed_paths=["assets","/","./assets","images","./images", "./images/prerendered"], favicon_path="./assets/favicon.ico", max_file_size="10mb")