Spaces:
Sleeping
Sleeping
import random | |
import time | |
import gradio as gr | |
from matt_function import create_VSI | |
logo_filepath = 'Readable.png' | |
robot_filepath = 'Robot.png' | |
custom_css = """ | |
.column { | |
align-items: center; | |
} | |
.column_el { | |
flex-grow: 1; | |
} | |
""" | |
def respond(message, chat_history, index): | |
bot_message = index.query(message) | |
chat_history.append((message, bot_message.response)) | |
return "", chat_history, display_question_and_answer(chat_history) | |
def display_question_and_answer(tuple_list): | |
qa_history = "" | |
for i, (q, a) in enumerate(tuple_list): | |
qa_history += f"<details><summary>Q: {q}</summary><p>A: {a}</p></details>" | |
return qa_history | |
with gr.Blocks(css=custom_css) as demo: | |
with gr.Tab("Home"): | |
with gr.Row(variant='compact'): | |
with gr.Column(scale=1, elem_classes='column'): | |
logo = gr.Image(value=logo_filepath, type='filepath', show_label=False, show_download_button=False, container=False, elem_classes='column_el') | |
VSidx = gr.State() | |
files = gr.File(file_count='multiple', show_label=False, elem_classes='column_el', height=75) | |
# name = gr.Textbox(label="Name", elem_classes='column_el') | |
# description = gr.Textbox(label="Description", elem_classes='column_el') | |
create_VSI_btn = gr.Button(value='Submit', elem_classes='column_el') | |
upload_status = gr.Textbox(label='Upload Status', value='No PDF uploaded') | |
create_VSI_btn.click(fn=create_VSI, inputs = [files, VSidx], outputs = [VSidx, upload_status]) | |
robot = gr.Image(value=robot_filepath, type='filepath', show_label=False, show_download_button=False, container=False, width=125, height=125, elem_classes='column_el') | |
with gr.Column(scale=3, elem_classes='column'): | |
chatbot = gr.Chatbot(elem_classes='column_el') | |
msg = gr.Textbox(show_label=False, elem_classes='column_el') | |
with gr.Tab("Chat History"): | |
history = gr.HTML() | |
msg.submit(respond, [msg, chatbot, VSidx], [msg, chatbot, history]) | |
demo.launch() | |