Update app.py

app.py

@@ -59,7 +59,7 @@ def convert_to_audio(video_filename):
 
 
 @st.cache_data
-def get_completion(prompt, model="gpt-3.5-turbo"):
+def summarise(prompt, model="gpt-3.5-turbo"):
     if openai.api_key is None:
         return "No OpenAI API Key provided."
     messages = [{"role": "user", "content": prompt}]
@@ -122,10 +122,9 @@ def main():
             - Fetch transcript from YouTube API (if available) by clicking the **Fetch Transcript** button.
             - Transcribe the video using the Whisper model by clicking the **Transcribe (Whisper)** button.
         - The transcript will be displayed in a text area below.
-        - You can download the video audio or transcript by clicking the respective buttons.
         - If you have an OpenAI API Key, you will be able to generate a summary of the transcript by ChatGPT.
         - The summary will be displayed in a text area below.
-        - You can download the summary by clicking the **Download** button.
+        - You can download the video, audio, transcript or summary by clicking the respective download buttons.
         """
     )
 
@@ -214,20 +213,20 @@ def process_youtube_video(video_id, col, emp, text_filename):
                 transcript_text = "\n".join(
                     [re.sub("\s+", " ", chunk["text"]) for chunk in transcript_text]
                 )
-                c1, c2 = st.columns(3)
+                c1, c2 = st.columns(2)
                 with c1:
                     modified_text = st.text_area(
                         "Transcript", transcript_text, height=500
                     )
-                    st.download_button("Download", modified_text, text_filename)
+                    st.download_button("Download Transcript", modified_text, text_filename)
                 with c2:
-                    openai_summarization = get_completion(
+                    openai_summarization = summarise(
                         PROMPT.format(modified_text)
                     )
                     summarized_text = st.text_area(
                         "Summarized Transcript", openai_summarization, height=500
                     )
-                    st.download_button("Download", summarized_text, text_filename)
+                    st.download_button("Download Summary", summarized_text, text_filename)
 
     except CouldNotRetrieveTranscript:
         emp.warning("Could Not Retrieve API Transcripts for this video.")