Spaces:
Runtime error
Runtime error
Added get_csv_file_name method
Browse files
utils.py
CHANGED
|
@@ -1,6 +1,7 @@
|
|
| 1 |
import os
|
| 2 |
import shutil
|
| 3 |
import time
|
|
|
|
| 4 |
|
| 5 |
from langchain import FAISS
|
| 6 |
from langchain.text_splitter import RecursiveCharacterTextSplitter
|
|
@@ -75,3 +76,12 @@ def get_csv_files(csv_file, source_column, field_names=None):
|
|
| 75 |
def index_exists(pickle_file, index_file):
|
| 76 |
return os.path.isfile(pickle_file) and os.path.isfile(index_file) and os.path.getsize(
|
| 77 |
pickle_file) > 0
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
import os
|
| 2 |
import shutil
|
| 3 |
import time
|
| 4 |
+
import glob
|
| 5 |
|
| 6 |
from langchain import FAISS
|
| 7 |
from langchain.text_splitter import RecursiveCharacterTextSplitter
|
|
|
|
| 76 |
def index_exists(pickle_file, index_file):
|
| 77 |
return os.path.isfile(pickle_file) and os.path.isfile(index_file) and os.path.getsize(
|
| 78 |
pickle_file) > 0
|
| 79 |
+
|
| 80 |
+
|
| 81 |
+
def get_csv_file_name():
|
| 82 |
+
output_dir = 'output'
|
| 83 |
+
if os.path.exists(output_dir):
|
| 84 |
+
csv_files = glob.glob(os.path.join(output_dir, '*.csv'))
|
| 85 |
+
if csv_files:
|
| 86 |
+
return csv_files[0] # return the first csv file found
|
| 87 |
+
return None
|