Update app.py

app.py

@@ -20,15 +20,37 @@ def predict_depth(image):
         -50,  # Frame's near plane offset
         api_name="/submit_depth_fn"
     )
-    return response['image']  # Adjust according to the actual key in the response
+    # Assuming the API returns a filepath or you save the result to a file
+    output_file_path = save_image_to_file(response['image'])  # Implement this function based on your API's response
+    return output_file_path
 
 # Gradio Interface
 iface = gr.Interface(
     fn=predict_depth,
     inputs=gr.Image(type='filepath', label="Upload your image"),
-    outputs=gr.Image(type='auto', label="Depth Map Image"),
+    outputs=gr.Image(type='filepath', label="Depth Map Image"),
     title="Depth Map Generator",
-    description="Upload an image to receive a depth map."
+    description="Upload an image to receive a depth map file."
 )
 
 iface.launch()
+
+def save_image_to_file(image_data):
+    # Assuming 'image_data' is the image data in a compatible format
+    # You would save the image to a file and return the path
+    import matplotlib.pyplot as plt
+    import matplotlib.image as mpimg
+    import numpy as np
+    import tempfile
+    import os
+
+    # Create a temporary file
+    fd, path = tempfile.mkstemp(suffix=".png")
+    try:
+        # Assume image_data is a numpy array
+        img = np.array(image_data).astype(np.uint8)
+        plt.imsave(path, img)
+    finally:
+        os.close(fd)
+
+    return path