Update app.py
Browse files
app.py
CHANGED
|
@@ -6,7 +6,6 @@ from langchain.chains.summarize import load_summarize_chain
|
|
| 6 |
from langchain.chat_models import ChatOpenAI
|
| 7 |
|
| 8 |
|
| 9 |
-
@st.cache_data
|
| 10 |
def setup_documents(pdf_file_path,chunk_size,chunk_overlap):
|
| 11 |
loader = PyPDFLoader(pdf_file_path)
|
| 12 |
docs_raw = loader.load()
|
|
@@ -37,7 +36,7 @@ def custom_summary(docs, llm, custom_prompt, chain_type, num_summaries):
|
|
| 37 |
|
| 38 |
return summaries
|
| 39 |
|
| 40 |
-
|
| 41 |
def color_chunks(text: str, chunk_size: int, overlap_size: int) -> str:
|
| 42 |
overlap_color = "#808080"
|
| 43 |
chunk_colors = ["#a8d08d", "#c6dbef", "#e6550d", "#fd8d3c", "#fdae6b", "#fdd0a2"] # Different shades of green for chunks
|
|
|
|
| 6 |
from langchain.chat_models import ChatOpenAI
|
| 7 |
|
| 8 |
|
|
|
|
| 9 |
def setup_documents(pdf_file_path,chunk_size,chunk_overlap):
|
| 10 |
loader = PyPDFLoader(pdf_file_path)
|
| 11 |
docs_raw = loader.load()
|
|
|
|
| 36 |
|
| 37 |
return summaries
|
| 38 |
|
| 39 |
+
|
| 40 |
def color_chunks(text: str, chunk_size: int, overlap_size: int) -> str:
|
| 41 |
overlap_color = "#808080"
|
| 42 |
chunk_colors = ["#a8d08d", "#c6dbef", "#e6550d", "#fd8d3c", "#fdae6b", "#fdd0a2"] # Different shades of green for chunks
|