added sliders for the first tab

app.py

@@ -1,54 +1,10 @@
 import cv2
-import ezdxf
 import gradio as gr
 import numpy as np
-from distortion import generate_matrix
+from distortion import generate_matrix, get_select_coords, correct_image, track
 
 
 
-def load_coefficients(path):
-    """Load camera matrix and distortion coefficients from file."""
-    cv_file = cv2.FileStorage(path.name, cv2.FILE_STORAGE_READ)
-    camera_matrix = cv_file.getNode('K').mat()
-    dist_matrix = cv_file.getNode('D').mat()
-    cv_file.release()
-    return [camera_matrix, dist_matrix]
-
-
-def correct_image(image, yaml):
-    image = cv2.imread(image)
-    mtx, dist = load_coefficients(yaml)
-    dst = cv2.undistort(image, mtx, dist, None, None)
-    return dst
-
-
-
-coordis =  []
-
-def get_select_coords(img, evt: gr.SelectData):
-
-    row, col = evt.index
-    coordis.append([row, col])
-    
-    if len(coordis) == 4 :
-        coordinates = np.array(coordis)
-        print("shape of second array:", coordinates.shape)
-        print(coordinates)
-        dwg = ezdxf.new("R2010")
-        msp = dwg.modelspace()
-        dwg.layers.new(name="greeny green lines", dxfattribs={"color": 3})
-        
-        msp.add_line((coordinates[0][0], -coordinates[0][1]), (coordinates[1][0], -coordinates[1][1]))
-        msp.add_line((coordinates[1][0], -coordinates[1][1]), (coordinates[2][0], -coordinates[2][1]))
-        msp.add_line((coordinates[2][0], -coordinates[2][1]), (coordinates[3][0], -coordinates[3][1]))
-        msp.add_line((coordinates[3][0], -coordinates[3][1]), (coordinates[0][0], -coordinates[0][1]))
-        
-        dwg_file = dwg.saveas("output.dxf") 
-        coordis.clear()
-
-        return "output.dxf"
-                
-
 SHARED_UI_WARNING = f'''##### Lorem ipsum dolor sit amet, consectetur adipiscing elit. Sed tempus dignissim odio, at elementum erat vulputate sit amet. Vestibulum sodales viverra fermentum. In ac hendrerit dolor, vitae mattis odio. Maecenas suscipit consectetur suscipit. Curabitur sodales dui eget neque venenatis tincidunt. In sed libero mi. Nam sollicitudin metus urna, sit amet sagittis ex laoreet sed.
 
 Pellentesque nunc turpis, porta ut accumsan eget, iaculis nec odio. Praesent fringilla a sem sed elementum. Proin orci justo, rutrum et feugiat eleifend, auctor sed odio. Maecenas posuere urna tortor, ut euismod ligula mattis sed. Sed ipsum velit, pretium sed lacinia sed, placerat eget urna. Mauris lobortis mi vitae odio luctus viverra sed eget urna. Pellentesque blandit semper felis sed congue. Aenean congue enim id euismod finibus.
@@ -61,6 +17,30 @@ with gr.Blocks() as demo:
     title = """<p><h1 align="center" style="font-size: 36px;">Auto Allign Quadrilateral Workpiece</h1></p>"""
     gr.HTML(title)
     with gr.Tab("Allign"):
+        
+        with gr.Row():
+            with gr.Column():
+                image_input = gr.Image(type="filepath")
+                image_output = gr.Image()
+            with gr.Column():
+                first_slider = gr.Slider(0,255, value=0, label="1. Slider")
+                second_slider = gr.Slider(0,255, value=0,label="2. Slider")
+                third_slider = gr.Slider(0,255, value=0,label="3. Slider")
+                forth_slider = gr.Slider(0,255, value=255,label="4. Slider")
+                fifth_slider = gr.Slider(0,255, value=255,label="5. Slider")
+                sixth_slider = gr.Slider(0,255, value=255,label="6. Slider")
+                current_values = gr.Textbox("Current vallues of sliders")
+            
+        
+
+        
+        first_slider.change(track, inputs=[image_input, first_slider, second_slider, third_slider, forth_slider, fifth_slider, sixth_slider], outputs=[image_output, current_values])
+        second_slider.change(track, inputs=[image_input, first_slider, second_slider, third_slider, forth_slider, fifth_slider, sixth_slider], outputs=[image_output, current_values])
+        third_slider.change(track, inputs=[image_input, first_slider, second_slider, third_slider, forth_slider, fifth_slider, sixth_slider], outputs=[image_output, current_values])
+        forth_slider.change(track, inputs=[image_input, first_slider, second_slider, third_slider, forth_slider, fifth_slider, sixth_slider], outputs=[image_output, current_values])
+        fifth_slider.change(track, inputs=[image_input, first_slider, second_slider, third_slider, forth_slider, fifth_slider, sixth_slider], outputs=[image_output, current_values])
+        sixth_slider.change(track, inputs=[image_input, first_slider, second_slider, third_slider, forth_slider, fifth_slider, sixth_slider], outputs=[image_output, current_values])
+
         with gr.Row():
             with gr.Column():
                 image_input = gr.Image(label="Select Image")

distortion.py

@@ -1,8 +1,10 @@
 import cv2
 import numpy as np
+import ezdxf
+import gradio as gr
 from pathlib import Path
 
-
+coordis =  []
 
 def save_coefficients(mtx, dist, path):
     """Save camera matrix and distortion coefficients to file."""
@@ -10,6 +12,23 @@ def save_coefficients(mtx, dist, path):
     cv_file.write('K', mtx)
     cv_file.write('D', dist)
     cv_file.release()
+    
+    
+def load_coefficients_yaml(path):
+    """Load camera matrix and distortion coefficients from file."""
+    cv_file = cv2.FileStorage(path.name, cv2.FILE_STORAGE_READ)
+    camera_matrix = cv_file.getNode('K').mat()
+    dist_matrix = cv_file.getNode('D').mat()
+    cv_file.release()
+    return [camera_matrix, dist_matrix]
+
+
+def correct_image(image, yaml):
+    image = cv2.imread(image)
+    mtx, dist = load_coefficients_yaml(yaml)
+    dst = cv2.undistort(image, mtx, dist, None, None)
+    return dst
+
 
 
 def load_coefficients(path):
@@ -20,6 +39,29 @@ def load_coefficients(path):
     cv_file.release()
     return [camera_matrix, dist_matrix]
 
+def get_select_coords(img, evt: gr.SelectData):
+
+    row, col = evt.index
+    coordis.append([row, col])
+    
+    if len(coordis) == 4 :
+        coordinates = np.array(coordis)
+        print("shape of second array:", coordinates.shape)
+        print(coordinates)
+        dwg = ezdxf.new("R2010")
+        msp = dwg.modelspace()
+        dwg.layers.new(name="greeny green lines", dxfattribs={"color": 3})
+        
+        msp.add_line((coordinates[0][0], -coordinates[0][1]), (coordinates[1][0], -coordinates[1][1]))
+        msp.add_line((coordinates[1][0], -coordinates[1][1]), (coordinates[2][0], -coordinates[2][1]))
+        msp.add_line((coordinates[2][0], -coordinates[2][1]), (coordinates[3][0], -coordinates[3][1]))
+        msp.add_line((coordinates[3][0], -coordinates[3][1]), (coordinates[0][0], -coordinates[0][1]))
+        
+        dwg_file = dwg.saveas("output.dxf") 
+        coordis.clear()
+
+        return "output.dxf"
+
 
 def generate_matrix(filename, board_vert, board_horz):
     """Main function to calibrate camera and undistort image."""
@@ -64,7 +106,28 @@ def generate_matrix(filename, board_vert, board_horz):
     except:
         print("Please check the Chessboard Dimensions")
         
-    
-    
+        
+def track(img, h1, s1, v1, h2, s2, v2):
+    # Load the image
+    img = cv2.imread(img)
+
+
+    while(1):
+        #hsv = cv2.cvtColor(img, cv2.COLOR_BGR2HSV)
+        img = cv2.cvtColor(img, cv2.COLOR_RGB2BGR)
+
+        # Create the NumPy arrays
+        lower_red = np.array([h1, s1, v1])
+        upper_red = np.array([h2, s2, v2])
+        
+        # Convert every element to integer using int() function
+        lower_red = np.array([int(x) for x in lower_red])
+        upper_red = np.array([int(x) for x in upper_red])
+        
+        total = np.array([h1, s1, v1, h2, s2, v2])
 
+        mask = cv2.inRange(img, lower_red, upper_red)
+        res = cv2.bitwise_and(img,img, mask= mask)
+        #hsv = cv2.cvtColor(res, cv2.COLOR_BGR2HSV)
     
+        return res, total