Update Utils.py

Utils.py

@@ -84,16 +84,29 @@ def generate_roadmap_image():
 
     roadmap.set_footer("Generated by Roadmapper")
     roadmap.draw()
-    roadmap.save("project_roadmap.png")
+    project_name = get_dynamic_filename()
+    roadmap.save(project_name)
+    
+    st.download_button(
+        label="Download Roadmap Image",
+        data=open(project_name, "rb").read(),
+        key="download_roadmap",
+        file_name=project_name,
+        mime="image/png",
+    )
     
-    image_buffer = BytesIO()
     
-    image_data = image_buffer.getvalue()
-    return image_data
 
 def get_binary_file_downloader_html(bin_file, file_label, button_label):
     with st.spinner("Processing..."):
         # Provide a download link for the image
         bin_str = base64.b64encode(bin_file).decode()
         href = f'<a href="data:application/octet-stream;base64,{bin_str}" download="{file_label}">{button_label}</a>'
-        return href
+        return href
+    
+    
+def get_dynamic_filename():
+    project_name = st.session_state["project_name"]
+    current_time = datetime.datetime.now().strftime("%Y%m%d%H%M%S")
+    filename = f"{project_name}_roadmap_{current_time}.png"
+    return filename