Spaces:
Build error
Build error
Update app.py
Browse files
app.py
CHANGED
@@ -6,7 +6,7 @@ import pdfplumber
|
|
6 |
from langchain_community.vectorstores import FAISS
|
7 |
from langchain_openai import OpenAIEmbeddings
|
8 |
from langchain.text_splitter import RecursiveCharacterTextSplitter
|
9 |
-
|
10 |
# Initialize system
|
11 |
@st.cache_resource
|
12 |
def init_system():
|
@@ -14,13 +14,10 @@ def init_system():
|
|
14 |
process_pdf("Q1FY24.pdf")
|
15 |
|
16 |
# 2. Load embeddings with secure API key
|
17 |
-
|
18 |
-
|
19 |
-
|
20 |
-
|
21 |
-
openai_api_key=os.getenv("OPENAI_API_KEY"),
|
22 |
-
model="text-embedding-ada-002" # Explicit model name
|
23 |
-
)
|
24 |
|
25 |
# 3. Load vector store
|
26 |
vector_store = FAISS.load_local("faiss_index", embeddings)
|
@@ -49,7 +46,10 @@ def process_pdf(pdf_path):
|
|
49 |
# Text Chunking & Embedding
|
50 |
splitter = RecursiveCharacterTextSplitter(chunk_size=1000)
|
51 |
chunks = splitter.split_text(full_text)
|
52 |
-
embeddings = OpenAIEmbeddings(
|
|
|
|
|
|
|
53 |
FAISS.from_texts(chunks, embeddings).save_local("faiss_index")
|
54 |
|
55 |
# Streamlit UI
|
|
|
6 |
from langchain_community.vectorstores import FAISS
|
7 |
from langchain_openai import OpenAIEmbeddings
|
8 |
from langchain.text_splitter import RecursiveCharacterTextSplitter
|
9 |
+
|
10 |
# Initialize system
|
11 |
@st.cache_resource
|
12 |
def init_system():
|
|
|
14 |
process_pdf("Q1FY24.pdf")
|
15 |
|
16 |
# 2. Load embeddings with secure API key
|
17 |
+
embeddings = OpenAIEmbeddings(
|
18 |
+
openai_api_key=os.getenv("OPENAI_API_KEY"),
|
19 |
+
model="text-embedding-ada-002"
|
20 |
+
)
|
|
|
|
|
|
|
21 |
|
22 |
# 3. Load vector store
|
23 |
vector_store = FAISS.load_local("faiss_index", embeddings)
|
|
|
46 |
# Text Chunking & Embedding
|
47 |
splitter = RecursiveCharacterTextSplitter(chunk_size=1000)
|
48 |
chunks = splitter.split_text(full_text)
|
49 |
+
embeddings = OpenAIEmbeddings(
|
50 |
+
openai_api_key=os.getenv("OPENAI_API_KEY"),
|
51 |
+
model="text-embedding-ada-002"
|
52 |
+
)
|
53 |
FAISS.from_texts(chunks, embeddings).save_local("faiss_index")
|
54 |
|
55 |
# Streamlit UI
|