run_backend.py

from run_gui import *
from run_api import *
from multiprocessing import Process
import argparse
def parse_args_port():
    parser = argparse.ArgumentParser()
    parser.add_argument("--port", default=7861, help="port for the api_backend")
    parser.add_argument("--services", default='all', help="what services to run")
    args = parser.parse_known_args()
    return args[0].port, args[0].services

def run_gradio():
    with warnings.catch_warnings():
        warnings.simplefilter("ignore")
        front_page.launch(
            server_name="0.0.0.0",
            auth=("wadset", "wadset"),
            # server_port=9085,
            show_api=False
        )

def run_flask(port):
    api_backend.run(host="0.0.0.0",port=port, debug=False)

if __name__ == "__main__":
    with warnings.catch_warnings():
        warnings.simplefilter("ignore")
        flask_port , services = parse_args_port()
        
        if services == 'all':
            flask_process = Process(target=run_flask,args=(flask_port,))
        else:
            flask_process = None

        try:
            if flask_process:
                # Starting the processes
                flask_process.start()
            run_gradio()
        except Exception as e:
            print(e)
        finally:            
            
            if flask_process:
                # gradio_process.join()
                flask_process.join()