Update app.py

app.py

@@ -247,6 +247,17 @@ def get_rag_response(question):
     
     return response
 
+def create_history_entry(page_content, metadata):
+    source = metadata.get('source', 'No source available')
+    page = metadata.get('page', 'No page information available')
+    download_link = f"https://example.com/download/{source.replace('pdf/', '')}"
+
+    return {
+        "page_content": page_content,
+        "page": page,
+        "download_link": download_link
+    }
+
 
 	
 #Eingaben der GUI verarbeiten	
@@ -274,7 +285,7 @@ def generate_auswahl(prompt_in, file, file_history, chatbot, history, anzahl_doc
         #kein Bild hochgeladen -> auf Text antworten...
         status = "Antwort der KI ..."
         if (file == None and file_history == None):
-            result, status = generate_text(prompt, chatbot, history,vektordatenbank, top_p=0.6, temperature=0.5, max_new_tokens=4048, max_context_length_tokens=2048, repetition_penalty=1.3, top_k=3)
+            results, status = generate_text(prompt, chatbot, history,vektordatenbank, top_p=0.6, temperature=0.5, max_new_tokens=4048, max_context_length_tokens=2048, repetition_penalty=1.3, top_k=3)
         else:
             #Es wurde ein File neu angehängt -> das hochladen und dann Prompt bearbeiten
             #das history_fiel muss neu gesetzt werden
@@ -283,7 +294,7 @@ def generate_auswahl(prompt_in, file, file_history, chatbot, history, anzahl_doc
                 neu_file = file
     
             #File hochladen in Chroma und dann Antwort generieren
-            result = generate_text_zu_doc(neu_file, prompt, k, rag_option, chatbot, history, vektordatenbank)
+            results = generate_text_zu_doc(neu_file, prompt, k, rag_option, chatbot, history, vektordatenbank)
             #########################################
             #result verarbeiten
             #aus result die Ergebnisse aufteilen und der History anhängen
@@ -297,28 +308,14 @@ def generate_auswahl(prompt_in, file, file_history, chatbot, history, anzahl_doc
                 #"documents": [{"link": link, "passage": passage} for link, passage in zip(links, passages)]
             #}
 
-        response =  result
-	
+    	print("hier...........................")
+        for text in results:
+            entry = create_history_entry(text['page_content'], text['metadata'])
+            #history.append(entry)
+            history = history + [[prompt, entry]]
+
+        return chatbot, history, None, file_history, ""
 	
-        #die history erweitern - abhängig davon, ob gerade ein file hochgeladen wurde oder nicht
-        if (file != None):    
-            history = history + [[(file,), None],[prompt, response]]
-        else:
-            history = history + [[prompt, response]]
-                
-
-				
-        chatbot[-1][1] = ""
-        for character in response:
-            chatbot[-1][1] += character
-            time.sleep(0.03)
-            yield chatbot, history, None, neu_file, status
-            if shared_state.interrupted:
-                shared_state.recover()
-                try:
-                    yield chatbot, history, None, neu_file, "Stop: Success"
-                except:
-                    pass
     else: #noch nicht validiert, oder kein Prompt
         return chatbot, history, None, file_history, "Erst validieren oder einen Prompt eingeben!"