Update app.py
Browse files
app.py
CHANGED
@@ -47,6 +47,11 @@ from qdrant_client.models import (
|
|
47 |
SparseIndexParams
|
48 |
)
|
49 |
|
|
|
|
|
|
|
|
|
|
|
50 |
def make_points(texts: List[str], metadatas: List[dict], dense: List[List[float]], sparse: List[SparseEmbedding])-> List[PointStruct]:
|
51 |
points = []
|
52 |
for idx, (text, metadata, sparse_vector, dense_vector) in enumerate(zip(texts, metadatas, sparse, dense)):
|
@@ -618,12 +623,14 @@ if __name__ == '__main__':
|
|
618 |
|
619 |
processing_time = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
|
620 |
base_name, ext = os.path.splitext(os.path.basename(f.name))
|
|
|
|
|
621 |
|
622 |
st.session_state.df.append(
|
623 |
{
|
624 |
-
'icon':
|
625 |
'document': base_name,
|
626 |
-
'type':
|
627 |
'path': os.path.realpath(uploaded_file.name),
|
628 |
'time': processing_time,
|
629 |
'toggle': True
|
|
|
47 |
SparseIndexParams
|
48 |
)
|
49 |
|
50 |
+
icon_to_types = {
|
51 |
+
'ppt: 'powerpoint.svg',
|
52 |
+
'pptx': 'powerpoint.svg'
|
53 |
+
}
|
54 |
+
|
55 |
def make_points(texts: List[str], metadatas: List[dict], dense: List[List[float]], sparse: List[SparseEmbedding])-> List[PointStruct]:
|
56 |
points = []
|
57 |
for idx, (text, metadata, sparse_vector, dense_vector) in enumerate(zip(texts, metadatas, sparse, dense)):
|
|
|
623 |
|
624 |
processing_time = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
|
625 |
base_name, ext = os.path.splitext(os.path.basename(f.name))
|
626 |
+
|
627 |
+
file_type = ext.lstrip('.')
|
628 |
|
629 |
st.session_state.df.append(
|
630 |
{
|
631 |
+
'icon': os.path.join(os.getenv('HF_HOME'), icon_to_types[file_type])
|
632 |
'document': base_name,
|
633 |
+
'type': file_type.upper(),
|
634 |
'path': os.path.realpath(uploaded_file.name),
|
635 |
'time': processing_time,
|
636 |
'toggle': True
|