Manishkumaryadav commited on
Commit
0a07fd2
·
verified ·
1 Parent(s): f5ff6a4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -9
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
- from langchain_openai import OpenAIEmbeddings
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
- # In both places where embeddings are initialized:
20
- embeddings = OpenAIEmbeddings(
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(openai_api_key=os.getenv("OPENAI_API_KEY"))
 
 
 
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