Merge branch 'master' of https://git.soton.ac.uk/gdp-project-4/AVVR-Pipeline-GDP4

mona-ssc @ a0816610

+Subproject commit a0816610da9fa1ed52674240d8343b5643f0e25c

scripts/GUI.py

@@ -34,17 +34,46 @@ class VRSceneCreatorGUI(QMainWindow):
     def setup_ui(self):
         # Create main widget and layout
         main_widget = QWidget()
+        main_widget.setStyleSheet("background-color: #1e1e1e;")
         self.setCentralWidget(main_widget)
         layout = QVBoxLayout(main_widget)
         
         # Add title/header
         header_label = QLabel("Immersive VR Scene Creator")
         header_label.setAlignment(Qt.AlignmentFlag.AlignCenter)
-        header_label.setStyleSheet("font-size: 24px; font-weight: bold; margin: 10px;")
+        header_label.setStyleSheet("""
+            font-size: 24px;
+            font-weight: bold;
+            margin: 10px;
+            color: white;
+        """)
         layout.addWidget(header_label)
         
         # Create tab widget
         self.tabs = QTabWidget()
+        self.tabs.setStyleSheet("""
+            QTabWidget::pane { /* The tab widget frame */
+                border-top: 2px solid #C2C7CB;
+            }
+            QTabBar::tab {
+                background: #3e3e3e;  
+                color: white;
+                padding: 10px;
+                border: 1px solid #C2C7CB;
+                border-bottom: none;
+                border-top-left-radius: 4px;
+                border-top-right-radius: 4px;
+            }
+            QTabBar::tab:selected {
+                background: #5e5e5e;
+                color: white;
+                border-color: #9B9B9B;
+            }
+            QTabBar::tab:hover {
+                background: #4e4e4e;
+                color: white;
+            }
+        """)
         layout.addWidget(self.tabs)
         
         # Initialize tabs

scripts/advanced_tab.py

@@ -44,44 +44,202 @@ class AdvancedTab(QWidget):
     def add_module_controls(self, layout):
         # Status section for overall progress
         status_group, self.status_text = create_group_with_text("Pipeline Status", 100)
+        status_group.setStyleSheet("""
+            QGroupBox {
+                font-weight: bold;
+                border: 2px solid grey;
+                border-radius: 10px;
+                margin-top: 10px;
+                background-color: #3e3e3e;
+                padding: 20px;
+            }
+            QGroupBox::title { 
+                margin: 10px;
+                background-color: transparent;
+                color: white;
+            }
+        """)
+        self.status_text.setStyleSheet("""
+            QTextEdit {
+                background-color: #1e1e1e;
+                color: white;
+                border: 2px solid grey;
+                border-radius: 10px;
+                padding: 10px;
+                font-size: 14px;
+            }
+            QTextEdit:focus {
+                border: 2px solid #05B8CC;
+            }
+        """)
         layout.addWidget(status_group)
         
         # Image Shifter Controls
         shifter_group = QGroupBox("Image Shifter")
         shifter_layout = QVBoxLayout(shifter_group)
+        shifter_group.setStyleSheet("""
+            QGroupBox {
+                font-weight: bold;
+                border: 2px solid grey;
+                border-radius: 10px;
+                margin-top: 10px;
+                background-color: #3e3e3e;
+                padding: 20px;
+            }
+            QGroupBox::title { 
+                margin: 10px;
+                background-color: transparent;
+                color: white;
+            }
+        """)
         # Add shifter specific controls
         layout.addWidget(shifter_group)
         
         # Depth Estimation Controls
         depth_group = QGroupBox("Depth Estimation")
         depth_layout = QVBoxLayout(depth_group)
