Update app.py

app.py

@@ -6,6 +6,7 @@ from typing import Tuple, List, Union
 import traceback
 import io
 import zipfile
+import tempfile
 
 # NTU Singapore colors
 NTU_BLUE = "#003D7C"
@@ -150,7 +151,7 @@ def create_scatter_plot(df: pd.DataFrame) -> Union[px.scatter, None]:
 def update_insights(df: pd.DataFrame, zip_data: Union[Tuple[io.BytesIO, str], None]) -> List[Union[gr.components.Component, None]]:
     try:
         if df.empty:
-            return [gr.Markdown("No data available. Please upload and process a file first.")] + [None] * 5
+            return [gr.Markdown("No data available. Please upload and process a file first.")] + [None] * 6
 
         stats = create_summary_stats(df)
         stats_md = gr.Markdown("\n".join([f"**{k}**: {v:.2f}" for k, v in stats.items()]))
@@ -163,14 +164,16 @@ def update_insights(df: pd.DataFrame, zip_data: Union[Tuple[io.BytesIO, str], No
         
         if zip_data:
             zip_file, zip_name = zip_data
-            download_button = gr.File(value=zip_file, visible=True, label="Download Results (ZIP)")
+            download_button = gr.File(value=zip_file, filename=zip_name, visible=True, label="Download Results")
+            download_text = gr.Markdown("Click the 'Download Results' button above to download the ZIP file containing all processed data.")
         else:
-            download_button = gr.File(visible=False, label="Download Results (ZIP)")
+            download_button = gr.File(visible=False, label="Download Results")
+            download_text = gr.Markdown("")
         
-        return [stats_md, users_activity_chart, users_courses_chart, scatter_plot, user_table, download_button]
+        return [stats_md, users_activity_chart, users_courses_chart, scatter_plot, user_table, download_button, download_text]
     except Exception as e:
         error_msg = f"Error updating insights: {str(e)}\n\nTraceback:\n{traceback.format_exc()}"
-        return [gr.Markdown(error_msg)] + [None] * 5
+        return [gr.Markdown(error_msg)] + [None] * 6
 
 def process_and_update(file):
     try:
@@ -179,10 +182,10 @@ def process_and_update(file):
         return [result_msg, csv_loc] + insights
     except Exception as e:
         error_msg = f"Error in process_and_update: {str(e)}\n\nTraceback:\n{traceback.format_exc()}"
-        return [error_msg, "Processing failed"] + [gr.Markdown(error_msg)] + [None] * 5
+        return [error_msg, "Processing failed"] + [gr.Markdown(error_msg)] + [None] * 6
 
 def clear_outputs():
-    return [""] * 2 + [None] * 6  # 2 text outputs and 6 graph/table/file outputs
+    return [""] * 2 + [None] * 6 + [""]  # 2 text outputs, 6 graph/table/file outputs, and 1 download text
 
 # Create a custom theme
 custom_theme = gr.themes.Base().set(
@@ -253,7 +256,10 @@ with gr.Blocks(theme=custom_theme, css=custom_css) as iface:
             
             scatter_plot = gr.Plot()
             user_table = gr.DataFrame()
-            download_button = gr.File(visible=False, label="Download Results (ZIP)")
+            
+            gr.Markdown("## Download Processed Data")
+            download_button = gr.File(visible=False, label="Download Results")
+            download_text = gr.Markdown("")
     
     clear_btn = gr.Button("Clear All Data", variant="secondary")
     gr.Markdown("Click 'Clear All Data' to reset the application and start over.")
@@ -261,13 +267,13 @@ with gr.Blocks(theme=custom_theme, css=custom_css) as iface:
     process_btn.click(
         process_and_update,
         inputs=[file_input],
-        outputs=[output_msg, csv_location, summary_stats, users_activity_chart, users_courses_chart, scatter_plot, user_table, download_button]
+        outputs=[output_msg, csv_location, summary_stats, users_activity_chart, users_courses_chart, scatter_plot, user_table, download_button, download_text]
     )
     
     clear_btn.click(
         clear_outputs,
         inputs=[],
-        outputs=[output_msg, csv_location, summary_stats, users_activity_chart, users_courses_chart, scatter_plot, user_table, download_button]
+        outputs=[output_msg, csv_location, summary_stats, users_activity_chart, users_courses_chart, scatter_plot, user_table, download_button, download_text]
     )
 
 if __name__ == "__main__":