Update src/utils/helper.py

src/utils/helper.py

@@ -125,10 +125,14 @@ def extract_transcript(video_link: str):
         # extract video id from video link
         video_id = video_link.split("v=")[1]
         
+        api_key = os.getenv("SUPADATA_API_KEY")
+        if not api_key:
+            raise ValueError("SUPADATA_API_KEY environment variable is not set")
+        
         # Call Supadata API
         url = f"https://api.supadata.ai/v1/youtube/transcript"
         headers = {
-            "x-api-key": os.getenv("SUPADATA_API_KEY")
+            "x-api-key": api_key
         }
         params = {
             "videoId": video_id
@@ -138,6 +142,10 @@ def extract_transcript(video_link: str):
         response.raise_for_status()  # Raise exception for non-200 status codes
         
         data = response.json()
+        logger.info(f"Data: {data}")
+        if not data.get("content"):
+            raise ValueError("No transcript content found in the API response")
+            
         text = ""
         for item in data["content"]:
             if "text" in item:
@@ -145,9 +153,15 @@ def extract_transcript(video_link: str):
         
         logger.info(f"Transcript: {text}")
         return text
+    except requests.exceptions.RequestException as e:
+        logger.error(f"API request failed: {str(e)}")
+        raise
+    except ValueError as e:
+        logger.error(str(e))
+        raise
     except Exception as e:
         logger.error(f"Failed to extract transcript: {str(e)}")
-        raise Exception(f"Failed to extract transcript: {str(e)}")
+        raise
 
 
 def extract_comment(video_link: str):