+        depth_group.setStyleSheet("""
+            QGroupBox {
+                font-weight: bold;
+                border: 2px solid grey;
+                border-radius: 10px;
+                margin-top: 10px;
+                background-color: #3e3e3e;
+                padding: 20px;
+            }
+            QGroupBox::title { 
+                margin: 10px;
+                background-color: transparent;
+                color: white;
+            }
+        """)
         # Add depth estimation specific controls
         layout.addWidget(depth_group)
         
         # Material Recognition Controls
         material_group = QGroupBox("Material Recognition")
         material_layout = QVBoxLayout(material_group)
+        material_group.setStyleSheet("""
+            QGroupBox {
+                font-weight: bold;
+                border: 2px solid grey;
+                border-radius: 10px;
+                margin-top: 10px;
+                background-color: #3e3e3e;
+                padding: 20px;
+            }
+            QGroupBox::title { 
+                margin: 10px;
+                background-color: transparent;
+                color: white;
+            }
+        """)
         # Add material recognition specific controls
         layout.addWidget(material_group)
         
         # EdgeNet Controls
         edgenet_group = QGroupBox("EdgeNet")
         edgenet_layout = QVBoxLayout(edgenet_group)
+        edgenet_group.setStyleSheet("""
+            QGroupBox {
+                font-weight: bold;
+                border: 2px solid grey;
+                border-radius: 10px;
+                margin-top: 10px;
+                background-color: #3e3e3e;
+                padding: 20px;
+            }
+            QGroupBox::title { 
+                margin: 10px;
+                background-color: transparent;
+                color: white;
+            }
+        """)
         # Add EdgeNet specific controls
         layout.addWidget(edgenet_group)
         
         # Mesh Processing Controls
         mesh_group = QGroupBox("Mesh Processing")
         mesh_layout = QVBoxLayout(mesh_group)
+        mesh_group.setStyleSheet("""
+            QGroupBox {
+                font-weight: bold;
+                border: 2px solid grey;
+                border-radius: 10px;
+                margin-top: 10px;
+                background-color: #3e3e3e;
+                padding: 20px;
+            }
+            QGroupBox::title { 
+                margin: 10px;
+                background-color: transparent;
+                color: white;
+            }
+        """)
         # Add mesh processing specific controls
         layout.addWidget(mesh_group)
         
         # Preview section
         preview_group = QGroupBox("Preview")
         preview_layout = QHBoxLayout(preview_group)
+        preview_group.setStyleSheet("""
+            QGroupBox {
+                font-weight: bold;
+                border: 2px solid grey;
+                border-radius: 10px;
+                margin-top: 10px;
+                background-color: #3e3e3e;
+                padding: 20px;
+            }
+            QGroupBox::title { 
+                margin: 10px;
+                background-color: transparent;
+                color: white;
+            }
+        """)
         
         input_group, self.input_preview = create_preview_group("Input")
         output_group, self.output_preview = create_preview_group("Output")
+        input_group.setStyleSheet("""
+            QGroupBox {
+                font-weight: bold;
+                border: 2px solid grey;
+                border-radius: 10px;
+                margin-top: 10px;
+                background-color: #1e1e1e;
+                padding: 20px;
+            }
+            QGroupBox::title { 
+                margin: 10px;
+                background-color: transparent;
+                color: white;
+            }
+            QLabel {
+                margin: 5px;
+                background-color: #1e1e1e;
+                color: white;
+            }
+        """)
+        output_group.setStyleSheet("""
+            QGroupBox {
+                font-weight: bold;
+                border: 2px solid grey;
+                border-radius: 10px;
+                margin-top: 10px;
+                background-color: #1e1e1e;
+                padding: 20px;
+            }
+            QGroupBox::title { 
+                margin: 10px;
+                background-color: transparent;
+                color: white;
+            }
+            QLabel {
+                margin: 5px;
+                background-color: #1e1e1e;
+                color: white;
+            }
+        """)
         
         preview_layout.addWidget(input_group)
         preview_layout.addWidget(output_group)

scripts/debug_tool/utils/clickable_label.py

