Update app.py

app.py

@@ -1,15 +1,9 @@
 import gradio as gr
-import torch
 import requests
 import json
-# from diffusers import StableDiffusionLDM3DPipeline
 import gradio as gr
 import torch 
 from PIL import Image
-import base64
-from io import BytesIO
-from tempfile import NamedTemporaryFile
-from pathlib import Path
 
 from gradio_client import Client
 
@@ -59,7 +53,7 @@ html_title = '''
 
 client = Client("http://198.175.88.247:17810/")
 
-def get_iframe(rgb_path: str, depth_path: str, viewer_mode: str = "6DOF"):
+def build_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>"""
@@ -74,7 +68,7 @@ def generate(
     randomize_seed: bool = True,
 ):
 
-    result = client.predict(
+    rgb_url, depth_url, generated_seed, _ = client.predict(
 		prompt,	# str  in 'Prompt' Textbox component
 		negative_prompt,	# str  in 'Negative Prompt' Textbox component
 		guidance_scale,	# int | float (numeric value between 0 and 10) in 'Guidance Scale' Slider component
@@ -84,9 +78,9 @@ def generate(
 		fn_index=1
     )
 
-    result[-1] = get_iframe(result[0], result[1])
+    iframe = build_iframe(rgb_url, depth_url)
     
-    return result
+    return rgb, depth, generated_seed, iframe
 
 with gr.Blocks(theme=theme, css=css) as demo:
     gr.HTML(value=html_title)