Update app.py

app.py

@@ -247,30 +247,56 @@ def download_markdown(n_clicks):
     return dcc.send_bytes(generated_file.getvalue(), f"release_notes_{datetime.now().strftime('%Y%m%d%H%M%S')}.md")
 
 @app.callback(
-    [Output("pr-button", "children"),
-     Output("pr-button", "disabled")],
-    Input("pr-button", "n_clicks"),
-    [State("repo-url", "value"),
-     State("folder-location", "value")],
-    prevent_initial_call=True
+    [Output("output-notes", "children"),
+     Output("download-button", "disabled"),
+     Output("pr-button", "disabled"),
+     Output("download-markdown", "data"),
+     Output("pr-button", "children")],
+    [Input("generate-button", "n_clicks"),
+     Input("download-button", "n_clicks"),
+     Input("pr-button", "n_clicks")],
+    [State("git-provider", "value"),
+     State("repo-url", "value"),
+     State("start-date", "value"),
+     State("end-date", "value"),
+     State("folder-location", "value")]
 )
-def create_pr(n_clicks, repo_url, folder_location):
-    if n_clicks is None:
-        return dash.no_update, dash.no_update
-    
-    global generated_file
-    if generated_file is None:
-        return "Error: No file generated", True
+def handle_all_actions(generate_clicks, download_clicks, pr_clicks, 
+                       git_provider, repo_url, start_date, end_date, folder_location):
+    global generated_file, pr_url
+    ctx = callback_context
 
-    file_name = f"{datetime.now().strftime('%m-%d-%Y')}.md"
-    markdown_content = generated_file.getvalue().decode()
-    
-    result = update_summary_and_create_pr(repo_url, folder_location, file_name, markdown_content)
-    
-    if pr_url:
-        return f"PR Created: {pr_url}", True
-    else:
-        return result, False
+    if not ctx.triggered:
+        return "", True, True, None, "Create PR"
+
+    button_id = ctx.triggered[0]['prop_id'].split('.')[0]
+
+    if button_id == "generate-button":
+        notes, file_name = generate_release_notes(git_provider, repo_url, start_date, end_date, folder_location)
+        return notes, False, False, None, "Create PR"
+
+    elif button_id == "download-button":
+        if generated_file is None:
+            return dash.no_update, dash.no_update, dash.no_update, None, dash.no_update
+        return (dash.no_update, dash.no_update, dash.no_update, 
+                dcc.send_bytes(generated_file.getvalue(), f"release_notes_{datetime.now().strftime('%Y%m%d%H%M%S')}.md"),
+                dash.no_update)
+
+    elif button_id == "pr-button":
+        if generated_file is None:
+            return dash.no_update, dash.no_update, dash.no_update, None, "Error: No file generated"
+
+        file_name = f"{datetime.now().strftime('%m-%d-%Y')}.md"
+        markdown_content = generated_file.getvalue().decode()
+        
+        result = update_summary_and_create_pr(repo_url, folder_location, file_name, markdown_content)
+        
+        if pr_url:
+            return dash.no_update, dash.no_update, True, None, f"PR Created: {pr_url}"
+        else:
+            return dash.no_update, dash.no_update, False, None, result
+
+    return dash.no_update, dash.no_update, dash.no_update, None, dash.no_update
 
 if __name__ == '__main__':
     print("Starting the Dash application...")