alexkueck commited on
Commit
077e8e9
·
verified ·
1 Parent(s): da38c91

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +25 -30
app.py CHANGED
@@ -60,19 +60,6 @@ CHROMA_WORD = './chroma/kkg/word'
60
  CHROMA_EXCEL = './chroma/kkg/excel'
61
  DOCS_DIR = "chroma/kkg"
62
 
63
- ###########################################
64
- # Beim Start der Anwendung - Vektorstore laden!!!!
65
- ###########################################
66
- print("Vektorstore laden.........................")
67
- vektordatenbank = load_vectorstore()
68
- if vektordatenbank is None:
69
- print("Vektorstore neu.......................")
70
- # Initialisieren Sie den Vektorstore bei Bedarf
71
- #vektordatenbank, SPLIT_TO_ORIGINAL_MAPPING werden aktualisiert
72
- create_vectorstore()
73
- else:
74
- SPLIT_TO_ORIGINAL_MAPPING = load_splits_and_metadata()
75
-
76
  ###########################################
77
  # Alternativen, um HF Modelle in der rAG Chain einzusetzen
78
  ###########################################
@@ -109,14 +96,35 @@ else:
109
  ###############################################
110
  #globale Variablen
111
  ##############################################
112
-
113
-
114
- #############################################
115
- # Allgemeine Konstanten
116
  #Filepath zu temp Folder (temp) mit File von ausgewähltem chatverlauf
117
  file_path_download = ""
118
 
 
 
 
 
 
 
 
 
 
 
 
 
 
119
 
 
 
 
 
 
 
 
 
 
 
 
 
120
 
121
  #################################################
122
  #################################################
@@ -213,19 +221,6 @@ def cancel_outputing():
213
  def reset_textbox():
214
  return gr.update(value=""),""
215
 
216
- ################################################
217
- # Erstellen des Vektorstores
218
- ################################################
219
- def create_vectorstore():
220
- global vektordatenbank, SPLIT_TO_ORIGINAL_MAPPING
221
- # Splits zu allen Dokumenten in den Verzeichnissen erstellen
222
- PREPROCESSED_SPLITS, SPLIT_TO_ORIGINAL_MAPPING = document_loading_splitting()
223
- if PREPROCESSED_SPLITS:
224
- # Vektordatenbank zu den Splits erstellen
225
- vektordatenbank = document_storage_chroma(PREPROCESSED_SPLITS)
226
- # Speichern der Splits und Metadaten
227
- save_splits_and_metadata((PREPROCESSED_SPLITS, SPLIT_TO_ORIGINAL_MAPPING))
228
-
229
 
230
 
231
  ####################################################
 
60
  CHROMA_EXCEL = './chroma/kkg/excel'
61
  DOCS_DIR = "chroma/kkg"
62
 
 
 
 
 
 
 
 
 
 
 
 
 
 
63
  ###########################################
64
  # Alternativen, um HF Modelle in der rAG Chain einzusetzen
65
  ###########################################
 
96
  ###############################################
97
  #globale Variablen
98
  ##############################################
 
 
 
 
99
  #Filepath zu temp Folder (temp) mit File von ausgewähltem chatverlauf
100
  file_path_download = ""
101
 
102
+ ################################################
103
+ # Erstellen des Vektorstores
104
+ ################################################
105
+ def create_vectorstore():
106
+ global vektordatenbank, SPLIT_TO_ORIGINAL_MAPPING
107
+ # Splits zu allen Dokumenten in den Verzeichnissen erstellen
108
+ PREPROCESSED_SPLITS, SPLIT_TO_ORIGINAL_MAPPING = document_loading_splitting()
109
+ if PREPROCESSED_SPLITS:
110
+ # Vektordatenbank zu den Splits erstellen
111
+ vektordatenbank = document_storage_chroma(PREPROCESSED_SPLITS)
112
+ # Speichern der Splits und Metadaten
113
+ save_splits_and_metadata((PREPROCESSED_SPLITS, SPLIT_TO_ORIGINAL_MAPPING))
114
+
115
 
116
+ ###########################################
117
+ # Beim Start der Anwendung - Vektorstore laden!!!!
118
+ ###########################################
119
+ print("Vektorstore laden.........................")
120
+ vektordatenbank = load_vectorstore()
121
+ if vektordatenbank is None:
122
+ print("Vektorstore neu.......................")
123
+ # Initialisieren Sie den Vektorstore bei Bedarf
124
+ #vektordatenbank, SPLIT_TO_ORIGINAL_MAPPING werden aktualisiert
125
+ create_vectorstore()
126
+ else:
127
+ SPLIT_TO_ORIGINAL_MAPPING = load_splits_and_metadata()
128
 
129
  #################################################
130
  #################################################
 
221
  def reset_textbox():
222
  return gr.update(value=""),""
223
 
 
 
 
 
 
 
 
 
 
 
 
 
 
224
 
225
 
226
  ####################################################