chat_bot/ui_simple_chat.py

import gradio as gr
import pandas as pd
from prompts import system_prompts, user_prompts
from utilities import constants
from chat_bot import simple_chat
import random
import time

def InitDF():
    global gitDF
    gitDF=pd.DataFrame({"role": [""], "content": [""] })
    simple_chat.CreateEmbeddings(constants.POLICY_INPUT, constants.POLICY_OUTPUT)

def Respond(message, chat_history):
    bot_message,df=simple_chat.QueryEmbeddingsSimple(message)
    chat_history.append((message, bot_message))
    return "", chat_history,df

with gr.Blocks() as ui:
    label = gr.Label(show_label=False, value=constants.UI_7, container=False)
    with gr.Row():
        with gr.Column():
            gr.Markdown(constants.PURPOSE_MD)
            gr.Markdown(constants.AR_ADVISOR_PURPOSE)
        with gr.Column():
            gr.Markdown(constants.DIRECTIONS_MD)
            gr.Markdown(constants.AR_ADVISOR_DIRECTIONS)
    question= gr.Textbox (label=constants.QUESTIONS_PREFIX, value=constants.QUESTIONS_AR_EXPERT)
    chatbot = gr.Chatbot(label=constants.CHAT_BOT, height=constants.CHAT_BOT_HEIGHT)
    msg = gr.Textbox(label=constants.CHAT_BOT_INPUT)    
    gitDF = gr.DataFrame(type="pandas", value=pd.DataFrame({"role": [""], "content": [""] }), wrap=True, label=constants.OPENAI_LOG)
    clear = gr.ClearButton([msg, chatbot])
    msg.submit(Respond, [msg, chatbot], [msg, chatbot,gitDF])

InitDF()