downloadable log files as tab

app.py

@@ -16,6 +16,9 @@ from utils import *
 from gptManager import ChatgptManager
 # from queryHelper import QueryHelper
 
+logs_dir = os.getenv("HF_HOME", "/data")
+
+
 pd.set_option('display.max_columns', None)
 pd.set_option('display.max_rows', None)
 
@@ -258,10 +261,40 @@ def onResetToDefaultSelection():
 
   return tableBoxes
 
+import gradio as gr
+import os
 
+# Define your function
+def getAllLogFilesUrls():
+    global logs_dir
+    # Save processed data to temporary file
+    logFiles = [file for file in os.listdir(logs_dir) if ('log' in file and '.csv' in file) ]
+    
+    downloadableFiles = [{"url":f"/download?file_path={logFilePath}", "name":logFilePath} for logFilePath in logFiles]
+    return downloadableFiles
 
-with gr.Blocks() as demo:
 
+
+# Download function
+def handle_button_click(download_url):
+    # Open the download URL in a new browser tab
+    js = f"window.open('{download_url}', '_blank')"
+    gr.run_js(js)
+
+# Connect button click to download
+download_button.click(handle_button_click, inputs=process_data, outputs=[0])
+
+# Build the Gradio interface
+with gr.Blocks() as interface:
+    
+
+# Launch the interface
+interface.launch()
+
+
+
+
+with gr.Blocks() as demo:
     # screen 1 : Chatbot for question answering to generate sql query from user input in english
     with gr.Tab("Query Helper"):
         gr.Markdown("""<h1><center> Query Helper</center></h1>""")
@@ -320,6 +353,15 @@ with gr.Blocks() as demo:
 
         resetToDefaultSelection = gr.Button("Reset to Default")
         resetToDefaultSelection.click(onResetToDefaultSelection, inputs=None, outputs=tableBoxes)
+    
+    #screen 4 for downloading logs
+    with gr.Tab("Log-files"):
+        downloadableFiles = getAllLogFilesUrls()
+        buttons = []
+        for downloadableFile in downloadableFiles:
+            button = gr.Button(f"Download {downloadableFile['name']}")
+            button.click(lambda url: gr.run_js(f"window.open('{url}', '_blank')"), inputs=[downloadableFile["url"]])
+            buttons.append(button)
 
 
 demo.launch(share=True, debug=True, ssl_verify=False, auth=checkAuth)