Update app.py

app.py

@@ -45,6 +45,8 @@ _ = load_dotenv(find_dotenv())
 ##############################################
 #nur bei ersten Anfrage splitten der Dokumente - um die Vektordatenbank entsprechend zu füllen
 splittet = False
+#für eine Session werden die chatsverläufe hier gespeichert...
+chats={}
 
 ##################################################
 #Für MongoDB statt Chroma als Vektorstore
@@ -121,8 +123,40 @@ general_assistant_suche= openai_assistant_suche(client)
 
 ##############################################
 #wenn löschen Button geklickt
-def clear_all():
-    return None, gr.Image(visible=False), [], []
+def clear_all(history):
+    global chats
+    dic_history = {schluessel: wert for schluessel, wert in history}
+    summary = "\n".join(f'{schluessel}:  {wert}' for schluessel, wert in dic_history.items())
+
+    id_neu = len(chats)+1
+    #chats ist ein dictionary
+    chats[id_neu]= summary
+
+    return None, gr.Image(visible=False), [], gr.CheckboxGroup(label="Wähle Chats zum Download", choices=update_chat_options())
+    
+    
+#########################################
+# Funktionen, um vergangene Chats anzuzeigen und zum Download anzubieten
+def update_chat_options():
+    global chats
+    # Diese Funktion aktualisiert die verfügbaren Chat-Optionen
+    if chats != {}:
+        return list(chats.keys())
+    else:
+        return None
+
+def download_chats(selected_chats):
+    global chats
+    if chats != {}:
+        # Diese Funktion bereitet die ausgewählten Chats zum Download vor
+        data = "\n\n".join(chats[chat] for chat in selected_chats)
+        # Erstelle eine temporäre Datei mit den Chat-Daten
+        with tempfile.NamedTemporaryFile(delete=False, mode='w', suffix='.txt') as tmp:
+            tmp.write(data)
+            tmp_path = tmp.name
+        return gr.File(tmp_path, label="Download-Chat", visible = True)
+    else:
+        return gr.File(visible=False)
 
 ##############################################
 #History - die Frage oder das File eintragen...
@@ -499,7 +533,6 @@ with gr.Blocks(css=customCSS, theme=small_and_beautiful_theme) as demo:
     # history parallel zu chatbot speichern - da in chatbot bei Bildern zum Anzeigen in der GUI die Bilder speziell formatiert werden, 
     # für die Übergabe an die ki aber der Pfad zum Bild behalten werden muss - was in der history der Fall ist!
     history = gr.State([])
-    long_history = gr.State([])
     #damit der Prompt auch nach dem upload in die History noch für predicts_args verfügbar ist
     user_question = gr.State("")
     #damit der Prompt auch nach dem upload in die History noch für predicts_args verfügbar ist
@@ -538,6 +571,10 @@ with gr.Blocks(css=customCSS, theme=small_and_beautiful_theme) as demo:
     
             with gr.Column():
                 with gr.Column(min_width=50, scale=1):
+                    with gr.Tab(label="Chats ..."):
+                        chat_selector = gr.CheckboxGroup(label="Wähle Chats zum Download", choices=update_chat_options())
+                        download_button = gr.Button("Download ausgewählte Chats")
+                        file_download = gr.File(label="Download-Chat",  visible=False)
                     with gr.Tab(label="Parameter Einstellung"):
                         #gr.Markdown("# Parameters")
                         rag_option = gr.Radio(["Aus", "An"], label="LI Erweiterungen (RAG)", value = "Aus")
@@ -662,12 +699,11 @@ with gr.Blocks(css=customCSS, theme=small_and_beautiful_theme) as demo:
     predict_event1 = user_input.submit(**transfer_input_args, queue=False,).then(**predict_args)
     predict_event2 = submitBtn.click(**transfer_input_args, queue=False,).then(**predict_args)
     predict_event3 = upload.upload(file_anzeigen, [upload], [image_display, image_display, attached_file] ) #.then(**predict_args)
-    emptyBtn.click(clear_all, [], [attached_file, image_display, history])
+    emptyBtn.click(clear_all, [history], [attached_file, image_display, history,  chat_selector])
     #Bild Anzeige neben dem Button wieder entfernen oder austauschen..
     image_display.select(file_loeschen, [], [attached_file, image_display])
-    # Interface-Elemente für Chats (vergangene)
-    chat_selector = gr.CheckboxGroup(label="Wähle Chats zum Download", choices=update_chat_options())
-    download_button = gr.Button("Download")
+    download_button.click(fn=download_chats, inputs=chat_selector, outputs=file_download)
+
     
     #Berechnung oder Ausgabe anhalten (kann danach fortgesetzt werden)
     cancelBtn.click(cancel_outputing, [], [status_display], cancels=[predict_event1,predict_event2, predict_event3])