File size: 847 Bytes
ecea303
 
3844992
ecea303
 
3844992
ecea303
 
3844992
ecea303
 
 
 
3844992
ecea303
 
 
 
3844992
 
54e7e68
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(debug=True,share=True)