+from PyQt6.QtWidgets import QInputDialog, QLabel
+from PyQt6.QtCore import Qt, pyqtSignal
+from PyQt6.QtGui import QPainter, QPen, QColor
+from collections import deque
+
+class ClickableLabel(QLabel):
+    point_added = pyqtSignal()
+    def __init__(self, parent=None):
+        super().__init__(parent)
+        self.points = deque(maxlen=2)  # Initialize a deque with a maximum length of 2
+
+    def mousePressEvent(self, event):
+        if event.button() == Qt.MouseButton.LeftButton:
+            x = event.position().x()
+            y = event.position().y()
+            number, ok = QInputDialog.getDouble(self, "Input Distance", "Enter a number (meters):", min=0)
+            if ok:
+                self.points.append((x, y, number))
+                print(f"Stored: x={x}, y={y}, number={number}")
+                self.point_added.emit()
+
+    def paintEvent(self, event):
+        super().paintEvent(event)
+        painter = QPainter(self)
+        pen = QPen(QColor("red"))
+        pen.setWidth(2)
+        painter.setPen(pen)
+        for i, point in enumerate(self.points):
+            x, y, _ = point
+            painter.drawEllipse(int(x) - 5, int(y) - 5, 10, 10)  # Draw a hollow circle
+            painter.drawText(int(x) + 10, int(y), str(i + 1))  # Draw the number next to the circle
+
+    
+    def get_points(self):
+        """
+        Returns the stored points from the image.
+
+        Returns:
+            list: List of tuples (x, y, number)
+        """
+        return list(self.points)  # Convert deque to list before returning
+    
+    def clear_points(self):
+        """
+        Clears the stored points from the image.
+        """
+        self.points.clear()
+        self.update()
+        self.point_added.emit()
\ No newline at end of file

scripts/scene_completion.py

