Query_PDFs / app.py
Dylan-Kaneshiro's picture
Update app.py
b7cfea3
raw
history blame contribute delete
No virus
2.05 kB
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()