Update app.py

app.py

@@ -3,31 +3,40 @@ import numpy as np
 
 # Define functions to wrap the private space's interfaces
 def cataract_analyzer(image):
-    private_space_id = "EyeMedicalDiagnosis"
+    private_space_id = "EyeMedicalDiagnosis/cataract_analyzer"
     private_space = gr.Interface.load(private_space_id)
     result = private_space(image)
-    if isinstance(result, list) and len(result) == 7:
-        return result
-    else:
-        return [np.zeros_like(image), "", np.zeros_like(image), 0, 0, 0, ""]
+    return parse_cataract_result(result, image)
 
 def glaucoma_analyzer(image):
-    private_space_id = "EyeMedicalDiagnosis"
+    private_space_id = "EyeMedicalDiagnosis/glaucoma_analyzer"
     private_space = gr.Interface.load(private_space_id)
     result = private_space(image)
-    if isinstance(result, list) and len(result) == 5:
-        return result
-    else:
-        return [np.zeros_like(image), "", "", "", ""]
+    return parse_glaucoma_result(result, image)
 
 def rlfd_analyzer(image):
-    private_space_id = "EyeMedicalDiagnosis"
+    private_space_id = "EyeMedicalDiagnosis/rlfd_analyzer"
     private_space = gr.Interface.load(private_space_id)
     result = private_space(image)
+    return parse_rlfd_result(result, image)
+
+def parse_cataract_result(result, image):
+    if isinstance(result, list) and len(result) == 7:
+        return result
+    else:
+        return [image, "Error: Invalid result", image, 0, 0, 0, ""]
+
+def parse_glaucoma_result(result, image):
+    if isinstance(result, list) and len(result) == 5:
+        return result
+    else:
+        return [image, "Error: Invalid result", "0", "0", "0"]
+
+def parse_rlfd_result(result, image):
     if isinstance(result, list) and len(result) == 8:
         return result
     else:
-        return [np.zeros_like(image), np.zeros_like(image), "", np.zeros_like(image), "", np.zeros_like(image), "", np.zeros_like(image), ""]
+        return [image, image, "Error: Invalid result", image, "0", image, "0", image, "0"]
 
 # Create the public interface with custom layout
 with gr.Blocks() as demo: