alexkueck commited on
Commit
3fa9bf8
·
verified ·
1 Parent(s): 9915a53

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -4
app.py CHANGED
@@ -289,14 +289,19 @@ def generate_auswahl(prompt_in, file, file_history, chatbot, history, anzahl_doc
289
  #?????????????????????????????????????????????? Nicht passend zum Promt???????????????????????????
290
  if vektordatenbank == None:
291
  print("db neu aufbauen!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!1")
292
- splits = document_loading_splitting()
293
- if splits:
294
- vektordatenbank, retriever = document_storage_chroma(splits)
295
 
296
  #kein Bild hochgeladen -> auf Text antworten...
297
  status = "Antwort der Vektordatenbank"
298
  results, status = generate_text(prompt, chatbot, history,vektordatenbank, retriever, top_p=0.6, temperature=0.5, max_new_tokens=4048, max_context_length_tokens=2048, repetition_penalty=1.3, top_k=3)
299
 
 
 
 
 
 
300
 
301
  #Ergebnisse für history und chatbot zusammenstellen
302
  summary = str(results['answer']) + "\n\n<b>Auszüge dazu: </b>"
@@ -305,7 +310,7 @@ def generate_auswahl(prompt_in, file, file_history, chatbot, history, anzahl_doc
305
  '(<b>Seite:</span> <span style="color: red;">' + str(doc['seite']) + '</b></span>)<br>'
306
  '<span><b>Auschnitt:</b> ' + str(doc["content"]) + '</span></div><br>'
307
  #'<div><span><b>Link: </b><span style="color: #BB70FC;"><a href="' + str(doc['download_link']) + '" target="_blank">' + str(doc['titel']) + '</a></span></div><br>'
308
- for doc in results['relevant_docs']])
309
 
310
  history = history + [[prompt_in, summary]]
311
 
 
289
  #?????????????????????????????????????????????? Nicht passend zum Promt???????????????????????????
290
  if vektordatenbank == None:
291
  print("db neu aufbauen!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!1")
292
+ preprocessed_splits, split_to_original_mapping = document_loading_splitting()
293
+ if preprocessed_splits:
294
+ vektordatenbank, retriever = document_storage_chroma(preprocessed_splits)
295
 
296
  #kein Bild hochgeladen -> auf Text antworten...
297
  status = "Antwort der Vektordatenbank"
298
  results, status = generate_text(prompt, chatbot, history,vektordatenbank, retriever, top_p=0.6, temperature=0.5, max_new_tokens=4048, max_context_length_tokens=2048, repetition_penalty=1.3, top_k=3)
299
 
300
+ #in results sind die preprocessed Splits enthalten, dargestellt werden sollen die orginalen:
301
+ relevant_docs_org=[]
302
+ for result in results['relevant_docs']:
303
+ original_split = split_to_original_mapping[result]
304
+ relevant_docs_org.append(original_split)
305
 
306
  #Ergebnisse für history und chatbot zusammenstellen
307
  summary = str(results['answer']) + "\n\n<b>Auszüge dazu: </b>"
 
310
  '(<b>Seite:</span> <span style="color: red;">' + str(doc['seite']) + '</b></span>)<br>'
311
  '<span><b>Auschnitt:</b> ' + str(doc["content"]) + '</span></div><br>'
312
  #'<div><span><b>Link: </b><span style="color: #BB70FC;"><a href="' + str(doc['download_link']) + '" target="_blank">' + str(doc['titel']) + '</a></span></div><br>'
313
+ for doc in relevant_docs_org])
314
 
315
  history = history + [[prompt_in, summary]]
316