Alex-Watchman commited on
Commit
383071b
·
verified ·
1 Parent(s): 2f5667f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +39 -0
app.py CHANGED
@@ -0,0 +1,39 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # -*- coding: utf-8 -*-
2
+ """Главный файл приложения для детекции дефектов сварки"""
3
+
4
+ import sys
5
+ from PyQt5.QtWidgets import QApplication, QMessageBox
6
+ from src import MainWindow
7
+ from src import check_yolo_availability
8
+
9
+
10
+ def main():
11
+ """Главная функция приложения"""
12
+ app = QApplication(sys.argv)
13
+ app.setStyle('Fusion')
14
+
15
+ # Проверяем доступность библиотек
16
+ if not check_yolo_availability():
17
+ msg = QMessageBox()
18
+ msg.setIcon(QMessageBox.Warning)
19
+ msg.setWindowTitle("Предупреждение")
20
+ msg.setText("Библиотека ultralytics не установлена!")
21
+ msg.setInformativeText(
22
+ "Для работы программы требуется установить библиотеку ultralytics.\n\n"
23
+ "Установите её, выполнив команду:\n"
24
+ "pip install ultralytics\n\n"
25
+ "Продолжить без детекции?"
26
+ )
27
+ msg.setStandardButtons(QMessageBox.Yes | QMessageBox.No)
28
+
29
+ if msg.exec_() == QMessageBox.No:
30
+ sys.exit(1)
31
+
32
+ window = MainWindow()
33
+ window.show()
34
+
35
+ sys.exit(app.exec_())
36
+
37
+
38
+ if __name__ == "__main__":
39
+ main()