alexkueck commited on
Commit
ef2a139
1 Parent(s): 9faf264

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -12
app.py CHANGED
@@ -61,6 +61,15 @@ CHROMA_WORD = './chroma/kkg/word'
61
  CHROMA_EXCEL = './chroma/kkg/excel'
62
  DOCS_DIR = "chroma/kkg"
63
 
 
 
 
 
 
 
 
 
 
64
  ###########################################
65
  # Alternativen, um HF Modelle in der rAG Chain einzusetzen
66
  ###########################################
@@ -176,14 +185,7 @@ def add_text(chatbot, history, prompt, file, file_history):
176
  chatbot = chatbot +[("Hochgeladenes Dokument: "+ get_filename(file) +"\n" + prompt, None)]
177
 
178
  return chatbot, history, prompt, file, file_history, gr.Image(visible = False), ""
179
- """
180
- def add_text2(chatbot, prompt):
181
- if (prompt == ""):
182
- chatbot = chatbot + [("", "Prompt fehlt!")]
183
- else:
184
- chatbot = chatbot + [(prompt, None)]
185
- return chatbot, prompt, ""
186
- """
187
 
188
  ############################################
189
  #nach dem Upload soll das zusätzliche Fenster mit dem image drinnen angezeigt werden
@@ -217,7 +219,6 @@ def generate_text (prompt, chatbot, history, retriever, top_p=0.6, temperature=0
217
  raise gr.Error("Prompt ist erforderlich.")
218
 
219
  try:
220
-
221
  #########################################
222
  #Prompt mit History Daten zusammenstellen:
223
  #Prompt an history anhängen und einen Text daraus machen
@@ -285,9 +286,6 @@ def generate_auswahl(prompt_in, file, file_history, chatbot, history, anzahl_doc
285
 
286
  #prompt normalisieren bevor er an die KIs geht
287
  prompt = preprocess_text(prompt_in)
288
-
289
- # Versuchen, den Vektorstore zu laden - nur wenn Dateien neu hochgeladen werden, werden die Splits neu erstellt
290
- vektordatenbank = load_vectorstore()
291
 
292
  if vektordatenbank is None:
293
  print("db neu aufbauen!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!1")
 
61
  CHROMA_EXCEL = './chroma/kkg/excel'
62
  DOCS_DIR = "chroma/kkg"
63
 
64
+ ###########################################
65
+ # Beim Start der Anwendung - Vektorstore laden!!!!
66
+ ###########################################
67
+ print("Vektorstore laden.........................")
68
+ vectorstore = load_vectorstore()
69
+ if vectorstore is None:
70
+ # Initialisieren Sie den Vektorstore bei Bedarf
71
+ vectorstore = create_vectorstore()
72
+
73
  ###########################################
74
  # Alternativen, um HF Modelle in der rAG Chain einzusetzen
75
  ###########################################
 
185
  chatbot = chatbot +[("Hochgeladenes Dokument: "+ get_filename(file) +"\n" + prompt, None)]
186
 
187
  return chatbot, history, prompt, file, file_history, gr.Image(visible = False), ""
188
+
 
 
 
 
 
 
 
189
 
190
  ############################################
191
  #nach dem Upload soll das zusätzliche Fenster mit dem image drinnen angezeigt werden
 
219
  raise gr.Error("Prompt ist erforderlich.")
220
 
221
  try:
 
222
  #########################################
223
  #Prompt mit History Daten zusammenstellen:
224
  #Prompt an history anhängen und einen Text daraus machen
 
286
 
287
  #prompt normalisieren bevor er an die KIs geht
288
  prompt = preprocess_text(prompt_in)
 
 
 
289
 
290
  if vektordatenbank is None:
291
  print("db neu aufbauen!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!1")