jeremierostan commited on
Commit
83bfdcf
·
verified ·
1 Parent(s): 68a7b6f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -6
app.py CHANGED
@@ -13,12 +13,6 @@ os.environ["ANTHROPIC_API_KEY"] = ANTHROPIC_API_KEY
13
  username = os.getenv('username')
14
  password = os.getenv('password')
15
 
16
- # Add the path to your desired knowledge base
17
- reference_document = "Rosenshine+Principles+red.pdf"
18
- reader = PdfReader(reference_document)
19
- full_text = ''.join(page.extract_text() for page in reader.pages)
20
- text_chunks = chunk_text(full_text)
21
-
22
  # Function to chunk the document
23
  def chunk_text(text, chunk_size=1000, overlap=100):
24
  chunks = []
@@ -38,6 +32,12 @@ def get_relevant_chunks(query, chunks, top_n=3):
38
  relevant_indices = cosine_similarities.argsort()[-top_n:][::-1]
39
  return [chunks[i] for i in relevant_indices]
40
 
 
 
 
 
 
 
41
  def chat_with_assistant(message, history):
42
  # Find relevant chunks based on the user message
43
  relevant_chunks = get_relevant_chunks(message, text_chunks)
 
13
  username = os.getenv('username')
14
  password = os.getenv('password')
15
 
 
 
 
 
 
 
16
  # Function to chunk the document
17
  def chunk_text(text, chunk_size=1000, overlap=100):
18
  chunks = []
 
32
  relevant_indices = cosine_similarities.argsort()[-top_n:][::-1]
33
  return [chunks[i] for i in relevant_indices]
34
 
35
+ # Add the path to your desired knowledge base
36
+ reference_document = "Rosenshine+Principles.pdf"
37
+ reader = PdfReader(reference_document)
38
+ full_text = ''.join(page.extract_text() for page in reader.pages)
39
+ text_chunks = chunk_text(full_text)
40
+
41
  def chat_with_assistant(message, history):
42
  # Find relevant chunks based on the user message
43
  relevant_chunks = get_relevant_chunks(message, text_chunks)