+import paramiko
+from dotenv import load_dotenv
+from scp import SCPClient
+import os
+import subprocess
+import time
+
+load_dotenv()
+hostname = os.getenv("HOSTNAME")
+username = os.getenv("USERNAME")
+password = os.getenv("PASSWORD")
+
+def send_files(shifted_disparity_path, shifted_t_path):
+    # sends two files needed to iridis
+    # files keep their names. NAMES MUST BE shifted_disparity.png AND shifted_t.png
+    
+    command = [
+        "scp",
+        shifted_disparity_path,
+        f"{username}@{hostname}:{os.getenv("REMOTE_INPUT_PATH")}"
+    ]
+    try:
+        subprocess.run(command, check=True)
+
+    except subprocess.CalledProcessError as e:
+        print(f"Error during SCP: {e}")
+        return False, False
+
+
+    command = [
+        "scp",
+        shifted_t_path,
+        f"{username}@{hostname}:{os.getenv("REMOTE_INPUT_PATH")}"
+    ]
+    try:
+        subprocess.run(command, check=True)
+
+    except subprocess.CalledProcessError as e:
+        print(f"Error during SCP: {e}")
+        return False, False
+    return True
+
+def get_completed_scene(shifted_disparity_path, shifted_t_path):
+    if send_files(shifted_disparity_path, shifted_t_path):
+        
+        client = paramiko.SSHClient() 
+        client.set_missing_host_key_policy(paramiko.AutoAddPolicy()) 
+
+        # Connect to the server 
+        try:
+            client.connect(hostname, username=username, password=password)
+            print("SSH connection established.")
+            
+            # Check if the connection is active
+            if client.get_transport().is_active():
+                print("Connection is active.")
+            else:
+                print("Connection is not active.")
+                return False, False
+        except Exception as e:
+            print(f"Failed to connect to {hostname}: {e}")
+            return False, False
+         
+        stdin, stdout, stderr = client.exec_command(
+            "cd mona/MDBNet360_GDP/" + 
+            " && module load conda" + 
+            " && source activate" + 
+            " && conda activate ssc_env" + 
+            " && sbatch --partition=ecsstudents --account=ecsstudents run_enhance360_job.sh"
+        )
+        output = stdout.read().decode()
+        print(output)
+
+        stdin, stdout, stderr = client.exec_command("squeue -lu kproject")
+        output = stdout.read().decode()
+        print(output)
+
+        time.sleep(30)
+
+        while "RUNNING" in output:
+            print("Sleeping for 30")
+            time.sleep(30)
+            stdin, stdout, stderr = client.exec_command("squeue -lu kproject")
+            output = stdout.read().decode()
+            print(output)
+
+        print("Finish Enhancing")
+
+        stdin, stdout, stderr = client.exec_command(
+            "cd mona/MDBNet360_GDP/" + 
+            " && module load conda" + 
+            " && source activate" + 
+            " && conda activate ssc_env" + 
+            " && sbatch --partition=ecsstudents --account=ecsstudents run_obj_job.sh"
+        )
+        
+        output = stdout.read().decode()
+        print(output)
+
+        stdin, stdout, stderr = client.exec_command("squeue -lu kproject")
+        output = stdout.read().decode()
+        print(output)
+        time.sleep(30)
+
+        while "RUNNING" in output:
+            print("Sleeping for 30")
+            time.sleep(30)
+            stdin, stdout, stderr = client.exec_command("squeue -lu kproject")
+            output = stdout.read().decode()
+            print(output)
+
+        remote_file_path = "/mainfs/ECShome/kproject/mona/MDBNet360_GDP/output/scene_completed_prediction.obj"
+        local_file_path = "edgenet-360/Data/Input/scene_completed_prediction.obj"  
+        with SCPClient(client.get_transport()) as scp:
+            scp.get(remote_file_path, local_file_path)  # Download file
+
+        print("OUTPUT DOWNLOADED")
+        # Close the SSH connection 
+        client.close() 
+        return True, local_file_path
+
+    else:
+        return False, False
+
+
+# out = get_completed_scene("C:\\Project\\AVVR-Pipeline-GDP4\\edgenet-360\\Data\\Courtyard\\shifted-disparity.png", "C:\\Project\\AVVR-Pipeline-GDP4\\edgenet-360\\Data\\Courtyard\\shifted_t.png")
+# print(out)
+
+if __name__ == "__main__":
+    if len(sys.argv) != 3:
+        print("Usage: python shifter.py <shifted_disparity_path> <shifted_t_path>")
+        sys.exit(1)
+
+    shifted_disparity_path = sys.argv[1]
+    shifted_t_path = sys.argv[2]
+
+    saved_path = get_completed_scene(shifted_disparity_path, shifted_t_path)
+    print(f"Shifted image saved as {saved_path}")
\ No newline at end of file

scripts/simple_tab.py

 from PyQt6.QtWidgets import (QWidget, QVBoxLayout, QHBoxLayout, 
                              QGroupBox, QCheckBox, QMessageBox, QPushButton,
-                             QProgressBar)
-from PyQt6.QtCore import Qt, QThread, pyqtSignal
+                             QProgressBar, QComboBox, QLabel, QSizePolicy, QScrollArea)
+from PyQt6.QtCore import Qt, QThread, pyqtSignal, QTimer
 import os
 import sys
 
@@ -9,7 +9,8 @@ import sys
 from debug_tool.utils.qt_widgets import (create_group_with_text, create_button_layout, 
                                        create_info_group, create_preview_group)
 from debug_tool.utils.file_handlers import select_file, clean_directory, copy_file, run_command
-from debug_tool.utils.image_handlers import update_preview
+from debug_tool.utils.image_handlers import update_preview, load_and_resize_image, convert_cv_to_pixmap
+from debug_tool.utils.clickable_label import ClickableLabel
 
 # Import existing module implementations
 from debug_tool.tabs.shifter_tab import ShifterTab
