Update app.py

app.py

@@ -574,7 +574,7 @@ def draw_mask(l, t, v, d, evt: gr.EventData):
     return gr.ImageEditor(value=d)
 
 
-def findNormals():
+def findNormals(format):
     global depths
     d_im = cv2.cvtColor(cv2.imread(depths[frame_selected]).astype(np.uint8), cv2.COLOR_BGR2GRAY)
     zy, zx = np.gradient(d_im)  
@@ -583,7 +583,10 @@ def findNormals():
     #zx = cv2.Sobel(d_im, cv2.CV_64F, 1, 0, ksize=5)     
     #zy = cv2.Sobel(d_im, cv2.CV_64F, 0, 1, ksize=5)
 
-    normal = np.dstack((np.ones_like(d_im), zy, -zx))
+    if format == "opengl":
+        zy = -zy
+        
+    normal = np.dstack((np.ones_like(d_im), -zy, -zx))
     n = np.linalg.norm(normal, axis=2)
     normal[:, :, 0] /= n
     normal[:, :, 1] /= n
@@ -831,8 +834,9 @@ with gr.Blocks(css=css, js=js) as demo:
               reset.click(fn=reset_mask, inputs=None, outputs=[output_mask, output_depth])
 
             normals_out = gr.Image(label="Normal map", interactive=False)
+            format_normals = gr.Radio(choices=["directx", "opengl"])
             find_normals = gr.Button("Find normals")
-            find_normals.click(fn=findNormals, inputs=None, outputs=[normals_out])
+            find_normals.click(fn=findNormals, inputs=[format_normals], outputs=[normals_out])
 
         with gr.Column():
             model_type = gr.Dropdown([("small", "vits"), ("base", "vitb"), ("large", "vitl")], type="value", value="vits", label='Model Type')