Update app.py

app.py

@@ -91,10 +91,19 @@ def draw_plot(pred_img, seg):
     plt.xticks([], [])
     ax.tick_params(width=0.0, labelsize=25)
     return fig
-    
+
 def on_mouseover(mouse_info):
     x, y = mouse_info["position"]
-    print(f"Mouse over position: ({x}, {y})")
+    
+    # 사용자 정의 함수를 호출하여 특정 영역을 강조
+    highlight_segmentation(x, y)
+
+def highlight_segmentation(x, y):
+    # 마우스 위치에 대한 세그멘테이션 정보 가져오기
+    seg_value = seg.numpy()[y, x]
+
+    # 세그멘테이션 정보 출력
+    print(f"Mouse over position: ({x}, {y}), Segmentation value: {seg_value}")
     
 def sepia(input_img):
     input_img = Image.fromarray(input_img)
@@ -107,6 +116,7 @@ def sepia(input_img):
     logits = tf.image.resize(
         logits, input_img.size[::-1]
     )  # We reverse the shape of `image` because `image.size` returns width and height.
+    global seg 
     seg = tf.math.argmax(logits, axis=-1)[0]
 
     color_seg = np.zeros(
@@ -120,7 +130,6 @@ def sepia(input_img):
     pred_img = pred_img.astype(np.uint8)
 
     fig = draw_plot(pred_img, seg)
-    #filename = input_image.name if hasattr(input_image, "name") else "Unknown filename"
     return fig
 
 demo = gr.Interface(fn=sepia,
@@ -132,5 +141,4 @@ demo = gr.Interface(fn=sepia,
                     allow_flagging='never',
                     events=[gr.Mouse(on_mouseover)])
 
-
 demo.launch()