71 lines
1.3 KiB
Python
71 lines
1.3 KiB
Python
import sys
|
|
|
|
from .main import EmotionPrediction
|
|
from .server import Server
|
|
|
|
|
|
def _real_main(argv=None):
|
|
"""
|
|
The main command-line processor
|
|
|
|
:param argv:
|
|
:return:
|
|
"""
|
|
if argv[1] == '--help':
|
|
_help_menu(argv)
|
|
if argv[1] == '--test':
|
|
_test_model(argv)
|
|
if argv[1] == '--start-server':
|
|
_start_server(argv)
|
|
|
|
|
|
def _start_server(argv=None):
|
|
"""
|
|
Starts the server
|
|
|
|
:param argv:
|
|
:return:
|
|
"""
|
|
server = Server()
|
|
server.start()
|
|
|
|
|
|
def _help_menu(argv=None):
|
|
"""
|
|
Displays the help menu and commandline usage
|
|
|
|
:param argv:
|
|
:return:
|
|
"""
|
|
print(
|
|
"CoffeeHouse EmotionPrediction CLI\n\n"
|
|
" --help\n"
|
|
" --test\n"
|
|
" --start-server\n"
|
|
)
|
|
sys.exit()
|
|
|
|
|
|
def _test_model(argv=None):
|
|
"""
|
|
Tests the model's prediction by allowing user input and displaying the
|
|
prediction output
|
|
|
|
:param argv:
|
|
:return:
|
|
"""
|
|
print("Loading")
|
|
emotion_prediction = EmotionPrediction()
|
|
print("Ready\n")
|
|
|
|
while True:
|
|
input_text = input("> ")
|
|
print(emotion_prediction.predict(input_text))
|
|
|
|
|
|
if __name__ == '__main__':
|
|
try:
|
|
_real_main(sys.argv)
|
|
except KeyboardInterrupt:
|
|
print('\nInterrupted by user')
|