@@ -149,33 +150,139 @@ class SimpleTab(QWidget):
         self.depth.hide()
         self.material.hide()
         self.edge_net.hide()
+
+        # Scroll Area
+        self.scroll_area = QScrollArea(self)
+        self.scroll_area.setWidgetResizable(True)
+
+        self.scroll_content = QWidget()
+
+        self.file_selected = False
+        self.flash_timer = QTimer(self)
+        self.flash_timer.timeout.connect(self.toggle_flash)
+        self.setup_ui(self.scroll_content)
         
-        self.setup_ui()
+        self.scroll_area.setWidget(self.scroll_content)
+        layout=QVBoxLayout(self)
+        layout.addWidget(self.scroll_area)
         
-    def setup_ui(self):
-        layout = QVBoxLayout(self)
+    def setup_ui(self,parent_widget):
+        layout = QVBoxLayout(parent_widget)
         
         # Controls section
         controls_group = QGroupBox("Pipeline Controls")
+        controls_group.setStyleSheet("""
+            QGroupBox {
+                font-weight: bold;
+                border: 2px solid grey;
+                border-radius: 20px;
+                margin-top: 10px;
+                background-color: #3e3e3e;
+                padding: 15px;
+            }
+            QGroupBox::title { 
+                margin: 10px;
+                background-color: transparent;
+                color: white;
+            }
+        """)
         controls_layout = QVBoxLayout(controls_group)
         
         # Info display
         info_rows = [
             ("Input Image:", "No file selected"),
-            ("Status:", "Ready")
+            ("Status:", "Ready - Waiting for input"),
         ]
         self.info_group, self.info_labels = create_info_group("Information", info_rows)
+        self.info_group.setStyleSheet("""
+            QGroupBox {
+                font-weight: bold;
+                border: 2px solid grey;
+                border-radius: 10px;
+                margin-top: 10px;
+                background-color: #3e3e3e;
+                padding: 20px;
+            }
+            QGroupBox::title { 
+                margin: 10px;
+                background-color: transparent;
+                color: white;
+            }
+            QLabel{
+                margin: 5px;
+                background-color: #3e3e3e;
+                color: white;
+            }                 
+        """)
+        for label in self.info_labels.values():
+            label.setStyleSheet("""
+                QLabel{
+                    margin: 5px;
+                    background-color: #3e3e3e;
+                    color: white;
+                }
+            """)
         controls_layout.addWidget(self.info_group)
         
         # Options
         options_layout = QHBoxLayout()
         
         self.include_top_check = QCheckBox("Include Top in Mesh")
+        self.include_top_check.setStyleSheet("""
+            QCheckBox {
+                margin: 5px;
+                padding: 5px;
+                background-color: #3e3e3e;
+                color: white;
+                border: none;
+                border-radius: 5px;
+            }
+        """)
         options_layout.addWidget(self.include_top_check)
         
         self.shift_image_check = QCheckBox("Shift Input Image")
+        self.shift_image_check.setStyleSheet("QCheckBox { margin: 5px; background-color: #3e3e3e;}")
         options_layout.addWidget(self.shift_image_check)
