Update app.py

app.py

@@ -1,19 +1,9 @@
-import os
 import openai
 import streamlit as st
 from youtube_transcript_api import YouTubeTranscriptApi
 from langchain.text_splitter import RecursiveCharacterTextSplitter
-from dotenv import load_dotenv, find_dotenv
 
-# Specify the path to your .env file
-#https://huggingface.co/spaces/Abu1998/YouTubeTranscriptApi1/resolve/main/.env
-env_path = 'YouTubeTranscriptApi1/resolve/main/.env' # Change the Path
-# Load the OpenAI API key from the .env file
-load_dotenv(env_path)
-openai.api_key = os.getenv('OPENAI_API_KEY')
-
-def get_transcript(youtube_url):
-    video_id = youtube_url.split("v=")[-1]
+def get_transcript(video_id):
     transcript_list = YouTubeTranscriptApi.list_transcripts(video_id)
 
     # Try fetching the manual transcript
@@ -33,12 +23,14 @@ def get_transcript(youtube_url):
     full_transcript = " ".join([part['text'] for part in transcript.fetch()])
     return full_transcript, language_code  # Return both the transcript and detected language
 
+def summarize_with_langchain_and_openai(transcript, language_code, openai_api_key, model_name='gpt-3.5-turbo'):
+    # Set the OpenAI API key
+    openai.api_key = openai_api_key
 
-def summarize_with_langchain_and_openai(transcript, language_code, model_name='gpt-3.5-turbo'):
     # Split the document if it's too long
     text_splitter = RecursiveCharacterTextSplitter(chunk_size=2000, chunk_overlap=0)
     texts = text_splitter.split_text(transcript)
-    text_to_summarize = " ".join(texts[:4]) # Adjust this as needed
+    text_to_summarize = " ".join(texts[:4])  # Adjust this as needed
 
     # Prepare the prompt for summarization
     system_prompt = 'I want you to act as a Life Coach that can create good summaries!'
@@ -57,15 +49,17 @@ def summarize_with_langchain_and_openai(transcript, language_code, model_name='g
         ],
         temperature=1
     )
-    
+
     return response['choices'][0]['message']['content']
 
 def main():
-    st.title('YouTube video summarizer')
-    link = st.text_input('Enter the link of the YouTube video you want to summarize:')
-
+    st.title('YouTube Video Summarizer')
+    
+    openai_api_key = st.text_input('Enter your OpenAI API Key:', type='password')
+    video_id = st.text_input('Enter the YouTube Video ID:')
+    
     if st.button('Start'):
-        if link:
+        if openai_api_key and video_id:
             try:
                 progress = st.progress(0)
                 status_text = st.empty()
@@ -74,13 +68,12 @@ def main():
                 progress.progress(25)
 
                 # Getting both the transcript and language_code
-                transcript, language_code = get_transcript(link)
+                transcript, language_code = get_transcript(video_id)
 
                 status_text.text(f'Creating summary...')
                 progress.progress(75)
 
-                model_name = 'gpt-3.5-turbo'
-                summary = summarize_with_langchain_and_openai(transcript, language_code, model_name)
+                summary = summarize_with_langchain_and_openai(transcript, language_code, openai_api_key)
 
                 status_text.text('Summary:')
                 st.markdown(summary)
@@ -88,7 +81,7 @@ def main():
             except Exception as e:
                 st.write(str(e))
         else:
-            st.write('Please enter a valid YouTube link.')
+            st.write('Please enter both your OpenAI API Key and YouTube Video ID.')
 
 if __name__ == "__main__":
-    main()
+    main()