File size: 826 Bytes
02a466f |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 |
from interface import create_demo
from pdfchatbot import PDFChatBot
# Create Gradio interface
demo, chat_history, show_img, txt, submit_button, uploaded_pdf = create_demo()
# Create PDFChatBot instance
pdf_chatbot = PDFChatBot()
# Set up event handlers
with demo:
# Event handler for uploading a PDF
uploaded_pdf.upload(pdf_chatbot.render_file, inputs=[uploaded_pdf], outputs=[show_img])
# Event handler for submitting text and generating response
submit_button.click(pdf_chatbot.add_text, inputs=[chat_history, txt], outputs=[chat_history], queue=False).\
success(pdf_chatbot.generate_response, inputs=[chat_history, txt, uploaded_pdf], outputs=[chat_history, txt]).\
success(pdf_chatbot.render_file, inputs=[uploaded_pdf], outputs=[show_img])
if __name__ == "__main__":
demo.launch() |