-        
+
+        # SSC Model selection
+        ssc_model_layout = QHBoxLayout()
+        ssc_model_label = QLabel("SSC Model:")
+        #ssc_model_label.setAlignment(Qt.AlignmentFlag.AlignRight | Qt.AlignmentFlag.AlignVCenter)
+        ssc_model_label.setStyleSheet("""
+            QLabel {
+                margin: 5px;
+                background-color: transparent;
+                color: white;
+                font-weight: bold;
+            }
+        """)
+        ssc_model_label.setFixedWidth(ssc_model_label.sizeHint().width())
+        ssc_model_layout.addWidget(ssc_model_label)
+
+        self.ssc_model_combo = QComboBox()
+        self.ssc_model_combo.addItems(["EdgeNet360", "MDBNet"])
+        self.ssc_model_combo.setStyleSheet("""
+            QComboBox {
+                margin: 5px;
+                padding: 5px;
+                background-color: #1e1e1e;
+                color: white;
+                border: none;
+                border-radius: 5px;
+            }
+            QComboBox QAbstractItemView {
+                background-color: #1e1e1e;
+                color: white;
+                selection-background-color: #5e5e5e;
+                border-radius: 5px;
+            }
+        """)
+        self.ssc_model_combo.setFixedWidth(150)
+        ssc_model_layout.addWidget(self.ssc_model_combo)
+        ssc_model_layout.setAlignment(Qt.AlignmentFlag.AlignLeft)
+        # Add the horizontal layout to the options layout
+        options_layout.addLayout(ssc_model_layout)
+
         controls_layout.addLayout(options_layout)
         
         # Progress Bar
@@ -183,38 +290,219 @@ class SimpleTab(QWidget):
         self.progress_bar.setMinimum(0)
         self.progress_bar.setMaximum(0)  # Makes it an indefinite progress bar
         self.progress_bar.hide()  # Hidden by default
+        self.progress_bar.setStyleSheet("""
+            QProgressBar {
+                border: 2px solid grey;
+                border-radius: 5px;
+                text-align: center;
+            }
+            QProgressBar::chunk {
+                background-color: #05B8CC;
+                width: 20px;
+            }
+        """)
         controls_layout.addWidget(self.progress_bar)
         
         # Buttons
         self.run_pipeline_btn = QPushButton("Run Pipeline")
         self.run_pipeline_btn.clicked.connect(self.run_full_pipeline)
         self.run_pipeline_btn.setEnabled(False)  # Disabled by default
+        self.run_pipeline_btn.setStyleSheet("""
+            QPushButton {
+                margin: 5px;
+                padding: 5px;
+                border-radius: 10px;
+            }
+            QPushButton:enabled {
+                background-color: green;
+                color: white;
+            }
+            QPushButton:disabled {
+            background-color: red;
+            color: white;
+            }
+        """)
+        self.run_pipeline_btn.setFixedSize(600, 40)  # Explicit size
         
         buttons_layout = QHBoxLayout()
-        select_btn = QPushButton("Select Input Image")
-        select_btn.clicked.connect(self.handle_file_select)
-        
-        buttons_layout.addWidget(select_btn)
+        self.select_btn = QPushButton("Select Input Image")
+        self.select_btn.clicked.connect(self.handle_file_select)
+        self.select_btn.setStyleSheet("""
+            QPushButton {
+                margin: 5px;
+                padding: 5px;
+                border-radius: 10px;
+            }
+        """)
+        self.select_btn.setFixedSize(600, 40)  # Explicit size
+        
+        buttons_layout.addWidget(self.select_btn)
         buttons_layout.addWidget(self.run_pipeline_btn)
         controls_layout.addLayout(buttons_layout)
         
         layout.addWidget(controls_group)
