Update UI/Main.py

UI/Main.py

@@ -16,6 +16,11 @@ demoImages = [
     "data/DJI_20240905094647_0003_Z.JPG",
     "data/DJI_20240905101846_0005_W.JPG",
     "data/16_3450.png", 
+    "data/16_3735.png",
+    "data/16_3900.png",
+    "data/19_00350.png",
+    "data/25_00272.png",
+    "data/67_02661.png"
 ]
 
 MODEL = CablePoleSegmentation(retina_mask=False)
@@ -27,35 +32,38 @@ with gr.Blocks(css="footer{display:none !important}") as block:
     gr.Markdown("## Yüksek Gerilim Hattı Kablo ve Direk Tespit ve Segmentasyon Uygulaması - Demo")
     
     with gr.Row():
-        image = gr.Image(label="Fotoğraf")
-        processedImageGallery = gr.Gallery(
-            label="Sonuçlar",
-            rows=1, 
-            columns=2, 
-            object_fit="contain", 
-            height="auto"
-        )
-    
-    with gr.Row() as imageGalleryRow:
-        with gr.Accordion("Demo Görsellerden Seçebilirsiniz", open=False):
-            imageGallery = gr.Gallery(
-                demoImages,
-                label="Demo Görseller",
-                rows=2, 
-                columns=3, 
-                object_fit="contain", 
-                height="auto"
-            )
-
+        
+        with gr.Column():
+            inputImage = gr.Image(label="Fotoğraf")
+            with gr.Accordion("Demo Görsellerden Seçebilirsiniz", open=False):
+                imageGallery = gr.Examples(
+                    examples=[
+                        os.path.join("data", img_name) for img_name in sorted(os.listdir("data"))
+                    ],
+                    inputs=[inputImage],
+                    label="Örnekler",
+                    cache_examples=False,
+                    examples_per_page=5
+                )
+
+        with gr.Column():
+            thresholdSlider = gr.Slider(0, 1, value=0.5, label="Model Eşik Değeri", info="0 ve 1 arası seçiniz.")
+
+            processButton = gr.Button("Tespit Et")
 
     results = gr.Textbox(label="Log")
+    gr.HTML("</hr>")
+    processedImageGallery = gr.Gallery(
+        label="Sonuçlar",
+        rows=1, 
+        columns=2, 
+        object_fit="contain", 
+        height="auto"
+    )
 
-    thresholdSlider = gr.Slider(0, 1, value=0.5, label="Model Eşik Değeri", info="0 ve 1 arası seçiniz.")
-
-    processButton = gr.Button("Tespit Et")
     annotatedImage = gr.AnnotatedImage(color_map=COLOR_MAP)
 
-    @processButton.click(outputs=[processedImageGallery, annotatedImage, results], inputs=[image, thresholdSlider])
+    @processButton.click(outputs=[processedImageGallery, annotatedImage, results], inputs=[inputImage, thresholdSlider])
     def Process(image, model_threshold):
         if image is None:
             raise gr.Warning("Lütfen görüntü yükleyiniz veya hazır seçiniz!", duration=3)
@@ -78,12 +86,6 @@ with gr.Blocks(css="footer{display:none !important}") as block:
         return [image, canvas], (image, sections), "Görüntü İşlendi!"
     
 
-    @imageGallery.select(inputs=imageGallery, outputs=[processedImageGallery, annotatedImage, results])
-    def GalleryProcess(image, evt_data: gr.EventData):
-        selectedIdx = evt_data._data["index"]
-        imagePath = demoImages[selectedIdx]
-        image = cv2.imread(imagePath)
-        return Process(image, thresholdSlider.value)
 
     
     block.load(Warning)
@@ -91,5 +93,3 @@ with gr.Blocks(css="footer{display:none !important}") as block:
 
 block.queue(max_size=10)
 block.launch(server_name="0.0.0.0", server_port=1071)
-
-