Update app.py

app.py

@@ -398,11 +398,18 @@ with gr.Blocks(title="Image Character Persona Generator") as iface:
                     # Copy to Chat Button
                     copy_to_chat_btn = gr.Button("Use This Persona in Chat")
             
-            # Function to refresh the character gallery
+           
+            # Modified function
             def refresh_character_gallery():
                 image_files = get_character_images()
-                return gr.Gallery(value=image_files)
-            
+                return image_files  # Return just the list of image paths
+                
+            # Modified function
+            def gallery_select(evt: gr.SelectData, gallery_images):
+                if gallery_images and evt.index < len(gallery_images):
+                    selected_image = gallery_images[evt.index]
+                    return selected_image, selected_image
+                return None, None
             # Function to handle gallery selection
             def gallery_select(evt: gr.SelectData, gallery_images):
                 if gallery_images and evt.index < len(gallery_images):
@@ -551,20 +558,20 @@ with gr.Blocks(title="Image Character Persona Generator") as iface:
                 inputs=[profile_persona],
                 outputs=[system_prompt]
             )
+
+            # Create a dedicated initialization function
+            def initialize_interface(profiles_data):
+                profile_names = [p["name"] for p in profiles_data["profiles"]] if profiles_data and "profiles" in profiles_data else []
+                image_files = get_character_images()
+                return profile_names, image_files  # Return raw values, not components
             
-            # Then use it:
+            # Then use it in place of your lambda
             iface.load(
                 fn=initialize_interface,
                 inputs=[profiles_state],
                 outputs=[profile_dropdown, character_gallery]
             )
-
-            # Initialize the profiles dropdown and character gallery
-            iface.load(
-                fn=lambda: (refresh_profiles_list, refresh_character_gallery),
-                inputs=[profiles_state],
-                outputs=[profile_dropdown, character_gallery]
-            )
+    
     
     # Function to update system prompt in Test tab when persona is generated
     def update_persona_state(caption, persona, time_output, img_path):