Update app.py
Browse files
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 |
-
|
293 |
-
if
|
294 |
-
vektordatenbank, retriever = document_storage_chroma(
|
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
|
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 |
|