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": [""] })
    ### changing to input -> pdf, output -> csv
    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()