Update app.py

app.py

@@ -68,11 +68,18 @@ def find_tail(points, tip):
 
     return None
 
+def draw_arrow(img_result, tip, tail):
+    # Draw arrow on the blank image
+    cv2.arrowedLine(img_result, tuple(tip), tuple(tail), (0, 255, 0), 3)
+    
 def infer(image_in):
     img = cv2.imread(image_in)
 
     contours, hierarchy = cv2.findContours(preprocess(img), cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)
 
+    # Create a blank image to accumulate arrows
+    img_result = np.zeros_like(img)
+
     for cnt in contours:
         peri = cv2.arcLength(cnt, True)
         approx = cv2.approxPolyDP(cnt, 0.025 * peri, True)
@@ -89,12 +96,16 @@ def infer(image_in):
                 arrow_tail = find_tail(approx[:,0,:], arrow_tip)
                 if arrow_tail :
                     cv2.circle(img, arrow_tail, 3, (255, 0, 0), cv2.FILLED)
+                    
+                    # Draw arrow on the same blank image
+                    draw_arrow(img_result, arrow_tip, arrow_tail)
 
 
 
     cv2.imwrite("Image_result.png", img)
+    cv2.imwrite("Arrows_on_same_blank.png", img_result)
 
-    return "Image_result.png", "1_gray.png", "2_blur.png", "3_canny.png", "4_dilate.png", "5_erode.png"
+    return "Image_result.png", "Arrows_on_same_blank.png"
 
 gr.Interface(
     fn=infer,
@@ -105,9 +116,6 @@ gr.Interface(
     outputs=[
         gr.Image(),
         gr.Image(),
-        gr.Image(),
-        gr.Image(),
-        gr.Image(),
-        gr.Image(),
+        
     ]
 ).launch()