Spaces:
Build error
Build error
Update app.py
Browse filesadded json import
app.py
CHANGED
|
@@ -1,4 +1,5 @@
|
|
| 1 |
import gradio as gr
|
|
|
|
| 2 |
import os
|
| 3 |
import pdfplumber
|
| 4 |
import together
|
|
@@ -57,14 +58,20 @@ def store_document(text):
|
|
| 57 |
# Generate a unique filename
|
| 58 |
filename = os.path.join(DOCUMENT_DIR, f"doc_{len(metadata) + 1}.txt")
|
| 59 |
|
| 60 |
-
|
|
|
|
|
|
|
| 61 |
with open(filename, "w") as f:
|
| 62 |
f.write(text)
|
|
|
|
|
|
|
| 63 |
|
| 64 |
# Generate and store embedding
|
| 65 |
embedding = embedding_model.encode([text]).astype(np.float32)
|
| 66 |
index.add(embedding)
|
| 67 |
|
|
|
|
|
|
|
| 68 |
# Update metadata
|
| 69 |
metadata[len(metadata)] = filename
|
| 70 |
with open(METADATA_FILE, "w") as f:
|
|
|
|
| 1 |
import gradio as gr
|
| 2 |
+
import json
|
| 3 |
import os
|
| 4 |
import pdfplumber
|
| 5 |
import together
|
|
|
|
| 58 |
# Generate a unique filename
|
| 59 |
filename = os.path.join(DOCUMENT_DIR, f"doc_{len(metadata) + 1}.txt")
|
| 60 |
|
| 61 |
+
print(filename)
|
| 62 |
+
|
| 63 |
+
# Save document in a file
|
| 64 |
with open(filename, "w") as f:
|
| 65 |
f.write(text)
|
| 66 |
+
|
| 67 |
+
print("document saved")
|
| 68 |
|
| 69 |
# Generate and store embedding
|
| 70 |
embedding = embedding_model.encode([text]).astype(np.float32)
|
| 71 |
index.add(embedding)
|
| 72 |
|
| 73 |
+
print("emeddings generated")
|
| 74 |
+
|
| 75 |
# Update metadata
|
| 76 |
metadata[len(metadata)] = filename
|
| 77 |
with open(METADATA_FILE, "w") as f:
|