Cachoups commited on
Commit
50d3706
·
verified ·
1 Parent(s): 67cafc1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +19 -18
app.py CHANGED
@@ -64,6 +64,9 @@ PDF_FOLDER = "data"
64
  def get_pdf_files(folder):
65
  return [f for f in os.listdir(folder) if f.endswith('.pdf')]
66
 
 
 
 
67
  stored_paragraphs_1 = []
68
  stored_paragraphs_2 = []
69
 
@@ -94,13 +97,6 @@ with gr.Blocks() as demo:
94
  # Process the selected paragraph from PDF 1
95
  with gr.Column():
96
  gr.Markdown("### PDF 1 Analysis")
97
- selected_paragraph_1 = gr.Textbox(label="Selected Paragraph 1 Content", lines=4)
98
- summarize_btn1 = gr.Button("Summarize Text from PDF 1")
99
- sentiment_btn1 = gr.Button("Classify Financial Tone from PDF 1")
100
- summary_textbox_1 = gr.Textbox(label="Summary for PDF 1", lines=4)
101
- sentiment_textbox_1 = gr.Textbox(label="Classification for PDF 1", lines=4)
102
- fin_spans_1 = gr.HighlightedText(label="Financial Tone Analysis for PDF 1")
103
-
104
  def process_paragraph_1(paragraph):
105
  try:
106
  paragraph_index = int(paragraph.split(':')[0].replace('Paragraph ', '')) - 1
@@ -111,22 +107,21 @@ with gr.Blocks() as demo:
111
  return selected_paragraph, summary, sentiment, fin_spans
112
  except (IndexError, ValueError):
113
  return "Invalid selection", "Error", "Error", []
114
-
 
 
115
  summarize_btn1.click(fn=lambda p: process_paragraph_1(p)[1], inputs=paragraph_1_dropdown, outputs=summary_textbox_1)
 
 
116
  sentiment_btn1.click(fn=lambda p: process_paragraph_1(p)[2], inputs=paragraph_1_dropdown, outputs=sentiment_textbox_1)
117
- analyze_btn1 = gr.Button("Analyze Financial Tone and FLS")
 
118
  analyze_btn1.click(fn=lambda p: process_paragraph_1(p)[3], inputs=paragraph_1_dropdown, outputs=fin_spans_1)
 
119
 
120
  # Process the selected paragraph from PDF 2
121
  with gr.Column():
122
  gr.Markdown("### PDF 2 Analysis")
123
- selected_paragraph_2 = gr.Textbox(label="Selected Paragraph 2 Content", lines=4)
124
- summarize_btn2 = gr.Button("Summarize Text from PDF 2")
125
- sentiment_btn2 = gr.Button("Classify Financial Tone from PDF 2")
126
- summary_textbox_2 = gr.Textbox(label="Summary for PDF 2", lines=4)
127
- sentiment_textbox_2 = gr.Textbox(label="Classification for PDF 2", lines=4)
128
- fin_spans_2 = gr.HighlightedText(label="Financial Tone Analysis for PDF 2")
129
-
130
  def process_paragraph_2(paragraph):
131
  try:
132
  paragraph_index = int(paragraph.split(':')[0].replace('Paragraph ', '')) - 1
@@ -137,10 +132,16 @@ with gr.Blocks() as demo:
137
  return selected_paragraph, summary, sentiment, fin_spans
138
  except (IndexError, ValueError):
139
  return "Invalid selection", "Error", "Error", []
140
-
 
 
141
  summarize_btn2.click(fn=lambda p: process_paragraph_2(p)[1], inputs=paragraph_2_dropdown, outputs=summary_textbox_2)
 
 
142
  sentiment_btn2.click(fn=lambda p: process_paragraph_2(p)[2], inputs=paragraph_2_dropdown, outputs=sentiment_textbox_2)
143
- analyze_btn2 = gr.Button("Analyze Financial Tone and FLS")
 
144
  analyze_btn2.click(fn=lambda p: process_paragraph_2(p)[3], inputs=paragraph_2_dropdown, outputs=fin_spans_2)
 
