server: reformat whole project using black

This commit is contained in:
Mathieu Virbel
2023-07-27 14:08:41 +02:00
parent 314321c603
commit 094ed696c4
12 changed files with 406 additions and 237 deletions

View File

@@ -17,6 +17,7 @@ class TitleSummaryInput:
Data class for the input to generate title and summaries.
The outcome will be used to send query to the LLM for processing.
"""
input_text = str
transcribed_time = float
prompt = str
@@ -25,8 +26,7 @@ class TitleSummaryInput:
def __init__(self, transcribed_time, input_text=""):
self.input_text = input_text
self.transcribed_time = transcribed_time
self.prompt = \
f"""
self.prompt = f"""
### Human:
Create a JSON object as response.The JSON object must have 2 fields:
i) title and ii) summary.For the title field,generate a short title
@@ -47,6 +47,7 @@ class IncrementalResult:
Data class for the result of generating one title and summaries.
Defines how a single "topic" looks like.
"""
title = str
description = str
transcript = str
@@ -65,6 +66,7 @@ class TitleSummaryOutput:
Data class for the result of all generated titles and summaries.
The result will be sent back to the client
"""
cmd = str
topics = List[IncrementalResult]
@@ -77,10 +79,7 @@ class TitleSummaryOutput:
Return the result dict for displaying the transcription
:return:
"""
return {
"cmd": self.cmd,
"topics": self.topics
}
return {"cmd": self.cmd, "topics": self.topics}
@dataclass
@@ -89,6 +88,7 @@ class ParseLLMResult:
Data class to parse the result returned by the LLM while generating title
and summaries. The result will be sent back to the client.
"""
title = str
description = str
transcript = str
@@ -98,8 +98,7 @@ class ParseLLMResult:
self.title = output["title"]
self.transcript = param.input_text
self.description = output.pop("summary")
self.timestamp = \
str(datetime.timedelta(seconds=round(param.transcribed_time)))
self.timestamp = str(datetime.timedelta(seconds=round(param.transcribed_time)))
def get_result(self) -> dict:
"""
@@ -107,10 +106,10 @@ class ParseLLMResult:
:return:
"""
return {
"title": self.title,
"description": self.description,
"transcript": self.transcript,
"timestamp": self.timestamp
"title": self.title,
"description": self.description,
"transcript": self.transcript,
"timestamp": self.timestamp,
}
@@ -120,6 +119,7 @@ class TranscriptionInput:
Data class to define the input to the transcription function
AudioFrames -> input
"""
frames = List[av.audio.frame.AudioFrame]
def __init__(self, frames):
@@ -132,6 +132,7 @@ class TranscriptionOutput:
Dataclass to define the result of the transcription function.
The result will be sent back to the client
"""
cmd = str
result_text = str
@@ -144,10 +145,7 @@ class TranscriptionOutput:
Return the result dict for displaying the transcription
:return:
"""
return {
"cmd": self.cmd,
"text": self.result_text
}
return {"cmd": self.cmd, "text": self.result_text}
@dataclass
@@ -156,6 +154,7 @@ class FinalSummaryResult:
Dataclass to define the result of the final summary function.
The result will be sent back to the client.
"""
cmd = str
final_summary = str
duration = str
@@ -171,9 +170,9 @@ class FinalSummaryResult:
:return:
"""
return {
"cmd": self.cmd,
"duration": self.duration,
"summary": self.final_summary
"cmd": self.cmd,
"duration": self.duration,
"summary": self.final_summary,
}
@@ -182,9 +181,14 @@ class BlackListedMessages:
Class to hold the blacklisted messages. These messages should be filtered
out and not sent back to the client as part of the transcription.
"""
messages = [" Thank you.", " See you next time!",
" Thank you for watching!", " Bye!",
" And that's what I'm talking about."]
messages = [
" Thank you.",
" See you next time!",
" Thank you for watching!",
" Bye!",
" And that's what I'm talking about.",
]
@dataclass