Marty/simple-summary
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Default to single line transcript

Browse source
This commit is contained in:
Marty Oehme 2025-01-18 14:34:02 +01:00
parent 5be71fc7da
commit 4df67bd475
Signed by: Marty
GPG key ID: EDBF2ED917B2EF6A
1 changed files with 4 additions and 2 deletions

6
summarize.py
View file

@ -13,7 +13,7 @@ def summarize_text_file(content: str, prompt: str | None) -> str:
return bot.chat(f"{prompt} {content}") return bot.chat(f"{prompt} {content}")
def extract_transcript_contents(content: str) -> str: def extract_transcript_contents(content: str, keep_newlines: bool = False) -> str:
jq_command = "jq '.events.[].segs | select(. != null).[].utf8'" jq_command = "jq '.events.[].segs | select(. != null).[].utf8'"
result = subprocess.run( result = subprocess.run(
jq_command, shell=True, capture_output=True, text=True, input=content jq_command, shell=True, capture_output=True, text=True, input=content
@ -22,7 +22,9 @@ def extract_transcript_contents(content: str) -> str:
result = result.replace("\n", "").split() result = result.replace("\n", "").split()
# Join lines back together with newlines # Join lines back together with newlines
processed = " ".join(result).replace('"', "").replace("\\n", "\n") processed = (
" ".join(result).replace('"', "").replace("\\n", "\n" if keep_newlines else " ")
)
return processed return processed