Penality commited on
Commit
6dee028
·
verified ·
1 Parent(s): 1e38ca3

Update app.py

Browse files

added json import

Files changed (1) hide show
  1. app.py +8 -1
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
- # Save document in a file
 
 
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: