Update app.py

app.py

@@ -44,7 +44,7 @@ def cohere_parse_command(command):
 
 # Global containers for pending tasks and generated files.
 task_bucket = []      # Queue of tasks waiting to be processed.
-generated_files = []  # List of filenames for files generated by completed tasks.
+generated_files = []  # Global list of filenames generated.
 bucket_lock = threading.Lock()
 
 def process_single_task(task):
@@ -157,9 +157,12 @@ def build_tasks_html(tasks):
     return html
 
 def refresh_ui(tasks, files):
-    """Refreshes the task list UI by rebuilding the HTML table."""
+    """Refreshes the UI by rebuilding the task list and updating the files state."""
+    # Update files state from the global generated_files list.
+    files = generated_files[:]
     tasks_html = build_tasks_html(tasks)
-    return tasks, files, tasks_html
+    # Return tasks, updated files state, tasks HTML, and update file downloads.
+    return tasks, files, tasks_html, files
 
 def update_dropdown(files):
     """Updates the dropdown choices for preview from the list of generated files."""
@@ -225,7 +228,7 @@ def main():
         tasks_state = gr.State([])
         files_state = gr.State([])
         
-        # On submit, add the task and then refresh the UI.
+        # Submit: add task then refresh UI immediately.
         submit_btn.click(
             submit_task,
             inputs=[command_input, tasks_state],
@@ -233,22 +236,25 @@ def main():
         ).then(
             refresh_ui,
             inputs=[tasks_state, files_state],
-            outputs=[tasks_state, files_state, tasks_html_output]
+            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]
+            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(
             refresh_ui,
             inputs=[tasks_state, files_state],
-            outputs=[tasks_state, files_state, tasks_html_output]
+            outputs=[tasks_state, files_state, tasks_html_output, file_output]
         )
         gr.HTML("""
             <script>
@@ -261,7 +267,7 @@ def main():
         """)
         auto_refresh_toggle.elem_id = "auto_refresh_toggle"
         
-        # Update file dropdown from files_state.
+        # Update file dropdown from files state.
         refresh_btn.click(
             update_dropdown,
             inputs=[files_state],