| import gradio as gr |
| import pandas as pd |
| from utilities import constants |
| from chat_bot import simple_chat_git |
|
|
| def InitDF(): |
| global gitDF |
| gitDF=pd.DataFrame({"role": [""], "content": [""] }) |
| |
| simple_chat_git.CreateEmbeddings(constants.POLICY_INPUT, constants.POLICY_OUTPUT) |
|
|
| def Respond(message, chat_history): |
| bot_message,df=simple_chat_git.QueryEmbeddingsSimple(message) |
| chat_history.append((message, bot_message)) |
| return "", chat_history,df |
|
|
| with gr.Blocks() as ui: |
| label = gr.Label(show_label=False, value="GIT CHAT", container=False) |
| 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() |
