Update app.py

app.py

@@ -48,6 +48,10 @@ def get_generated_files():
     files_list = utils.format_directory(OUTPUT_DIR)  # This function should list files in the directory
     return gr.HTML(f"<h3>Generated Files:</h3><pre><code style='overflow-x: auto'>{files_list}</pre></code>")
 
+def get_download_btn():
+    """Get download all files button."""
+    return gr.Button("Download All Files", elem_id="download-btn")
+
 class AutoAPI:
     def __init__(self, huggingface_key, ai_name, ai_role, top_5_goals):
         self.huggingface_key = huggingface_key
@@ -73,7 +77,7 @@ def start(huggingface_key, ai_name, ai_role, top_5_goals, description):
         return gr.Column.update(visible=False), gr.Column.update(visible=True), gr.update(value=tasks)
     except Exception as e:
         logger.error("Failed to start AutoAPI: %s", str(e))
-        return gr.Column.update(visible=True ), gr.Column.update(visible=False), gr.update(value=[])
+        return gr.Column.update(visible=True), gr.Column.update(visible=False), gr.update(value=[])
 
 # Main Gradio Interface
 with gr.Blocks(css=CSS) as demo:
@@ -95,7 +99,7 @@ with gr.Blocks(css=CSS) as demo:
     
     start_btn.click(
         start,
-        inputs=[api_key, ai_name, ai_role, top_5_goals, description],
+        inputs=[api_key, ai_name , ai_role, top_5_goals, description],
         outputs=[setup_pane, main_pane, inferred_tasks]
     )