Update app.py

app.py

@@ -1,6 +1,7 @@
 # importing the required libraries and classes 
 import os 
 import gradio as gr
+from dotenv import load_dotenv
 from langchain_openai import OpenAI
 from langchain.prompts import PromptTemplate
 from pytube import YouTube
@@ -8,8 +9,11 @@ from youtube_transcript_api import YouTubeTranscriptApi
 from langchain.chains import LLMChain
 from langchain_openai import ChatOpenAI
 
-# setting the openai_api_key
-os.environ['OPENAI_API_KEY'] = "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"
+# Load environment variables from .env file
+load_dotenv()
+
+# Access the API key
+openai_api_key = os.getenv("OPENAI_API_KEY")
 
 
 
@@ -24,8 +28,10 @@ def get_transcript(youtube_url):
         transcript_text = " ".join([entry["text"] for entry in transcript])
 
         return transcript_text
-    except Exception as e:
-        return f"Error: {e}"
+
+    # returning -1, if the URL is not valid
+    except:
+        return -1
 
 def get_title(youtube_url):
     """ This function will return the Title of the given YouTube URL """ 
@@ -34,8 +40,9 @@ def get_title(youtube_url):
         yt_video = YouTube(youtube_url)
         return yt_video.title
 
-    except Exception as e:
-        return f'Error: {e}'
+    # returning -1, if the URL is not valid 
+    except:
+        return -1
 
 def get_result(url,content):
     """ This function will return the content(type of content provided by the user)"""
@@ -43,6 +50,10 @@ def get_result(url,content):
     # getting the title and transcript of the Youtube URL
     title,transcript = get_title(url) , get_transcript(url)
 
+    # if the URL is not valid 
+    if title == -1 and transcript == -1:
+        return f'The provided url:  "{url}"  is not valid. Please provide a valid URL.'
+
     # this template is for the Article or Stories
     basic_template = "Imagine you're a content creator. Your task is to create {user_input} by leveraging the provided title and transcript creatively. Craft a compelling {user_input} that engages the audience and leverages the information from the video in an about 800-1000 words. Your output should be creative. Specifics: The YouTube video is titled '{title}', and you have access to its transcript: {transcript}"
 
@@ -67,7 +78,10 @@ def get_result(url,content):
     )
 
     # invoking the chain and return the output i.e. content
-    return chain.invoke({'title':title,'transcript':transcript,'user_input':content})['text']
+    try:
+        return chain.invoke({'title':title,'transcript':transcript,'user_input':content})['text']
+    except Exception as e:
+        return f'Error: {e}'