| from interface import create_demo | |
| from pdfchatbot import PDFChatBot | |
| demo, chat_history, show_img, txt, submit_button, uploaded_pdf, slider_chunk_size,slider_overlap_percentage,slider_temp,slider_k = create_demo() | |
| pdf_chatbot = PDFChatBot() | |
| with demo: | |
| uploaded_pdf.upload(pdf_chatbot.render_file, inputs=[uploaded_pdf,slider_chunk_size,slider_overlap_percentage,slider_temp,slider_k], outputs=[show_img]) | |
| 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,slider_chunk_size,slider_overlap_percentage,slider_temp,slider_k], outputs=[chat_history,txt]).\ | |
| success(pdf_chatbot.render_file, inputs=[uploaded_pdf], outputs=[show_img]) | |
| if __name__ == "__main__": | |
| demo.launch() |