Update app.py

app.py

@@ -56,12 +56,13 @@ def process_single_task(task):
         time.sleep(1)
     intent = cohere_parse_command(task["command"])
     cmd = task["command"].lower()
+    # Pass the entire command as "subject" for dynamic content.
     if intent == "report":
         if "csv" in cmd:
-            filename = generate_csv_report()
+            filename = generate_csv_report(subject=task["command"])
             task["result"] = f"CSV report generated for '{task['command']}'"
         elif "xlsx" in cmd:
-            filename = generate_xlsx_report()
+            filename = generate_xlsx_report(subject=task["command"])
             task["result"] = f"XLSX report generated for '{task['command']}'"
         else:
             filename = generate_pdf_report(subject=task["command"])
@@ -161,7 +162,6 @@ def refresh_ui(tasks, files):
     # Update files state from the global generated_files list.
     files = generated_files[:]
     tasks_html = build_tasks_html(tasks)
-    # Return tasks, updated files state, tasks HTML, and update file downloads.
     return tasks, files, tasks_html, files
 
 def update_dropdown(files):
@@ -228,7 +228,7 @@ def main():
         tasks_state = gr.State([])
         files_state = gr.State([])
         
-        # Submit: add task then refresh UI immediately.
+        # Submit: add task then immediately refresh UI.
         submit_btn.click(
             submit_task,
             inputs=[command_input, tasks_state],
@@ -238,17 +238,14 @@ def main():
             inputs=[tasks_state, files_state],
             outputs=[tasks_state, files_state, tasks_html_output, file_output]
         )
-        
         sample1.click(lambda: "generate a report on unemployment in the United States in 2024", None, command_input)
         sample2.click(lambda: "generate diagram of sales data", None, command_input)
         sample3.click(lambda: "generate csv report of user activity", None, command_input)
-        
         refresh_btn.click(
             refresh_ui,
             inputs=[tasks_state, files_state],
             outputs=[tasks_state, files_state, tasks_html_output, file_output]
         )
-        
         # Hidden auto-refresh button.
         auto_refresh_btn = gr.Button("Auto Refresh", visible=False, elem_id="auto_refresh_btn")
         auto_refresh_btn.click(