gui.py

import sys
from PyQt5.QtWidgets import QApplication, QMainWindow, QLabel, QLineEdit, QPushButton, QTextEdit, QVBoxLayout, QWidget
from PyQt5.QtGui import QPalette, QColor
from PyQt5.QtCore import Qt

class VulnerabilityScannerGUI(QMainWindow):
    def __init__(self):
        super().__init__()
        self.initUI()

    def initUI(self):
        self.setWindowTitle('Vulnerability Scanner')
        self.setGeometry(100, 100, 800, 600)

        # Set theme colors
        self.setTheme()

        # Create widgets
        self.url_label = QLabel('Enter URL:')
        self.url_input = QLineEdit()
        self.scan_button = QPushButton('Start Scan')
        self.results_area = QTextEdit()
        self.results_area.setReadOnly(True)

        # Connect button to scan function
        self.scan_button.clicked.connect(self.start_scan)

        # Layout
        layout = QVBoxLayout()
        layout.addWidget(self.url_label)
        layout.addWidget(self.url_input)
        layout.addWidget(self.scan_button)
        layout.addWidget(self.results_area)

        container = QWidget()
        container.setLayout(layout)
        self.setCentralWidget(container)

    def setTheme(self):
        palette = QPalette()
        palette.setColor(QPalette.Window, QColor(45, 45, 45))
        palette.setColor(QPalette.WindowText, Qt.white)
        palette.setColor(QPalette.Base, QColor(30, 30, 30))
        palette.setColor(QPalette.AlternateBase, QColor(45, 45, 45))
        palette.setColor(QPalette.ToolTipBase, Qt.white)
        palette.setColor(QPalette.ToolTipText, Qt.white)
        palette.setColor(QPalette.Text, Qt.white)
        palette.setColor(QPalette.Button, QColor(45, 45, 45))
        palette.setColor(QPalette.ButtonText, Qt.white)
        palette.setColor(QPalette.BrightText, Qt.red)
        palette.setColor(QPalette.Highlight, QColor(100, 100, 150).lighter())
        palette.setColor(QPalette.HighlightedText, Qt.black)
        self.setPalette(palette)

    def start_scan(self):
        url = self.url_input.text()
        if url:
            self.results_area.append(f'Scanning {url}...')
            # Call your existing scanning function here and display the results
            # For example:
            # results = run_vulnerability_scan(url)
            # self.results_area.append(results)

if __name__ == '__main__':
    app = QApplication(sys.argv)
    gui = VulnerabilityScannerGUI()
    gui.show()
    sys.exit(app.exec_())