+
+        # Image Distance section
+        self.image_distance_group = QGroupBox("Image Distance")
+        distance_layout = QVBoxLayout(self.image_distance_group)
+        info_label = QLabel("Please select two point on the image and input the distance from the camera to that point.")
+        self.counter_label = QLabel("(0/2)")
+        self.distance_preview = ClickableLabel()
+        self.distance_preview.setAlignment(Qt.AlignmentFlag.AlignCenter)
+        self.counter_label.setAlignment(Qt.AlignmentFlag.AlignBottom | Qt.AlignmentFlag.AlignCenter)
+        self.points_info_label = QLabel()  # Label to display points and distances
+        self.points_info_label.setAlignment(Qt.AlignmentFlag.AlignTop | Qt.AlignmentFlag.AlignCenter)
+        self.image_distance_group.setStyleSheet("""
+            QGroupBox {
+                font-weight: bold;
+                border: 2px solid grey;
+                border-radius: 20px;
+                margin-top: 10px;
+                background-color: #3e3e3e;
+                padding: 20px;
+            }
+            QGroupBox::title { 
+                margin: 10px;
+                background-color: transparent;
+                color: white;
+            }
+            QLabel {
+                margin: 5px;
+                background-color: #3e3e3e;
+                color: white;
+            }
+        """)
+        # Center the ClickableLabel within its parent layout
+        distance_preview_layout = QHBoxLayout()
+        distance_preview_layout.addStretch()
+        distance_preview_layout.addWidget(self.distance_preview)
+        distance_preview_layout.addStretch()
+
+        self.distance_reset_btn = QPushButton("Reset Points")
+        self.distance_reset_btn.clicked.connect(self.distance_preview.clear_points)
+        self.distance_reset_btn.setFixedSize(150, 40)
+        self.distance_reset_btn.setStyleSheet("""
+            QPushButton {
+                margin: 5px;
+                padding: 5px;
+                border-radius: 10px;
+            }
+        """)
+        distance_btn_layout = QHBoxLayout()
+        distance_btn_layout.addStretch()
+        distance_btn_layout.addWidget(self.distance_reset_btn)
+        distance_btn_layout.addStretch()
+
+        distance_layout.addWidget(info_label)
+        distance_layout.addLayout(distance_preview_layout)
+        distance_layout.addWidget(self.points_info_label)
+        distance_layout.addWidget(self.counter_label)
+        distance_layout.addWidget(self.distance_reset_btn)
+        self.image_distance_group.hide()
+        layout.addWidget(self.image_distance_group)
+
+        self.distance_preview.point_added.connect(self.update_counter_Label)
         
         # Status section
-        status_group, self.status_text = create_group_with_text("Pipeline Status", 150)
+        status_group, self.status_text = create_group_with_text("Pipeline Status", 300)
+        status_group.setStyleSheet("""
+            QGroupBox {
+                font-weight: bold;
+                border: 2px solid grey;
+                border-radius: 20px;
+                margin-top: 10px;
+                background-color: #3e3e3e;
+                padding: 20px;
+            }
+            QGroupBox::title { 
+                margin: 10px;
+                background-color: transparent;
+                color: white;
+            }
+        """)
+        self.status_text.setStyleSheet("""
+            QTextEdit {
+                background-color: #1e1e1e;
+                color: white;
+                border: 2px solid grey;
+                border-radius: 10px;
+                padding: 10px;
+                font-size: 14px;
+            }
+            QTextEdit:focus {
+                border: 2px solid #05B8CC;
+            }
+        """)
         layout.addWidget(status_group)
         
         # Preview section
         preview_group = QGroupBox("Preview")
+        preview_group.setStyleSheet("""
+            QGroupBox {
+                font-weight: bold;
+                border: 2px solid grey;
+                border-radius: 20px;
+                margin-top: 10px;
+                background-color: #3e3e3e;
+                padding: 20px;
+            }
+            QGroupBox::title { 
+                margin: 10px;
+                background-color: transparent;
+                color: white;
+            }
+        """)
         preview_layout = QHBoxLayout(preview_group)
         
         input_group, self.input_preview = create_preview_group("Input Image")
         output_group, self.output_preview = create_preview_group("Current Output")
-        
+        input_group.setStyleSheet("""
+            QGroupBox {
+                font-weight: bold;
+                border: 2px solid grey;
+                border-radius: 10px;
+                margin-top: 10px;
+                background-color: #1e1e1e;
+                padding: 20px;
+            }
+            QGroupBox::title { 
+                margin: 10px;
+                background-color: transparent;
+                color: white;
+            }
+            QLabel {
+                margin: 5px;
+                background-color: #1e1e1e;
+                color: white;
+            }
+        """)
+        output_group.setStyleSheet("""
+            QGroupBox {
+                font-weight: bold;
+                border: 2px solid grey;
+                border-radius: 10px;
+                margin-top: 10px;
+                background-color: #1e1e1e;
+                padding: 20px;
+            }
+            QGroupBox::title { 
+                margin: 10px;
+                background-color: transparent;
+                color: white;
+            }
+            QLabel {
+                margin: 5px;
+                background-color: #1e1e1e;
+                color: white;
+            }
+        """)
         preview_layout.addWidget(input_group)
         preview_layout.addWidget(output_group)
         layout.addWidget(preview_group)
 
