create_zip_archive() extracted

src/plan/app_text2plan.py

@@ -8,12 +8,9 @@ import time
 import sys
 import threading
 from math import ceil
-import shutil
-import zipfile
-import copy
-
 from src.llm_factory import get_available_llms
 from src.plan.generate_run_id import generate_run_id
+from src.plan.create_zip_archive import create_zip_archive
 from src.plan.filenames import FilenameEnum
 from src.plan.plan_file import PlanFile
 from src.plan.speedvsdetail import SpeedVsDetailEnum
@@ -111,26 +108,6 @@ class SessionState:
         """
         return self
 
-def create_zip_archive(directory_to_zip: str) -> str:
-    """
-    Creates a zip archive of the given directory, excluding 'log.txt'.
-    Returns the path to the zip file, or None if an error occurred.
-    """
-    zip_file_path = os.path.join(os.path.dirname(directory_to_zip), os.path.basename(directory_to_zip) + ".zip")
-    try:
-        with zipfile.ZipFile(zip_file_path, 'w', zipfile.ZIP_DEFLATED) as zipf:
-            for root, _, files in os.walk(directory_to_zip):
-                for file in files:
-                    if file == "log.txt":
-                        continue  # Skip the log file
-
-                    file_path = os.path.join(root, file)
-                    zipf.write(file_path, os.path.relpath(file_path, directory_to_zip))
-        return zip_file_path
-    except Exception as e:
-        print(f"Error creating zip archive: {e}")
-        return None
-
 def run_planner(submit_or_retry_button, plan_prompt, llm_model, speedvsdetail, session_state: SessionState):
     """
     Generator function for launching the pipeline process and streaming updates.

src/plan/create_zip_archive.py

@@ -0,0 +1,27 @@
+import zipfile
+import os
+
+def create_zip_archive(directory_to_zip: str) -> str:
+    """
+    Creates a zip archive of the given directory, excluding 'log.txt'.
+    Returns the path to the zip file, or None if an error occurred.
+    """
+    zip_file_path = os.path.join(os.path.dirname(directory_to_zip), os.path.basename(directory_to_zip) + ".zip")
+    try:
+        with zipfile.ZipFile(zip_file_path, 'w', zipfile.ZIP_DEFLATED) as zipf:
+            for root, _, files in os.walk(directory_to_zip):
+                for file in files:
+                    if file == "log.txt":
+                        continue  # Skip the log file
+
+                    file_path = os.path.join(root, file)
+                    zipf.write(file_path, os.path.relpath(file_path, directory_to_zip))
+        return zip_file_path
+    except Exception as e:
+        print(f"Error creating zip archive: {e}")
+        return None
+
+if __name__ == "__main__":
+    dir_path = os.path.join(os.path.dirname(__file__), '..', 'expert', 'test_data')
+    zip_path = create_zip_archive(dir_path)
+    print(f"Zip archive created at: {zip_path}")