def create_srt(split_df, cris_stt_df):
abs_path = os.path.dirname(split_df)
df1 = pd.read_csv(split_df)
df2 = pd.read_excel(cris_stt_df)
df1.rename(columns={'wav_filename': 'wav_name'}, inplace=True)
# This df3 contains all the info for srt creation
df3 = pd.merge(df1, df2, how='inner', on='wav_name')
print("Creating the srt:")
new_srt = SubRipFile()
for index, row in df3.iterrows():
text = str(row['transcripts'] if \
type(row['transcripts']) != float \
else "")
new_srt.append(
SubRipItem(index=index + 1,
start=SubRipTime(milliseconds=row['start']),
end=SubRipTime(milliseconds=row['end']),
text=text[:-1] if text.endswith(".") else text))
new_srt.save(os.path.join(abs_path, "stt_converted.srt"))
print("successfully written")
