Spaces:
Running
Running
Peter
commited on
Commit
•
079d1ca
1
Parent(s):
34de38e
🎨 format
Browse filesSigned-off-by: Peter <74869040+pszemraj@users.noreply.github.com>
app.py
CHANGED
@@ -299,11 +299,11 @@ if __name__ == "__main__":
|
|
299 |
)
|
300 |
|
301 |
text_file = gr.File(
|
302 |
-
|
303 |
-
|
304 |
-
|
305 |
-
|
306 |
-
|
307 |
|
308 |
gr.Markdown("---")
|
309 |
with gr.Column():
|
|
|
299 |
)
|
300 |
|
301 |
text_file = gr.File(
|
302 |
+
label="Download Summary as Text File",
|
303 |
+
file_count="single",
|
304 |
+
type="file",
|
305 |
+
interactive=False,
|
306 |
+
)
|
307 |
|
308 |
gr.Markdown("---")
|
309 |
with gr.Column():
|
utils.py
CHANGED
@@ -8,13 +8,16 @@ from datetime import datetime
|
|
8 |
from natsort import natsorted
|
9 |
import subprocess
|
10 |
|
11 |
-
|
|
|
12 |
"""
|
13 |
get_timestamp - get a timestamp for the current time
|
14 |
Returns:
|
15 |
str, the timestamp
|
16 |
"""
|
17 |
return datetime.now().strftime("%Y%m%d_%H%M%S")
|
|
|
|
|
18 |
def truncate_word_count(text, max_words=512):
|
19 |
"""
|
20 |
truncate_word_count - a helper function for the gradio module
|
@@ -74,7 +77,8 @@ def load_example_filenames(example_path: str or Path):
|
|
74 |
examples = {f.name: f for f in example_path.glob("*.txt")}
|
75 |
return examples
|
76 |
|
77 |
-
|
|
|
78 |
"""
|
79 |
|
80 |
saves_summary - save the summary generated from summarize_via_tokenbatches() to a text file
|
@@ -87,8 +91,11 @@ def saves_summary(summarize_output, outpath:str or Path=None, add_signature=True
|
|
87 |
)
|
88 |
"""
|
89 |
|
90 |
-
|
91 |
-
|
|
|
|
|
|
|
92 |
sum_text = [s["summary"][0] for s in summarize_output]
|
93 |
sum_scores = [f"\n - {round(s['summary_score'],4)}" for s in summarize_output]
|
94 |
scores_text = "\n".join(sum_scores)
|
|
|
8 |
from natsort import natsorted
|
9 |
import subprocess
|
10 |
|
11 |
+
|
12 |
+
def get_timestamp() -> str:
|
13 |
"""
|
14 |
get_timestamp - get a timestamp for the current time
|
15 |
Returns:
|
16 |
str, the timestamp
|
17 |
"""
|
18 |
return datetime.now().strftime("%Y%m%d_%H%M%S")
|
19 |
+
|
20 |
+
|
21 |
def truncate_word_count(text, max_words=512):
|
22 |
"""
|
23 |
truncate_word_count - a helper function for the gradio module
|
|
|
77 |
examples = {f.name: f for f in example_path.glob("*.txt")}
|
78 |
return examples
|
79 |
|
80 |
+
|
81 |
+
def saves_summary(summarize_output, outpath: str or Path = None, add_signature=True):
|
82 |
"""
|
83 |
|
84 |
saves_summary - save the summary generated from summarize_via_tokenbatches() to a text file
|
|
|
91 |
)
|
92 |
"""
|
93 |
|
94 |
+
outpath = (
|
95 |
+
Path.cwd() / f"document_summary_{get_timestamp()}.txt"
|
96 |
+
if outpath is None
|
97 |
+
else Path(outpath)
|
98 |
+
)
|
99 |
sum_text = [s["summary"][0] for s in summarize_output]
|
100 |
sum_scores = [f"\n - {round(s['summary_score'],4)}" for s in summarize_output]
|
101 |
scores_text = "\n".join(sum_scores)
|