Update src/app.py
Browse files- src/app.py +6 -5
src/app.py
CHANGED
@@ -1,9 +1,9 @@
|
|
1 |
-
import csv
|
2 |
-
import gradio as gr
|
3 |
from interface import create_demo
|
4 |
from pdfchatbot import PDFChatBot
|
|
|
|
|
5 |
|
6 |
-
demo, chat_history, show_img, txt, submit_button, uploaded_pdf,
|
7 |
|
8 |
pdf_chatbot = PDFChatBot()
|
9 |
|
@@ -21,10 +21,11 @@ with demo:
|
|
21 |
submit_button.click(pdf_chatbot.add_text, inputs=[chat_history, txt], outputs=[chat_history], queue=False).\
|
22 |
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]).\
|
23 |
success(pdf_chatbot.render_file, inputs=[uploaded_pdf], outputs=[show_img])
|
24 |
-
|
25 |
download_button = gr.Button("Download Chat History")
|
26 |
-
download_output = gr.
|
27 |
download_button.click(fn=download_chat_history, inputs=[chat_history], outputs=download_output)
|
28 |
|
|
|
29 |
if __name__ == "__main__":
|
30 |
demo.launch()
|
|
|
|
|
|
|
1 |
from interface import create_demo
|
2 |
from pdfchatbot import PDFChatBot
|
3 |
+
import csv
|
4 |
+
import gradio as gr
|
5 |
|
6 |
+
demo, chat_history, show_img, txt, submit_button, uploaded_pdf, slider_chunk_size,slider_overlap_percentage,slider_temp,slider_k = create_demo()
|
7 |
|
8 |
pdf_chatbot = PDFChatBot()
|
9 |
|
|
|
21 |
submit_button.click(pdf_chatbot.add_text, inputs=[chat_history, txt], outputs=[chat_history], queue=False).\
|
22 |
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]).\
|
23 |
success(pdf_chatbot.render_file, inputs=[uploaded_pdf], outputs=[show_img])
|
24 |
+
|
25 |
download_button = gr.Button("Download Chat History")
|
26 |
+
download_output = gr.File()
|
27 |
download_button.click(fn=download_chat_history, inputs=[chat_history], outputs=download_output)
|
28 |
|
29 |
+
|
30 |
if __name__ == "__main__":
|
31 |
demo.launch()
|