Update app.py

app.py

@@ -3,10 +3,12 @@ import time
 import numpy as np
 import cv2
 import torch
+import gradio as gr
 from transformers import DPTFeatureExtractor, DPTForDepthEstimation
 from fastapi import FastAPI, File, UploadFile
 from PIL import Image
 import uvicorn
+import threading
 
 # 🟢 Tạo FastAPI
 app = FastAPI()
@@ -83,6 +85,29 @@ def detect_path(depth_map):
     else:
         return "backward"
 
-# 🟢 Chạy server FastAPI
+# 🟢 Hàm Gradio hiển thị ảnh Depth Map
+def get_depth_map():
+    """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ý!"
+
+# 🟢 Tạo giao diện Gradio trên Hugging Face Spaces
+def launch_gradio():
+    gr.Interface(
+        fn=get_depth_map,
+        inputs=[],
+        outputs=gr.Image(type="pil"),
+        title="🔍 Depth Map Viewer",
+        description="Hiển thị ảnh Depth Map mới nhất từ FastAPI",
+        live=True,  # 🎯 **Tự động cập nhật ảnh mới**
+    ).launch(share=True)
+
+# 🟢 Chạy cả FastAPI và Gradio cùng lúc
 if __name__ == "__main__":
+    # Chạy Gradio trong một luồng riêng
+    threading.Thread(target=launch_gradio, daemon=True).start()
+
+    # Chạy FastAPI
     uvicorn.run(app, host="0.0.0.0", port=7860)