Update app.py

app.py

@@ -4,9 +4,8 @@
 # 输出图片
 # 使用 depth-anything + teed 生成外轮廓
 # 使用 teed + 腐蚀算法 生成内边缘
-
-
 from PIL import Image
+
 import cv2
 import numpy as np
 import os
@@ -16,16 +15,11 @@ from torchvision.transforms import Compose
 from tqdm import tqdm
 import TEED.main as teed
 from TEED.main import parse_args
-import gradio as gr 
-
 
 from depthAnything.depth_anything.dpt import DepthAnything
 from depthAnything.depth_anything.util.transform import Resize, NormalizeImage, PrepareForNet
 import shutil
 
-import sys
-print(sys.path)
-
 
 def multiply_blend(image1, image2):
     # 将图片转换为浮点数，方便计算
@@ -358,28 +352,38 @@ def process_line(img_path='./input', outdir='./output'):
 
 
 if __name__ == '__main__':
-    # depth_anything()
-    # teed_imgs('./input', './output/teed_imgs', [1,7,2])
-    # teed_imgs('./output/depth_anything', './output/dp_teed_imgs', [0,7,2])
-    # merge_images_in_2_folder('./output/teed_imgs', './output/dp_teed_imgs', './output/merged_imgs','_depth', 1, 'multiply', [[2,0],[2,1]],[1,0])
+    # 原有的处理逻辑（如果需要）
+    # img_path = "input"
+    # outdir = "output"
+    # process_line(img_path, outdir)
 
-    # erosion_img_from_path('./output/teed_imgs', './output/erosion_imgs', 2, 1, True)
-    # guassian_blur_path('./input', './output/guassian_blur', 7, 2)
-    # erosion_img_from_path('./output/merged_imgs', './output/erosion_merged_imgs', 2, 1, False)
-    # erosion_img_from_path('./output/erosion_merged_imgs', './output/erosion2_merged_imgs', 2, 1, True)
-    img_path = "input"
-    outdir = "output"
-    process_line(img_path,outdir)
+    # Gradio 接口代码
+    import gradio as gr
+    from PIL import Image
 
+    def gradio_process_line(img):
+        # 保存输入图像到临时文件
+        img_path = './temp_input.png'
+        img.save(img_path)
 
-if __name__ == '__main__':
-    # 创建 Gradio 接口
-    interface = gr.Interface(fn=process_line,
-                             inputs=gr.Image(type="numpy", label="Upload Image"),
-                             outputs=gr.Image(label="Output Image"),
-                             title="Image Processing",
-                             description="Upload an image to process it using depth-anything and TEED.")
-    
-    # 启动 Gradio 接口
-    interface.launch()
+        # 调用处理函数
+        process_line(img_path, './output')
+
+        # 返回输出图像
+        output_image_path = './output/merged_imgs/your_final_image_name.png'  # 修改为实际输出路径
+        return Image.open(output_image_path)
+
+    # 定义 Gradio 接口
+    iface = gr.Interface(
+        fn=gradio_process_line,
+        inputs=gr.inputs.Image(type="pil"),
+        outputs="image",
+        title="Image Processing with Depth Anything and TEED",
+        description="Upload an image to process it with depth estimation and edge detection."
+    )
 
+    # 启动 Gradio 应用
+    iface.launch()
+
+
+