add user workspace

app.py

@@ -5,12 +5,33 @@ from typing import Dict
 import streamlit as st
 import yaml
 from loguru import logger as _logger
+import shutil
+import uuid
 
 from metagpt.const import METAGPT_ROOT
 from metagpt.ext.spo.components.optimizer import PromptOptimizer
 from metagpt.ext.spo.utils.llm_client import SPO_LLM, RequestType
 
 
+
+def get_user_workspace():
+    if "user_id" not in st.session_state:
+        st.session_state.user_id = str(uuid.uuid4())
+    
+    workspace_dir = Path("workspace") / st.session_state.user_id
+    workspace_dir.mkdir(parents=True, exist_ok=True)
+    return workspace_dir
+
+def cleanup_workspace(workspace_dir: Path) -> None:
+    try:
+        if workspace_dir.exists():
+            shutil.rmtree(workspace_dir)
+            _logger.info(f"Cleaned up workspace directory: {workspace_dir}")
+    except Exception as e:
+        _logger.error(f"Error cleaning up workspace: {e}")
+
+
+
 def load_yaml_template(template_path: Path) -> Dict:
     if template_path.exists():
         with open(template_path, "r", encoding="utf-8") as f:
@@ -76,6 +97,8 @@ def main():
     if "optimization_results" not in st.session_state:
         st.session_state.optimization_results = []
 
+    workspace_dir = get_user_workspace()
+
     st.title("SPO | Self-Supervised Prompt Optimization 🤖")
 
     # Sidebar for configurations
@@ -213,7 +236,7 @@ def main():
 
                 # Create optimizer instance
                 optimizer = PromptOptimizer(
-                    optimized_path="workspace",
+                    optimized_path=str(workspace_dir),
                     initial_round=initial_round,
                     max_rounds=max_rounds,
                     template=f"{template_name}.yaml",