Spaces:
Runtime error
Runtime error
import gradio as gr | |
import time | |
from utils import Bot | |
from utils import make_documens, make_descriptions | |
def init_bot(table=None,tittle=None,pdf=None,key=None): | |
if key is None: | |
return 'You must submit OpenAI key' | |
if pdf is None: | |
return 'You must submit pdf file' | |
if table is None: | |
return 'You must submit table file' | |
if tittle is None: | |
return 'You must submit tittle of table' | |
table = table.name | |
pdf = pdf.name | |
table_descriptions = make_descriptions(tittle=tittle,table=table) | |
documents = make_documens(pdf) | |
print(documents[0]) | |
global bot | |
bot = Bot( | |
openai_api_key=key, | |
table_descriptions=table_descriptions, | |
text_documents=documents, | |
verbose=False | |
) | |
return 'Chat bot successfully initialized' | |
def msg_bot(history): | |
message = history[-1][0] | |
bot_message = bot(message)['output'] | |
history[-1][1] = "" | |
for character in bot_message: | |
history[-1][1] += character | |
time.sleep(0.05) | |
yield history | |
def user(user_message, history): | |
return "", history + [[user_message, None]] | |
with gr.Blocks() as demo: | |
with gr.Tab("Chat bot initialization"): | |
key = gr.Textbox(label='OpenAI key') | |
with gr.Row(variant='panel'): | |
with gr.Column(): | |
with gr.Row(): | |
title = gr.Textbox(label='Table tittle') | |
with gr.Row(): | |
table = gr.File(label='csv table') | |
pdf = gr.File(label='pdf') | |
with gr.Row(variant='panel'): | |
init_button = gr.Button('submit') | |
init_output = gr.Textbox(label="Initialization status") | |
init_button.click(fn=init_bot,inputs=[table,title,pdf,key],outputs=init_output,api_name='init') | |
with gr.Tab("Chat bot"): | |
chatbot = gr.Chatbot() | |
msg = gr.Textbox(label='Ask the bot') | |
clear = gr.Button('Clear') | |
msg.submit(user,[msg,chatbot],[msg,chatbot],queue=False).then( | |
msg_bot, chatbot, chatbot | |
) | |
clear.click(lambda: None, None, chatbot, queue=False) | |
demo.queue() | |
demo.launch() | |