Update app.py
Browse files
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)
|