Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -72,11 +72,6 @@ api_key = os.getenv("OPENAI_API_KEY")
|
|
| 72 |
|
| 73 |
import chromadb
|
| 74 |
|
| 75 |
-
# Initialize CromA client
|
| 76 |
-
chroma_client = chromadb.Client()
|
| 77 |
-
|
| 78 |
-
# Later in your code, use chroma_client to create a collection
|
| 79 |
-
collection = chroma_client.create_collection(name="Kosten_Strukturdaten")
|
| 80 |
|
| 81 |
|
| 82 |
# Function to extract text from a PDF file
|
|
@@ -90,13 +85,6 @@ def extract_text_from_pdf(pdf_path):
|
|
| 90 |
# Example usage
|
| 91 |
pdf_text = extract_text_from_pdf(pdf_path3)
|
| 92 |
|
| 93 |
-
# Add the extracted text from PDF to the Chroma collection
|
| 94 |
-
collection.add(
|
| 95 |
-
documents=[pdf_text],
|
| 96 |
-
metadatas=[{"source": pdf_path3}], # Add any relevant metadata for your document
|
| 97 |
-
ids=[("Kosten_Strukturdaten")]
|
| 98 |
-
)
|
| 99 |
-
|
| 100 |
|
| 101 |
@st.cache_resource
|
| 102 |
def load_vector_store(file_path, store_name, force_reload=False):
|
|
@@ -539,6 +527,13 @@ def page3():
|
|
| 539 |
chroma_client = chromadb.Client()
|
| 540 |
collection = chroma_client.create_collection(name="Kosten_Strukturdaten")
|
| 541 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 542 |
display_chat_history(st.session_state['chat_history_page3'])
|
| 543 |
|
| 544 |
st.write("<!-- Start Spacer -->", unsafe_allow_html=True)
|
|
|
|
| 72 |
|
| 73 |
import chromadb
|
| 74 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 75 |
|
| 76 |
|
| 77 |
# Function to extract text from a PDF file
|
|
|
|
| 85 |
# Example usage
|
| 86 |
pdf_text = extract_text_from_pdf(pdf_path3)
|
| 87 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 88 |
|
| 89 |
@st.cache_resource
|
| 90 |
def load_vector_store(file_path, store_name, force_reload=False):
|
|
|
|
| 527 |
chroma_client = chromadb.Client()
|
| 528 |
collection = chroma_client.create_collection(name="Kosten_Strukturdaten")
|
| 529 |
|
| 530 |
+
# Add the extracted text from PDF to the Chroma collection
|
| 531 |
+
collection.add(
|
| 532 |
+
documents=[pdf_text],
|
| 533 |
+
metadatas=[{"source": pdf_path3}], # Add any relevant metadata for your document
|
| 534 |
+
ids=[("Kosten_Strukturdaten")]
|
| 535 |
+
)
|
| 536 |
+
|
| 537 |
display_chat_history(st.session_state['chat_history_page3'])
|
| 538 |
|
| 539 |
st.write("<!-- Start Spacer -->", unsafe_allow_html=True)
|