Update app.py

app.py

@@ -149,15 +149,18 @@ def generate_csv(start_date, end_date):
         df.to_csv(tmp.name, index=False)
         df_per_month.to_csv(tmp_month.name, index=False)
 
-        return tmp.name, tmp_month.name, df
+        return tmp.name, tmp_month.name, df, df_per_month
 
 
 with gr.Blocks() as demo:
     gr.Markdown("# Fake SaaS Data Generator")
     start_date = gr.Textbox(label="Start Date (YYYY-MM-DD)", value="2023-01-01")
     end_date = gr.Textbox(label="End Date (YYYY-MM-DD)", value="2025-01-31")
-    output_df = gr.Dataframe(
-        label="Generated Data Preview", interactive=False, wrap=True
+    output_df_per_day = gr.Dataframe(
+        label="Generated Data Preview (per day)", interactive=False, wrap=True
+    )
+    output_df_per_month = gr.Dataframe(
+        label="Generated Data Preview (per month)", interactive=False, wrap=True
     )
     output_file_per_day = gr.File(label="Download Generated CSV (per day)")
     output_file_per_month = gr.File(label="Download Generated CSV (per month)")
@@ -165,7 +168,12 @@ with gr.Blocks() as demo:
     generate_btn.click(
         fn=generate_csv,
         inputs=[start_date, end_date],
-        outputs=[output_file_per_day, output_file_per_month, output_df],
+        outputs=[
+            output_file_per_day,
+            output_file_per_month,
+            output_df_per_day,
+            output_df_per_month,
+        ],
     )
 
 if __name__ == "__main__":