Update app.py

app.py

@@ -59,6 +59,13 @@ html_title = '''
 
 client = Client("http://198.175.88.247:17810/")
 
+def get_iframe(rgb_path: str, depth_path: str, viewer_mode: str = "6DOF"):
+
+    if viewer_mode == "6DOF":
+        return f"""<iframe src="file=static/three6dof.html" width="100%" height="500px" data-rgb="{rgb_path}" data-depth="{depth_path}"></iframe>"""
+    else:
+        return f"""<iframe src="file=static/depthmap.html" width="100%" height="500px" data-rgb="{rgb_path}" data-depth="{depth_path}"></iframe>"""
+        
 def generate(
     prompt: str,
     negative_prompt: str,
@@ -77,6 +84,8 @@ def generate(
 		fn_index=1
     )
 
+    result[-1] = get_iframe(result[0], result[1])
+    
     return result
 
 with gr.Blocks(theme=theme, css=css) as demo: