run.py - master - sofnuk2004/server1

run.py
88 строк3 KB
Sofya Nikolaeva
изменения пробьую заново
03 мар 2026, 17:53
03 мар 2026, 17:536bbc67f
from app import create_app, db
from app.db_models import User, Patient, Doctor, ModelInfo
import os
import shutil
import sys

app = create_app()

@app.shell_context_processor
def make_shell_context():
    return {'db': db, 'User': User, 'Patient': Patient, 'Doctor': Doctor, 'ModelInfo': ModelInfo}

if __name__ == '__main__':
    print("=" * 50)
    print("Skin Diagnosis Server")
    print("=" * 50)
    print(f"Host: {app.config['HOST']}")
    print(f"Port: {app.config['PORT']}")
    print(f"Debug: {app.config['DEBUG']}")
    print(f"Database: {app.config['SQLALCHEMY_DATABASE_URI']}")
    print(f"Schema: {app.config.get('DB_SCHEMA', 'public')}")
    print("=" * 50)
    print("Available endpoints:")
    print(f"http://{app.config['HOST']}:{app.config['PORT']}/")
    print(f"http://{app.config['HOST']}:{app.config['PORT']}/health")
    print(f"http://{app.config['HOST']}:{app.config['PORT']}/test")
    print("=" * 50)
    
    # Создаем необходимые директории с обработкой ошибок
    try:
        upload_folder = app.config['UPLOAD_FOLDER']
        models_folder = app.config['MODELS_FOLDER']
        
        # Проверяем и создаем директории
        if not os.path.exists(upload_folder):
            os.makedirs(upload_folder, exist_ok=True)
            print(f" Created upload folder: {upload_folder}")
        else:
            print(f" Upload folder already exists: {upload_folder}")
            
        if not os.path.exists(models_folder):
            os.makedirs(models_folder, exist_ok=True)
            print(f" Created models folder: {models_folder}")
        else:
            print(f" Models folder already exists: {models_folder}")
            
    except Exception as e:
        print(f" Warning: Could not create directories: {e}")
    
    # Копируем модели из папки VKR/models в static/models
    source_models_dir = os.path.join(os.path.dirname(__file__), 'VKR', 'models')
    target_models_dir = app.config['MODELS_FOLDER']
    
    print(f"\nLooking for models in: {source_models_dir}")
    
    if os.path.exists(source_models_dir):
        for model_dir in os.listdir(source_models_dir):
            src_path = os.path.join(source_models_dir, model_dir)
            dst_path = os.path.join(target_models_dir, model_dir)
            
            if os.path.isdir(src_path):
                if not os.path.exists(dst_path):
                    try:
                        shutil.copytree(src_path, dst_path)
                        print(f" Copied model: {model_dir}")
                    except Exception as e:
                        print(f" Error copying {model_dir}: {e}")
                else:
                    print(f" Model already exists: {model_dir}")
    else:
        print(f" Models directory not found: {source_models_dir}")
        print("   Please make sure models are in VKR/models/")
    
    print("\n" + "=" * 50)
    print(" Starting server...")
    print("=" * 50 + "\n")
    
    # Запускаем сервер
    try:
        app.run(
            host=app.config['HOST'],
            port=app.config['PORT'],
            debug=app.config['DEBUG'],
            threaded=True
        )
    except Exception as e:
        print(f" Failed to start server: {e}")
        sys.exit(1)