Update rag_chain.py

rag_chain.py

@@ -51,15 +51,18 @@ Answer:""",
     input_variables=["context", "question"],
 )
 
-# Updated to optionally accept proxies
-def fetch_transcript(video_id: str, proxies: dict = None) -> str:
+def fetch_transcript(video_id: str) -> str:
     try:
-        # If proxies are provided, patch requests.Session to use them
-        if proxies:
-            session = requests.Session()
-            session.proxies.update(proxies)
-            # Monkey patch the YouTubeTranscriptApi's internal session to use proxy
-            YouTubeTranscriptApi._requests = session
+        # ✅ Define your proxy here
+        proxies = {
+            "http": "http://219.65.73.81:80",
+            "https": "http://219.65.73.81:80"
+        }
+
+        # Patch requests session with proxy
+        session = requests.Session()
+        session.proxies.update(proxies)
+        YouTubeTranscriptApi._requests = session  # monkey patch
 
         transcript = YouTubeTranscriptApi.get_transcript(video_id, languages=["en", "hi"])
         return " ".join([t["text"] for t in transcript])
@@ -69,9 +72,10 @@ def fetch_transcript(video_id: str, proxies: dict = None) -> str:
     except Exception as e:
         raise Exception(f"Error fetching transcript: {str(e)}")
 
+
 # Build RAG chain from transcript
-def build_chain(video_id: str, proxies: dict = None) -> RetrievalQA:
-    text = fetch_transcript(video_id, proxies=proxies)
+def build_chain(video_id: str) -> RetrievalQA:
+    text = fetch_transcript(video_id)
 
     splitter = RecursiveCharacterTextSplitter(chunk_size=500, chunk_overlap=50)
     docs = splitter.create_documents([text])