devve1 commited on
Commit
0cdc7ec
1 Parent(s): 6de185f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -2
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': ext.lstrip('.').upper(),
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