Update app.py

Zip file to download

app.py

@@ -12,7 +12,7 @@ NTU_BLUE = "#003D7C"
 NTU_RED = "#C11E38"
 NTU_GOLD = "#E7B820"
 
-def process_data(file: gr.File, progress=gr.Progress()) -> Tuple[str, str, pd.DataFrame, Union[io.BytesIO, None]]:
+def process_data(file: gr.File, progress=gr.Progress()) -> Tuple[str, str, pd.DataFrame, Union[Tuple[io.BytesIO, str], None]]:
     try:
         # Check if file is uploaded
         if file is None:
@@ -97,7 +97,7 @@ def process_data(file: gr.File, progress=gr.Progress()) -> Tuple[str, str, pd.Da
 
         zip_buffer.seek(0)
         progress(1.0, desc="Processing complete")
-        return "Processing complete. You can now download the results.", "Results are packaged in the zip file below.", user_info, zip_buffer
+        return "Processing complete. You can now download the results.", "Results are packaged in the zip file below.", user_info, (zip_buffer, f"gradebook_results_{base_filename}.zip")
     except Exception as e:
         error_msg = f"Error: {str(e)}\n\nTraceback:\n{traceback.format_exc()}"
         return error_msg, "Processing failed", pd.DataFrame(), None
@@ -147,7 +147,7 @@ def create_scatter_plot(df: pd.DataFrame) -> Union[px.scatter, None]:
         print(f"Error creating scatter plot: {str(e)}")
         return None
 
-def update_insights(df: pd.DataFrame, zip_file: Union[io.BytesIO, None]) -> List[Union[gr.components.Component, 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
@@ -161,10 +161,11 @@ def update_insights(df: pd.DataFrame, zip_file: Union[io.BytesIO, None]) -> List
         
         user_table = gr.DataFrame(value=df)
         
-        if zip_file:
-            download_button = gr.File(value=zip_file, filename="gradebook_results.zip", visible=True, label="Download Results")
+        if zip_data:
+            zip_file, zip_name = zip_data
+            download_button = gr.File(value=zip_file, visible=True, label="Download Results (ZIP)")
         else:
-            download_button = gr.File(visible=False, label="Download Results")
+            download_button = gr.File(visible=False, label="Download Results (ZIP)")
         
         return [stats_md, users_activity_chart, users_courses_chart, scatter_plot, user_table, download_button]
     except Exception as e:
@@ -173,8 +174,8 @@ def update_insights(df: pd.DataFrame, zip_file: Union[io.BytesIO, None]) -> List
 
 def process_and_update(file):
     try:
-        result_msg, csv_loc, df, zip_file = process_data(file)
-        insights = update_insights(df, zip_file)
+        result_msg, csv_loc, df, zip_data = process_data(file)
+        insights = update_insights(df, zip_data)
         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()}"
@@ -252,7 +253,7 @@ 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")
+            download_button = gr.File(visible=False, label="Download Results (ZIP)")
     
     clear_btn = gr.Button("Clear All Data", variant="secondary")
     gr.Markdown("Click 'Clear All Data' to reset the application and start over.")