Update session.py

session.py

@@ -1,28 +1,48 @@
-import uuid
-
-class Session:
-    def __init__(self, project_name):
-        self.session_id = str(uuid.uuid4())
-        self.project_name = project_name
-        self.variables = {}
-        self.auth_tokens = {}
-        self.last_intent = None
-        self.awaiting_variable = None
-        self.chat_history = []
-
-class SessionStore:
-    def __init__(self):
-        self.sessions = {}
-
-    def create_session(self, project_name):
-        session = Session(project_name)
-        self.sessions[session.session_id] = session
-        return session
-
-    def get_session(self, session_id):
-        return self.sessions.get(session_id)
-
-    def remove_session(self, session_id):
-        if session_id in self.sessions:
-            del self.sessions[session_id]
-
+import uuid
+import threading
+from log import log
+
+
+class Session:
+    def __init__(self, project_name):
+        self.session_id = str(uuid.uuid4())
+        self.project_name = project_name
+        self.last_intent = None
+        self.variables = {}  # Extracted parameters
+        self.awaiting_parameters = []  # Still-missing parameters
+        self.state = "intent_detection"  # intent_detection, parameter_extraction, api_call, humanization
+        self.chat_history = []  # [{'role': 'user'/'assistant', 'content': text}]
+        self.auth_tokens = {}  # Per-API auth tokens
+
+
+class SessionStore:
+    def __init__(self):
+        self.sessions = {}
+        self.lock = threading.Lock()
+
+    def create_session(self, project_name):
+        with self.lock:
+            session = Session(project_name)
+            self.sessions[session.session_id] = session
+            log(f"🆕 Created session: {session.session_id} for project: {project_name}")
+            return session
+
+    def get_session(self, session_id):
+        with self.lock:
+            return self.sessions.get(session_id)
+
+    def update_session(self, session_id, **kwargs):
+        with self.lock:
+            session = self.sessions.get(session_id)
+            if not session:
+                return None
+            for key, value in kwargs.items():
+                setattr(session, key, value)
+            log(f"🔄 Updated session: {session_id} with {kwargs}")
+            return session
+
+    def remove_session(self, session_id):
+        with self.lock:
+            if session_id in self.sessions:
+                del self.sessions[session_id]
+                log(f"❌ Removed session: {session_id}")