Spaces:
Sleeping
Sleeping
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:
|