main.py

from src.MainNode.Communication.communication import *
import settings
import socket
import threading
from src.SocialStudy import questions
from src.Pi.PiBot.PiBot2 import PiBot2
"""
determine if main or pi
"""
hostname = socket.gethostname()
print(hostname)
if hostname in settings.PI_HOST_NAMES:
    IS_PI = True
    # from src.DRL.run_session import train_and_test_bot
    from src.DRL.train import train
    from src.DRL.test import test
elif hostname == settings.PC:
    IS_PI = False
    # from src.MainNode.Utils.OverHead import OverHead
    # from src.MainNode.Utils.stream_oh_inference import run
else:
    raise Exception("Must configure this device in settings.py")

def main_train_and_test():
    if not IS_PI: # main node
        #oh_stream = OverHead()
        # model = ('ssd_mobilenet_v2_320x320_coco17_tpu-8', 3)
        # run(model, oh_stream)
        threading.Thread(target=subscribe, args=()).start()
        threading.Thread(target=questions.get_user_input, args=()).start()
    else:
        pibot = PiBot2()
        steps = 200
        train.train_session(pibot, steps, continue_prompt=True)
        # publish_data(actions)
        test.run_policy(pibot, steps, continue_prompt=True)




if __name__ == "__main__":
    # main()
    main_train_and_test()