+        # Start flashing if no file is selected
+        self.start_flashing()
+
     def handle_file_select(self):
         
         file_path = select_file(
@@ -230,15 +518,53 @@ class SimpleTab(QWidget):
             self.update_status(f"Selected input file: {file_path}")
             update_preview(self.input_preview, file_path, 
                          error_callback=self.update_status)
-            
+            update_preview(self.distance_preview,file_path,max_size=1500)
+            pixmap = load_and_resize_image(file_path, 1500)
+            pixmap = convert_cv_to_pixmap(pixmap)
+            self.distance_preview.setFixedSize(pixmap.size())
+            self.image_distance_group.show()
+            self.update_status("Waiting for distance points...")
+            self.info_labels["Status:"].setText("Waiting for distance points...")
             # Enable the run pipeline button
-            self.run_pipeline_btn.setEnabled(True)
+            #self.run_pipeline_btn.setEnabled(True)
             
             # Provide input path to all modules
             self.shifter.input_file_path = file_path
             self.depth.depth_input_path = file_path
             self.material.input_file_path = file_path
             # self.edge_net.input_path = file_path # edgenet have default input path
+        
+        self.file_selected = True
+        self.flash_timer.stop()
+        self.select_btn.setStyleSheet("QPushButton { margin: 5px; padding: 5px; border-radius: 10px;}")
+    
+    def start_flashing(self):
+        if not self.file_selected:
+            self.flash_timer.start(1000)  # Flash every 1000 milliseconds
+
+    def toggle_flash(self):
+        current_style = self.select_btn.styleSheet()
+        if "background-color: DarkOrange;" in current_style:
+            self.select_btn.setStyleSheet("QPushButton { margin: 5px; padding: 5px; border-radius: 10px;}")
+        else:
+            self.select_btn.setStyleSheet("QPushButton { margin: 5px; padding: 5px; background-color: DarkOrange; border-radius: 10px;}")
+
+    def update_counter_Label(self):
+        count = len(self.distance_preview.get_points())
+        self.counter_label.setText(f"({count}/2)")
+        points_info = "\n".join([f"Point {i+1}: (x={x:.2f}, y={y:.2f}), Distance: {distance:.2f} meters)"
+                                for i, (x, y, distance) in enumerate(self.distance_preview.get_points())])
+        self.points_info_label.setText(points_info)
+
+        #enable run pipeline button if 2 points are selected
+        if count == 2:
+            self.run_pipeline_btn.setEnabled(True)
+            self.update_status("Distance points selected. Ready to run pipeline.")
+        else:
+            #pass
+            self.run_pipeline_btn.setEnabled(False)
+            self.update_status("Waiting for distance points...")
+            self.info_labels["Status:"].setText("Waiting for distance points...")
 
     def run_full_pipeline(self):
         if not self.input_path:
@@ -263,6 +589,14 @@ class SimpleTab(QWidget):
         self.setEnabled(False)
         self.progress_bar.setEnabled(True)  # Keep progress bar enabled
         
+        #TODO: Add model selection for EdgeNet or MDBNet
+        # Set the SSC model
+        self.selected_model = self.ssc_model_combo.currentText()
+
+        #TODO: Add distance points to the pipeline for depth estimation
+        # Get the distance points
+        self.distance_points = self.distance_preview.get_points()
+        
         # Start the pipeline
         self.pipeline_thread.start()