Update app.py

app.py

@@ -49,19 +49,24 @@ def check_password():
 if not check_password():
     st.stop()  # Do not continue if check_password is not True.
 
-def generate_zip(filenames: list):
+def generate_zip(dir_path):
     """Generates a zip file containing the input files."""
     zip_buffer = io.BytesIO()
     with zipfile.ZipFile(zip_buffer, 'w') as zip_file:
-        for filename in filenames:
-            with open(filename, 'rb') as f:
-                file_content = f.read()
-                zip_file.writestr(filename, file_content)
+        for root, _, files in os.walk(dir_path):
+            # Add each file to the zip archive
+            for file in files:
+                file_path = os.path.join(root, file)
+                with open(file_path, 'rb') as f:
+                    file_content = f.read()
+                    zip_file.writestr(filename, file_content, arcname=os.path.relpath(file_path, dir_path))
     zip_buffer.seek(0)
     return zip_buffer
 
+if "default" not in st.session_state:
+    st.session_state["default"] = 2000
 text_input = st.text_area("Enter your software development requirements here:", 
-                          height=2000 if 'button' not in st.session_state else None,
+                          height=st.session_state["default"],
                           value="""The goal.
 
 The instructions of the task for each file example, things to do with input and output.
@@ -123,7 +128,8 @@ comments from Nick/Marc/...
 """)
 
 if st.button("Generate Code"):
-    st.session_state.button = True
+    st.balloons()
+    st.session_state["default"] = 200
     def download_content(url):
         response = requests.get(url)
         response.raise_for_status()
@@ -139,10 +145,12 @@ if st.button("Generate Code"):
                 except requests.exceptions.RequestException as e:
                     print(f"Error downloading content from {line}: {e}")
         return '\n'.join(lines)
+
+    st.info('Working on it.. A download button will show up below once work done.')
     
     text_input = replace_urls_with_content(text_input)
     repo: ProjectRepo = generate_repo(text_input)
-    zip_buffer = generate_zip(repo.all_files)
+    zip_buffer = generate_zip(repo.workdir())
     st.download_button(
         label="Download Code Repository",
         data=zip_buffer,