MAGI/main.py

from dataclasses import dataclass
import os
from dash_extensions.enrich import Dash, Input, Output, State, Trigger, callback, ALL, MATCH, callback_context
import dash_core_components as dcc
import dash_html_components as html
from dash_local_react_components import load_react_component
import ai

app = Dash(__name__)

Magi = load_react_component(app, 'components', 'magi.js')
WiseMan = load_react_component(app, 'components', 'wise_man.js')
Response = load_react_component(app, 'components', 'response.js')
Modal = load_react_component(app, 'components', 'modal.js')

app.layout = html.Div(
    className='system',
    children=[
        Magi(id='magi', children=[
              html.Div(className='connection casper-balthasar'),
              html.Div(className='connection casper-melchior'),
              html.Div(className='connection balthasar-melchior'),
              html.Div(className='header left', children=[
                  html.Hr(),
                  html.Hr(),
                  html.Span('質 問 '),
                  html.Hr(),
                  html.Hr()
              ]),
            html.Div(className='header right', children=[
                html.Hr(),
                html.Hr(),
                html.Span('解 決 '),
                html.Hr(),
                html.Hr()
            ]),
            html.Div(
                  className='system-status',
                  children=[
                      html.Div(children='CODE:473'),
                      html.Div(children='FILE:MAGI_SYS'),
                      html.Div(id='extention', children='EXTENTION:????'),
                      html.Div(children='EX_MODE:OFF'),
                      html.Div(children='PRIORITY:AAA')]),
            WiseMan(
                  id={'type': 'wise-man', 'name': 'melchior'},
                  name='melchior',
                  order_number=1,
                  personality='You are a scientist. Your goal is to further our understanding of the universe and advance our technological progress.',
                  question_id=0,
                  answer={'id': 0, 'response': 'yes', 'status': 'yes'}),
            WiseMan(
                  id={'type': 'wise-man', 'name': 'balthasar'},
                  name='balthasar',
                  order_number=2,
                  personality='You are a mother. Your goal is to protect your children and ensure their well-being.',
                  question_id=0,
                  answer={'id': 0, 'response': 'yes', 'status': 'yes'}),
            WiseMan(
                  id={'type': 'wise-man', 'name': 'casper'},
                  name='casper',
                  order_number=3,
                  personality='You are a woman. Your goal is to pursue love, dreams and desires.',
                  question_id=0,
                  answer={'id': 0, 'response': 'yes', 'status': 'yes'}),
            Response(id='response', status='info'),
            html.Div(className='title', children='MAGI')
        ]),
        html.Div(className='input-container', children=[
            html.Label('access code: '),
            dcc.Input(id='key', autoComplete='off', type='password', value=os.getenv('OPENAI_API_KEY', '')),
            html.Label('question: '),
            dcc.Input(id='query', type='text', value='', debounce=True, autoComplete='off'),
        ]),
        Modal(id={'type': 'modal', 'name': 'melchior'}, name='melchior'),
        Modal(id={'type': 'modal', 'name': 'balthasar'}, name='balthasar'),
        Modal(id={'type': 'modal', 'name': 'casper'}, name='casper'),

        dcc.Store(id='question', data={'id': 0, 'query': ''}),
        dcc.Store(id='annotated-question', data={'id': 0, 'query': '', 'is_yes_or_no_question': False}),
        dcc.Store(id='is_yes_or_no_question', data=False),
        dcc.Store(id='question-id', data=0),
    ])


@callback(
    Output('question', 'data'),
    Input('query', 'value'),
    State('question', 'data'),
    prevent_initial_call=True)
def question(query: str, question: dict):
    print('question')

    return {'id': question['id'] + 1, 'query': query}


@callback(
    Output('annotated-question', 'data'),
    Input('question', 'data'),
    State('key', 'value'),
    prevent_initial_call=True)
def annotated_question(question: dict, key: str):
    print('annotated_question')

    try:
        is_yes_or_no_question = ai.is_yes_or_no_question(question['query'], key)

        print(is_yes_or_no_question)

        return {
            'id': question['id'],
            'query': question['query'],
            'is_yes_or_no_question': is_yes_or_no_question,
            'error': None
        }
    except Exception as e:
        print(e)
        return {
            'id': question['id'],
            'query': question['query'],
            'is_yes_or_no_question': False,
            'error': str(e)
        }


@callback(
    Output('extention', 'children'),
    Input('question', 'data'),
    Input('annotated-question', 'data'))
def extention(question: dict, annotated_question: dict):
    if question['id'] != annotated_question['id']:
        return 'EXTENTION:????'

    code = '7312' if annotated_question['is_yes_or_no_question'] else '3023'
    return f'EXTENTION:{code}'


@callback(
    Output({'type': 'wise-man', 'name': MATCH}, 'answer'),
    Input('annotated-question', 'data'),
    State({'type': 'wise-man', 'name': MATCH}, 'personality'),
    State('key', 'value'),
    prevent_initial_call=True)
def wise_man_answer(question: dict, personality: str, key: str):
    print('wise_man_answer')

    if question['error']:
        return {'id': question['id'], 'response': question['error'], 'status': 'error'}

    try:
        answer = ai.get_answer(question['query'], personality, key)
        print(answer)

        if question['is_yes_or_no_question']:
            classification = ai.classify_answer(question['query'], personality, answer, key)
        else:
            classification = {'status': 'info', 'conditions': None}

        print(classification)

        return {'id': question['id'], 'response': answer, 'status': classification['status'], 'conditions': classification['conditions'], 'error': None}

    except Exception as e:
        print(e)
        return {'id': question['id'], 'response': None, 'status': 'error', 'conditions': 'None', 'error': str(e)}


@callback(
    Output({'type': 'wise-man', 'name': MATCH}, 'question_id'),
    Input('question', 'data'))
def question_id(question: dict):
    return question['id']


@callback(
    Output('response', 'status'),
    Input({'type': 'wise-man', 'name': ALL}, 'answer'))
def response_status(answers: list):
    if all([answer['status'] == 'yes' for answer in answers]):
        return 'yes'
    elif any([answer['status'] == 'no' for answer in answers]):
        return 'no'
    else:
        return 'info'


@callback(
    Output({'type': 'modal', 'name': MATCH}, 'is_open'),
    Trigger({'type': 'wise-man', 'name': MATCH}, 'n_clicks'),
    prevent_initial_call=True)
def modal():
    return True


@callback(
    Output({'type': 'modal', 'name': MATCH}, 'question'),
    Output({'type': 'modal', 'name': MATCH}, 'answer'),
    Input('question', 'data'),
    Input({'type': 'wise-man', 'name': MATCH}, 'answer'))
def modal(question: dict, answer: dict):
    return question, answer


if __name__ == '__main__':
    app.run_server(debug=True)