Commit 5d7aec99 authored by dmMaze's avatar dmMaze
Browse files

update headless mode

parent 268b2ade

launch.py

+5 −5
Original line number Diff line number Diff line
@@ -126,7 +126,6 @@ def main(): 


    if args.debug:

        os.environ['BALLOONTRANS_DEBUG'] = '1'

    os.environ['BT_HEADLESS'] = '1' if args.headless else '0'



    if not args.qt_api in QT_APIS:

        os.environ['QT_API'] = 'pyqt6'
@@ -154,11 +153,15 @@ def main(): 


    from qtpy.QtCore import QTranslator, QLocale, Qt

    shared.DEFAULT_DISPLAY_LANG = QLocale.system().name().replace('en_CN', 'zh_CN')



    shared.HEADLESS = args.headless

    shared.load_cache()

    program_config.load_config()

    config = program_config.pcfg



    if args.headless:

        config.module.load_model_on_demand = True

        config.module.empty_runcache = False



    from modules.prepare_local_files import prepare_local_files_forall

    prepare_local_files_forall()
@@ -230,9 +233,6 @@ def main(): 


    from ui.mainwindow import MainWindow



    if args.headless:

        config.module.load_model_on_demand = True

        config.module.empty_runcache = True

    ballontrans = MainWindow(app, config, open_dir=args.proj_dir, **vars(args))

    global BT

    BT = ballontrans

ui/mainwindow.py

+4 −5
Original line number Diff line number Diff line
@@ -53,8 +53,7 @@ class PageListView(QListWidget): 


        return super().contextMenuEvent(e)



RUN_HEADLESS = os.environ['BT_HEADLESS'] == '1'

mainwindow_cls = Widget if RUN_HEADLESS else FramelessWindow

mainwindow_cls = Widget if shared.HEADLESS else FramelessWindow

class MainWindow(mainwindow_cls):



    imgtrans_proj: ProjImgTrans = ProjImgTrans()
@@ -88,7 +87,7 @@ class MainWindow(mainwindow_cls): 
                if osp.exists(proj_dir):

                    self.OpenProj(proj_dir)



        if RUN_HEADLESS:

        if shared.HEADLESS:

            self.run_batch(**exec_args)



    def setStyleSheet(self, styleSheet: str) -> None:
@@ -839,9 +838,9 @@ class MainWindow(mainwindow_cls): 


    def on_imgtrans_pipeline_finished(self):

        self.postprocess_mt_toggle = True

        if pcfg.module.empty_runcache and not RUN_HEADLESS:

        if pcfg.module.empty_runcache and not shared.HEADLESS:

            self.module_manager.unload_all_models()

        if RUN_HEADLESS:

        if shared.HEADLESS:

            self.run_next_dir()



    def postprocess_translations(self, blk_list: List[TextBlock]) -> None:

utils/shared.py

+1 −1
Original line number Diff line number Diff line
@@ -82,7 +82,7 @@ DEFAULT_DISPLAY_LANG = 'English' 
USE_PYSIDE6 = False

ON_MACOS = sys.platform == 'darwin'

ON_WINDOWS = sys.platform == 'win32'



HEADLESS = False

DEBUG = False



cache_data: Dict = None