Update app.py

app.py

@@ -1,5 +1,5 @@
 import gradio as gr
-from langchain.document_loaders.base import Document
+from langchain.document_loaders.base import Document, BaseLoader
 from langchain.indexes import VectorstoreIndexCreator
 from apify_client import ApifyClient
 import os
@@ -31,10 +31,21 @@ def fetch_website_content(website_url):
     items = list(apify_client.dataset(run["defaultDatasetId"]).iterate_items())
     return items if items else None
 
-# Fetch and index website content
+# Custom loader for our documents
+class CustomLoader(BaseLoader):
+    def __init__(self, documents):
+        self.documents = documents
+
+    def load(self):
+        return self.documents
+
+# Fetch content
 content = fetch_website_content("https://python.langchain.com/en/latest/")
 documents = [Document(page_content=item["content"] or "", metadata={"source": item.get("url", "Unknown URL")}) for item in content]
-index = VectorstoreIndexCreator().from_loaders([documents])
+
+# Use custom loader
+loader = CustomLoader(documents)
+index = VectorstoreIndexCreator().from_loaders([loader])
 
 # Function for the Gradio UI
 def ask_langchain(question):