Update app/transcript_extractor.py

app/transcript_extractor.py

@@ -16,20 +16,26 @@ import requests
 logging.basicConfig(level=logging.DEBUG, format='%(asctime)s - %(levelname)s - %(message)s')
 logger = logging.getLogger(__name__)
 
-# Get the directory of the current script
-current_dir = os.path.dirname(os.path.abspath(__file__))
-# Construct the path to the .env file (one directory up from the current script)
-dotenv_path = os.path.join(os.path.dirname(current_dir), '.env')
-logger.info(f"The .env path is: {dotenv_path}")
-# Load environment variables from .env file
-load_dotenv(dotenv_path)
+# Try to load from .env file if it exists, but don't fail if it doesn't
+try:
+    current_dir = os.path.dirname(os.path.abspath(__file__))
+    dotenv_path = os.path.join(os.path.dirname(current_dir), '.env')
+    if os.path.exists(dotenv_path):
+        load_dotenv(dotenv_path)
+        logger.info(f"Loaded environment variables from {dotenv_path}")
+except Exception as e:
+    logger.warning(f"Could not load .env file: {e}")
 
 # Get API key from environment variable
 API_KEY = os.getenv('YOUTUBE_API_KEY')
-logger.info(f"API_KEY: {API_KEY[:5]}...{API_KEY[-5:]}")  # Log first and last 5 characters for verification
 
-if not API_KEY:
-    raise ValueError("YouTube API key not found. Make sure it's set in your .env file in the parent directory of the 'app' folder.")
+# Safe logging of API key
+if API_KEY:
+    masked_key = f"{API_KEY[:3]}...{API_KEY[-3:]}" if len(API_KEY) > 6 else "***"
+    logger.info(f"API_KEY found (masked): {masked_key}")
+else:
+    logger.error("YouTube API key not found in environment variables")
+    raise ValueError("YouTube API key not found. Make sure it's set in your environment variables or .env file.")
 
 def get_youtube_client():
     try:
@@ -49,6 +55,7 @@ def get_youtube_client():
         logger.error(f"Error initializing YouTube API client: {str(e)}")
         raise
 
+# Rest of your existing functions remain the same...
 def extract_video_id(url):
     if not url:
         return None
@@ -69,7 +76,6 @@ def get_video_metadata(video_id):
             video = response['items'][0]
             snippet = video['snippet']
             
-            # Get the description and set default if it's blank
             description = snippet.get('description', '').strip()
             if not description:
                 description = 'Not Available'
@@ -82,7 +88,7 @@ def get_video_metadata(video_id):
                 'like_count': video['statistics'].get('likeCount', '0'),
                 'comment_count': video['statistics'].get('commentCount', '0'),
                 'duration': video['contentDetails']['duration'],
-                'description': description  # Add the description to the metadata
+                'description': description
             }
         else:
             logger.error(f"No video found with id: {video_id}")
@@ -120,7 +126,7 @@ def get_channel_videos(channel_url):
             part="id,snippet",
             channelId=channel_id,
             type="video",
-            maxResults=50  # Adjust as needed
+            maxResults=50
         )
         response = request.execute()
 
@@ -147,8 +153,8 @@ def extract_channel_id(url):
     return None
 
 def test_api_key():
-    youtube = get_youtube_client()
     try:
+        youtube = get_youtube_client()
         request = youtube.videos().list(part="snippet", id="dQw4w9WgXcQ")
         response = request.execute()
         if 'items' in response: