Cachoups commited on
Commit
37af58a
·
verified ·
1 Parent(s): 570b200

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -0
app.py CHANGED
@@ -3,6 +3,7 @@ import gradio as gr
3
  from transformers import pipeline
4
  import spacy
5
  import lib.read_pdf
 
6
  import pandas as pd
7
  import re
8
  import matplotlib.pyplot as plt
@@ -469,6 +470,8 @@ with gr.Blocks(theme='gradio/soft',js=js_func) as demo:
469
 
470
  selected_paragraph_1 = gr.Textbox(label="Selected Paragraph 1 Content", lines=4)
471
  paragraph_1_dropdown.select(fn=show1, inputs = paragraph_1_dropdown, outputs=selected_paragraph_1)
 
 
472
  with gr.Group():
473
  summarize_btn1 = gr.Button("Summarize Text from PDF 1")
474
  summary_textbox_1 = gr.Textbox(label="Summary for PDF 1", lines=2)
@@ -501,6 +504,8 @@ with gr.Blocks(theme='gradio/soft',js=js_func) as demo:
501
 
502
  selected_paragraph_2 = gr.Textbox(label="Selected Paragraph 2 Content", lines=4)
503
  paragraph_2_dropdown.select(fn=show2, inputs = paragraph_2_dropdown, outputs=selected_paragraph_2)
 
 
504
  with gr.Group():
505
  # Display selected paragraph from PDF 2
506
  selected_paragraph_2.change(fn=show2, inputs=paragraph_2_dropdown, outputs=selected_paragraph_2)
 
3
  from transformers import pipeline
4
  import spacy
5
  import lib.read_pdf
6
+ import lib.comparison
7
  import pandas as pd
8
  import re
9
  import matplotlib.pyplot as plt
 
470
 
471
  selected_paragraph_1 = gr.Textbox(label="Selected Paragraph 1 Content", lines=4)
472
  paragraph_1_dropdown.select(fn=show1, inputs = paragraph_1_dropdown, outputs=selected_paragraph_1)
473
+ close_paragraph_1 = gr.Textbox(label="Closest Paragraph 1 Content", lines=4)
474
+ paragraph_1_dropdown.select(fn=lib.comparison.compare_summaries, inputs = [paragraph_1_dropdown, stored_paragraphs_2], outputs=close_paragraph_1)
475
  with gr.Group():
476
  summarize_btn1 = gr.Button("Summarize Text from PDF 1")
477
  summary_textbox_1 = gr.Textbox(label="Summary for PDF 1", lines=2)
 
504
 
505
  selected_paragraph_2 = gr.Textbox(label="Selected Paragraph 2 Content", lines=4)
506
  paragraph_2_dropdown.select(fn=show2, inputs = paragraph_2_dropdown, outputs=selected_paragraph_2)
507
+ close_paragraph_2 = gr.Textbox(label="Closest Paragraph 2 Content", lines=4)
508
+ paragraph_2_dropdown.select(fn=lib.comparison.compare_summaries, inputs = [paragraph_2_dropdown, stored_paragraphs_1], outputs=close_paragraph_2)
509
  with gr.Group():
510
  # Display selected paragraph from PDF 2
511
  selected_paragraph_2.change(fn=show2, inputs=paragraph_2_dropdown, outputs=selected_paragraph_2)