Update run.py
Browse files
run.py
CHANGED
@@ -1,60 +1,60 @@
|
|
1 |
-
import os
|
2 |
-
import sys
|
3 |
-
import subprocess
|
4 |
-
|
5 |
-
def check_dependencies():
|
6 |
-
"""Проверяет, установлены ли все необходимые зависимости"""
|
7 |
-
try:
|
8 |
-
import flask
|
9 |
-
import torch
|
10 |
-
import transformers
|
11 |
-
print("✅ Все необходимые зависимости установлены.")
|
12 |
-
return True
|
13 |
-
except ImportError as e:
|
14 |
-
print(f"❌ Отсутствуют зависимости: {e}")
|
15 |
-
return False
|
16 |
-
|
17 |
-
def install_dependencies():
|
18 |
-
"""Устанавливает необходимые зависимости из requirements.txt"""
|
19 |
-
print("📦 Установка зависимостей...")
|
20 |
-
try:
|
21 |
-
subprocess.check_call([sys.executable, "-m", "pip", "install", "-r", "requirements.txt"])
|
22 |
-
print("✅ Зависимости успешно установлены!")
|
23 |
-
return True
|
24 |
-
except subprocess.CalledProcessError as e:
|
25 |
-
print(f"❌ Ошибка при установке зависимостей: {e}")
|
26 |
-
return False
|
27 |
-
|
28 |
-
def run_app():
|
29 |
-
"""Запускает Flask-приложение"""
|
30 |
-
print("🚀 Запуск
|
31 |
-
print("⚠️ Первый запуск может занять некоторое время, пока загружается модель.")
|
32 |
-
try:
|
33 |
-
from app import app
|
34 |
-
app.run(debug=True, port=5000)
|
35 |
-
except Exception as e:
|
36 |
-
print(f"❌ Ошибка при запуске приложения: {e}")
|
37 |
-
|
38 |
-
def main():
|
39 |
-
print("="*50)
|
40 |
-
print("🤖
|
41 |
-
print("="*50)
|
42 |
-
|
43 |
-
# Проверяем зависимости
|
44 |
-
if not check_dependencies():
|
45 |
-
print("\nНеобходимо установить зависимости.")
|
46 |
-
choice = input("Установить зависимости сейчас? (y/n): ").lower()
|
47 |
-
if choice == 'y':
|
48 |
-
if not install_dependencies():
|
49 |
-
print("❌ Не удалось установить зависимости. Пожалуйста, установите их вручную.")
|
50 |
-
return
|
51 |
-
else:
|
52 |
-
print("⚠️ Для работы приложения необходимо установить зависимости.")
|
53 |
-
print("Выполните: pip install -r requirements.txt")
|
54 |
-
return
|
55 |
-
|
56 |
-
# Запускаем приложение
|
57 |
-
run_app()
|
58 |
-
|
59 |
-
if __name__ == "__main__":
|
60 |
main()
|
|
|
1 |
+
import os
|
2 |
+
import sys
|
3 |
+
import subprocess
|
4 |
+
|
5 |
+
def check_dependencies():
|
6 |
+
"""Проверяет, установлены ли все необходимые зависимости"""
|
7 |
+
try:
|
8 |
+
import flask
|
9 |
+
import torch
|
10 |
+
import transformers
|
11 |
+
print("✅ Все необходимые зависимости установлены.")
|
12 |
+
return True
|
13 |
+
except ImportError as e:
|
14 |
+
print(f"❌ Отсутствуют зависимости: {e}")
|
15 |
+
return False
|
16 |
+
|
17 |
+
def install_dependencies():
|
18 |
+
"""Устанавливает необходимые зависимости из requirements.txt"""
|
19 |
+
print("📦 Установка зависимостей...")
|
20 |
+
try:
|
21 |
+
subprocess.check_call([sys.executable, "-m", "pip", "install", "-r", "requirements.txt"])
|
22 |
+
print("✅ Зависимости успешно установлены!")
|
23 |
+
return True
|
24 |
+
except subprocess.CalledProcessError as e:
|
25 |
+
print(f"❌ Ошибка при установке зависимостей: {e}")
|
26 |
+
return False
|
27 |
+
|
28 |
+
def run_app():
|
29 |
+
"""Запускает Flask-приложение"""
|
30 |
+
print("🚀 Запуск AI Assistant...")
|
31 |
+
print("⚠️ Первый запуск может занять некоторое время, пока загружается модель.")
|
32 |
+
try:
|
33 |
+
from app import app
|
34 |
+
app.run(debug=True, port=5000)
|
35 |
+
except Exception as e:
|
36 |
+
print(f"❌ Ошибка при запуске приложения: {e}")
|
37 |
+
|
38 |
+
def main():
|
39 |
+
print("="*50)
|
40 |
+
print("🤖 AI Assistant - Установка и запуск")
|
41 |
+
print("="*50)
|
42 |
+
|
43 |
+
# Проверяем зависимости
|
44 |
+
if not check_dependencies():
|
45 |
+
print("\nНеобходимо установить зависимости.")
|
46 |
+
choice = input("Установить зависимости сейчас? (y/n): ").lower()
|
47 |
+
if choice == 'y':
|
48 |
+
if not install_dependencies():
|
49 |
+
print("❌ Не удалось установить зависимости. Пожалуйста, установите их вручную.")
|
50 |
+
return
|
51 |
+
else:
|
52 |
+
print("⚠️ Для работы приложения необходимо установить зависимости.")
|
53 |
+
print("Выполните: pip install -r requirements.txt")
|
54 |
+
return
|
55 |
+
|
56 |
+
# Запускаем приложение
|
57 |
+
run_app()
|
58 |
+
|
59 |
+
if __name__ == "__main__":
|
60 |
main()
|