Update app.py

app.py

@@ -1,6 +1,7 @@
 import os
 import gradio as gr
 from PIL import Image
+import json
 
 # Define global paths
 BASE_PATH = "ContraCLIP/experiments/wip/"
@@ -47,8 +48,9 @@ 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}"
-    with open(html_file, "r") as file:
-        return file.read()
+    # Embed the HTML file using an iframe
+    iframe = f'<iframe src="{html_file}" width="100%" height="600px" style="border:none;"></iframe>'
+    return iframe
 
 # Gradio Interface
 def build_interface():