Spaces:
Running
on
Zero
Running
on
Zero
File size: 3,089 Bytes
7a4927a |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 |
from src.data.chapters import Chapters, sec_to_hms
class Prompt:
def __init__(
self,
chapters: Chapters,
):
self.chapters = chapters
def __contains__(self, vid_id):
raise NotImplementedError(
"Subclasses must implement the '__contains__' method."
)
def get_duration_prompt(self, vid_id: str) -> str:
duration = self.chapters.get_duration(vid_id, hms=True)
return f"Given the complete transcript of a video of duration {duration}, "
def get_task_prompt(self) -> str:
raise NotImplementedError(
"Subclasses must implement the 'get_task_prompt' method."
)
def get_format_instruction(self):
return "Identify the approximate start time of each chapter in the format 'hh:mm:ss - Title'. "
def get_new_line_instruction(self):
return "Ensure each chapter entry is on a new line. "
def get_focus_instruction(self):
return "Focus on significant topic changes that would merit a new chapter in a video, "
def get_no_summaries_instruction(self):
return "but do not provide summaries of the chapters.\n"
def get_transcript_introduction(self):
return "Here is the transcript to analyze:\n"
def get_transcript(self, vid_id: str) -> str:
# By default, the transcript is the same for train and test
raise NotImplementedError(
"Subclasses must implement the 'get_transcript' method."
)
def get_transcript_train(self, vid_id: str) -> str:
return self.get_transcript(vid_id)
def get_transcript_test(self, vid_id: str) -> str:
return self.get_transcript(vid_id)
def get_base_prompt(self, vid_id: str) -> str:
prompt_parts = [
self.get_duration_prompt(vid_id),
self.get_task_prompt(),
self.get_format_instruction(),
self.get_new_line_instruction(),
self.get_focus_instruction(),
self.get_no_summaries_instruction(),
self.get_transcript_introduction(),
]
return "".join(prompt_parts)
def get_prompt_train(self, vid_id: str) -> str:
return self.get_base_prompt(vid_id)
def get_prompt_test(self, vid_id: str) -> str:
return self.get_base_prompt(vid_id)
def get_output(self, vid_id: str) -> str:
vid_chapters = self.chapters.get_chapters(vid_id)
answers = []
for chp_time, chp_title in vid_chapters.items():
chp_time = sec_to_hms(chp_time)
answers.append(f"{chp_time} - {chp_title}")
return "\n".join(answers)
def get_dialog(self, vid_id: str) -> str:
prompt = self.get_prompt_train(vid_id)
transcript = self.get_transcript_train(vid_id)
output = self.get_output(vid_id)
dialog = [
{
"role": "user",
"content": prompt + transcript,
},
{
"role": "assistant",
"content": output,
},
]
return dialog
|