el/youtubesummarizer
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

put in a function the cleanup

Browse Source
This commit is contained in:
Leonard Excoffier
2024-02-13 22:30:16 -05:00
parent a4e735563f
commit 8177f9eef2
1 changed files with 9 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
main.py
View File

@@ -30,8 +30,6 @@ def transcription(OPENAI_API_KEY, audio_filename):
print("Transcription finished") print("Transcription finished")
os.remove(audio_filename)
return transcript return transcript
except Exception as e: except Exception as e:
print(f"Failed to transcript audio: {e}") print(f"Failed to transcript audio: {e}")
@@ -56,12 +54,21 @@ def summarize(OPENAI_API_KEY, transcript):
print(f"Failed to summarize transcript: {e}") print(f"Failed to summarize transcript: {e}")
return None return None
def cleanup(audio_filename):
try:
os.remove(audio_filename)
print(f"Deleted audio: {audio_filename}")
except Exception as e:
print(f"Failed to delete audio: {audio_filename}: {e}")
return None
def main(): def main():
OPENAI_API_KEY = os.environ.get('OPENAI_API_KEY') OPENAI_API_KEY = os.environ.get('OPENAI_API_KEY')
video_url = str(input("video url:")) video_url = str(input("video url:"))
audio_filename = download_audio(video_url) audio_filename = download_audio(video_url)
transcript = transcription(OPENAI_API_KEY, audio_filename) transcript = transcription(OPENAI_API_KEY, audio_filename)
summarize(OPENAI_API_KEY, transcript) summarize(OPENAI_API_KEY, transcript)
cleanup(audio_filename)
if __name__ == "__main__": if __name__ == "__main__":
main() main()