Update app.py

app.py

@@ -13,6 +13,8 @@ DEFAULT_IMAGE = "original_image.jpg"
 # Load semantic dipoles
 with open(SEMANTIC_DIPOLES_FILE, "r") as f:
     semantic_dipoles = json.load(f)
+    # Transform semantic_dipoles into "A -> B" format
+    formatted_dipoles = [f"{pair[0]} -> {pair[1]}" for pair in semantic_dipoles]
 
 # Helper to list all latent code folders
 latent_code_folders = sorted(
@@ -34,23 +36,28 @@ def load_dipole_paths(latent_code):
     )
     return paths
 
-# Function to display images
-def display_image(latent_code, semantic_dipole, frame_idx):
-    index = semantic_dipoles.index(semantic_dipole)
-    path_dir = os.path.join(LATENT_CODES_DIR, latent_code, "paths_images", f"path_{index:03d}")
+def display_image(latent_code, formatted_dipole, frame_idx):
+    # Reverse-map "A -> B" format back to the index in semantic_dipoles
+    try:
+        index = formatted_dipoles.index(formatted_dipole)
+    except ValueError:
+        return f"Error: Semantic dipole '{formatted_dipole}' not found in the list."
+
+    path_dir = os.path.join(
+        LATENT_CODES_DIR, latent_code, "paths_images", f"path_{index:03d}"
+    )
     frame_image_path = os.path.join(path_dir, f"{frame_idx:06d}.jpg")
+
     if not os.path.exists(frame_image_path):
-        return f"Image not found: {frame_image_path}"
+        return f"Image not found: {frame_image_path}."
+
     return Image.open(frame_image_path)
 
 # Function to display GAN latent space interactive plot
 def display_interactive_plot(latent_code):
-    html_file = os.path.join(LATENT_CODES_DIR, latent_code, f"interactive_latent_space_{latent_code}.html")
-    if not os.path.exists(html_file):
-        return f"Interactive file not found: {html_file}"
-    # Embed the HTML file using an iframe
-    iframe = f'<iframe src="{html_file}" width="100%" height="600px" style="border:none;"></iframe>'
-    return iframe
+    file_path = f"files/{LATENT_CODES_DIR}/{latent_code}/interactive_latent_space_{latent_code}.html"
+    iframe_html = f'<iframe src="{file_path}" width="800" height="600" frameborder="0"></iframe>'
+    return iframe_html
 
 # Gradio Interface
 def build_interface():
@@ -66,9 +73,9 @@ def build_interface():
                     value=latent_code_folders[0],
                 )
                 semantic_dipole_dropdown = gr.Dropdown(
-                    semantic_dipoles,
+                    formatted_dipoles,
                     label="Semantic Dipole",
-                    value=semantic_dipoles[0],
+                    value=formatted_dipoles[0],  # Set default value
                 )
                 frame_slider = gr.Slider(
                     0, 32, step=1, label="Frame Index"