Update app.py

app.py

@@ -4,10 +4,7 @@ import numpy as np
 import torch
 from PIL import Image
 import gradio as gr
-from gradio_imageslider import ImageSlider
 from depth_anything_v2.dpt import DepthAnythingV2
-import matplotlib.pyplot as plt
-import matplotlib
 
 # ===============================
 # Device & Model
@@ -29,22 +26,19 @@ def predict_depth(frame_rgb):
     return depth.astype(np.float32)
 
 # ===============================
-# Colormap for preview
+# Normalize to 0-255 grayscale
 # ===============================
-cmap = matplotlib.cm.get_cmap('magma')  # nice perceptual colormap
-
-def apply_colormap(depth):
-    """Scale depth to 0-1 and apply colormap, return uint8 RGB"""
+def depth_to_grayscale(depth):
     norm = (depth - depth.min()) / (depth.max() - depth.min() + 1e-8)
-    colored = (cmap(norm)[:, :, :3] * 255).astype(np.uint8)
-    return colored
+    gray = (norm * 255).astype(np.uint8)
+    return gray
 
 # ===============================
 # Process video
 # ===============================
 def process_video(video_file):
     """
-    Render depthmap video with colormap.
+    Render grayscale depthmap video.
     Keep original resolution & FPS.
     """
     OUTPUT_DIR = "output"
@@ -66,7 +60,7 @@ def process_video(video_file):
     # Video output path
     output_video_path = os.path.join(OUTPUT_DIR, os.path.basename(video_path).replace(".mp4","_depth.mp4"))
     fourcc = cv2.VideoWriter_fourcc(*'mp4v')
-    out = cv2.VideoWriter(output_video_path, fourcc, fps, (width,height), isColor=True)
+    out = cv2.VideoWriter(output_video_path, fourcc, fps, (width,height), isColor=False)
 
     # Slider preview (sample frames)
     slider_frames = []
@@ -82,14 +76,12 @@ def process_video(video_file):
 
         frame_rgb = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
         depth_map = predict_depth(frame_rgb)
-
-        # Apply colormap for video output
-        colored_frame = apply_colormap(depth_map)
-        out.write(cv2.cvtColor(colored_frame, cv2.COLOR_RGB2BGR))
+        gray_frame = depth_to_grayscale(depth_map)
+        out.write(gray_frame)
 
         # Add sampled frames for slider preview
         if idx % step == 0:
-            slider_frames.append(Image.fromarray(colored_frame))
+            slider_frames.append(Image.fromarray(gray_frame))
         idx += 1
 
     cap.release()
@@ -100,15 +92,15 @@ def process_video(video_file):
 # Gradio Interface
 # ===============================
 with gr.Blocks() as demo:
-    gr.Markdown("# Depth Anything V2 – Depth Video (vitb)")
+    gr.Markdown("# Depth Anything V2 – Grayscale Depth Video (vitb)")
     gr.Markdown(
-        "Upload an MP4 video to generate a **colored DepthMap video**.\n\n"
-        "**Model:** vitb – fast and high quality for real-time processing.\n"
+        "Upload an MP4 video to generate a **grayscale DepthMap video**.\n\n"
+        "**Model:** vitb – fast and high quality for video processing.\n"
         "Resolution and FPS are preserved from the original video."
     )
 
     video_input = gr.File(label="Upload MP4", file_types=['.mp4'])
-    depth_slider = ImageSlider(label="DepthMap Slider Preview")
+    depth_slider = gr.Gallery(label="DepthMap Slider Preview", elem_id="depth_slider")
     video_output = gr.Video(label="DepthMap Video")
     submit = gr.Button("Render DepthMap")