Update app.py

app.py

@@ -168,6 +168,14 @@ def download_chats(selected_chats):
     if chats != {}:
         # Diese Funktion bereitet die ausgewählten Chats zum Download vor
         data = "\n\n".join(chats[chat] for chat in selected_chats)
+
+        file_path = save_and_download(data)
+        cleanup_button = gr.Button("Cleanup")
+        cleanup_button.click(lambda: cleanup(file_path), inputs=[], outputs=[])
+        return gr.File(file_path, label="Download-Chat"), cleanup_button
+
+
+        """
         # Dateipfad festlegen (hier wird die Datei im aktuellen Verzeichnis gespeichert)
         file_path = FILE_PATH  #'data/chatverlauf.txt'
 
@@ -176,8 +184,9 @@ def download_chats(selected_chats):
             # String in die Datei schreiben
             file.write(data)
         return gr.File(file_path, label="Download-Chat", visible = True, show_label=True)
+        """
     else:
-        return gr.File(visible=False)
+        return gr.File(visible=False), gr.Button("Cleanup", visible=False)
 
 ##############################################
 #History - die Frage oder das File eintragen...
@@ -550,7 +559,10 @@ with gr.Blocks(css=customCSS, theme=small_and_beautiful_theme) as demo:
                     with gr.Tab(label="Chats ..."):
                         chat_selector = gr.CheckboxGroup(label="", choices=update_chat_options())
                         download_button = gr.Button("Download ausgewählte Chats")
-                        file_download = gr.File(label="Download-Chat",  visible=False)
+                        with gr.Column(scale=4):
+                            file_download = gr.File(label="Download-Chat",  visible=False)
+                        with gr.Column(scale=1):
+                            loeschen=gr.Button(visible = False)
                     with gr.Tab(label="Parameter Einstellung"):
                         #gr.Markdown("# Parameters")
                         rag_option = gr.Radio(["Aus", "An"], label="LI Erweiterungen (RAG)", value = "Aus")
@@ -675,7 +687,7 @@ with gr.Blocks(css=customCSS, theme=small_and_beautiful_theme) as demo:
     emptyBtn.click(clear_all, [history], [attached_file, image_display, history,  chat_selector, file_download])
     #Bild Anzeige neben dem Button wieder entfernen oder austauschen..
     image_display.select(file_loeschen, [], [attached_file, image_display])
-    download_button.click(fn=download_chats, inputs=chat_selector, outputs=file_download)
+    download_button.click(fn=download_chats, inputs=chat_selector, outputs=[file_download, loeschen])
     
     #Berechnung oder Ausgabe anhalten (kann danach fortgesetzt werden)
     cancelBtn.click(cancel_outputing, [], [status_display], cancels=[predict_event1,predict_event2, predict_event3])