145
 
146
  demo.launch()
 
64
  def get_pdf_files(folder):
65
  return [f for f in os.listdir(folder) if f.endswith('.pdf')]
66
 
67
+ def show(name):
68
+ return f"{name}"
69
+
70
  stored_paragraphs_1 = []
71
  stored_paragraphs_2 = []
72
 
 
97
  # Process the selected paragraph from PDF 1
98
  with gr.Column():
99
  gr.Markdown("### PDF 1 Analysis")
 
 
 
 
 
 
 
100
  def process_paragraph_1(paragraph):
101
  try:
102
  paragraph_index = int(paragraph.split(':')[0].replace('Paragraph ', '')) - 1
 
107
  return selected_paragraph, summary, sentiment, fin_spans
108
  except (IndexError, ValueError):
109
  return "Invalid selection", "Error", "Error", []
110
+ selected_paragraph_1 = gr.Textbox(label="Selected Paragraph 1 Content", lines=4)
111
+ selected_paragraph_1.change(show, paragraph_1_dropdown, selected_paragraph_1)
112
+ summarize_btn1 = gr.Button("Summarize Text from PDF 1")
113
  summarize_btn1.click(fn=lambda p: process_paragraph_1(p)[1], inputs=paragraph_1_dropdown, outputs=summary_textbox_1)
114
+ summary_textbox_1 = gr.Textbox(label="Summary for PDF 1", lines=2)
115
+ sentiment_btn1 = gr.Button("Classify Financial Tone from PDF 1")
116
  sentiment_btn1.click(fn=lambda p: process_paragraph_1(p)[2], inputs=paragraph_1_dropdown, outputs=sentiment_textbox_1)
117
+ sentiment_textbox_1 = gr.Textbox(label="Classification for PDF 1", lines=1)
118
+ analyze_btn1 = gr.Button("Analyze Financial Tone on each sentence")
119
  analyze_btn1.click(fn=lambda p: process_paragraph_1(p)[3], inputs=paragraph_1_dropdown, outputs=fin_spans_1)
120
+ fin_spans_1 = gr.HighlightedText(label="Financial Tone Analysis for PDF 1")
121
 
122
  # Process the selected paragraph from PDF 2
123
  with gr.Column():
124
  gr.Markdown("### PDF 2 Analysis")
 
 
 
 
 
 
 
125
  def process_paragraph_2(paragraph):
126
  try:
127
  paragraph_index = int(paragraph.split(':')[0].replace('Paragraph ', '')) - 1
 
132
  return selected_paragraph, summary, sentiment, fin_spans
133
  except (IndexError, ValueError):
134
  return "Invalid selection", "Error", "Error", []
135
+ selected_paragraph_2 = gr.Textbox(label="Selected Paragraph 2 Content", lines=4)
136
+ selected_paragraph_2.change(show, paragraph_2_dropdown, selected_paragraph_2)
137
+ summarize_btn2 = gr.Button("Summarize Text from PDF 2")
138
  summarize_btn2.click(fn=lambda p: process_paragraph_2(p)[1], inputs=paragraph_2_dropdown, outputs=summary_textbox_2)
139
+ summary_textbox_2 = gr.Textbox(label="Summary for PDF 2", lines=2)
140
+ sentiment_btn2 = gr.Button("Classify Financial Tone from PDF 2")
141
  sentiment_btn2.click(fn=lambda p: process_paragraph_2(p)[2], inputs=paragraph_2_dropdown, outputs=sentiment_textbox_2)
142
+ sentiment_textbox_2 = gr.Textbox(label="Classification for PDF 2", lines=1)
143
+ analyze_btn2 = gr.Button("Analyze Financial Tone on each sentence")
144
  analyze_btn2.click(fn=lambda p: process_paragraph_2(p)[3], inputs=paragraph_2_dropdown, outputs=fin_spans_2)
145
+ fin_spans_2 = gr.HighlightedText(label="Financial Tone Analysis for PDF 2")
146
 
147
  demo.launch()