Update gradio_app.py

gradio_app.py

@@ -1,11 +1,12 @@
 import gradio as gr
-from app import depth_map_global  # Import ảnh Depth Map từ FastAPI
+from PIL import Image
 
 def get_depth_map():
-    """Trả về ảnh Depth Map mới nhất để hiển thị trên Gradio"""
-    if depth_map_global is None:
+    """Trả về ảnh Depth Map mới nhất từ FastAPI"""
+    try:
+        return Image.open("depth_map.png")  # 🟢 Lấy ảnh đã lưu từ FastAPI
+    except FileNotFoundError:
         return "Chưa có ảnh Depth Map nào được xử lý!"
-    return depth_map_global
 
 # 🟢 Tạo UI trên Hugging Face Spaces với Gradio
 gr.Interface(
@@ -14,5 +15,5 @@ gr.Interface(
     outputs=gr.Image(type="pil"),
     title="🔍 Depth Map Viewer",
     description="Hiển thị ảnh Depth Map mới nhất từ FastAPI",
-    live=True,  # Cập nhật theo thời gian thực
-).launch(share=True)  # `share=True` để tạo link truy cập từ bên ngoài
+    live=True,  # 🎯 **Tự động cập nhật ảnh mới**
+).launch(share=True)