adjust session ID

app.py

@@ -17,51 +17,44 @@ with open("system_prompt.txt", "r") as f:
 client = InferenceClient(MODEL_NAME)
 api = HfApi()
 
-# ---- Session ID and local file setup ----
-SESSION_ID = 0
-TODAY_DATE = datetime.now().strftime("%Y-%m-%d")
-LOCAL_LOG_PATH = f"chatlog_{TODAY_DATE}_{SESSION_ID}.jsonl"
-REMOTE_LOG_PATH = f"sessions/{TODAY_DATE}/{SESSION_ID}.jsonl"  # stored in a folder in the dataset
 
-def increment_session_id():
-    global SESSION_ID
-    SESSION_ID += 1
-    return SESSION_ID
+def append_to_session_log(session_id, user_message, assistant_message):
+    today = datetime.now().strftime("%Y-%m-%d")
+    local_path = f"chatlog_{today}_{session_id}.jsonl"
+    remote_path = f"sessions/{today}/{session_id}.jsonl"
 
-# ---- Logging per session ----
-def append_to_session_log(user_message, assistant_message):
     row = {
         "timestamp": datetime.now().isoformat(),
         "user": user_message,
         "assistant": assistant_message,
         "system_prompt": SYSTEM_PROMPT,
-        "session_id": SESSION_ID
+        "session_id": session_id,
     }
-    with open(LOCAL_LOG_PATH, "a", encoding="utf-8") as f:
+    with open(local_path, "a", encoding="utf-8") as f:
         f.write(json.dumps(row) + "\n")
 
-    # Push to HF dataset
     api.upload_file(
-        path_or_fileobj=LOCAL_LOG_PATH,
-        path_in_repo=REMOTE_LOG_PATH,
+        path_or_fileobj=local_path,
+        path_in_repo=remote_path,
         repo_id=DATASET_REPO,
         repo_type="dataset",
         token=HF_TOKEN
     )
 
-# ---- Chatbot function ----
-def respond(message, history):
+
+def respond(message, history, session_id):
+    if session_id is None:
+        session_id = str(uuid.uuid4())  # unique per user session
+
     messages = [{"role": "system", "content": SYSTEM_PROMPT}]
-    
     for user_msg, bot_msg in history:
         if user_msg:
             messages.append({"role": "user", "content": user_msg})
         if bot_msg:
             messages.append({"role": "assistant", "content": bot_msg})
-    
     messages.append({"role": "user", "content": message})
-    response = ""
 
+    response = ""
     for chunk in client.chat_completion(
         messages,
         max_tokens=512,
@@ -72,15 +65,15 @@ def respond(message, history):
         token = chunk.choices[0].delta.content
         if token:
             response += token
-            yield response
+            yield response, session_id  # return state too
+
+    append_to_session_log(session_id, message, response)
 
-    # Save after each message pair
-    append_to_session_log(message, response)
-    increment_session_id()
 
-# ---- Gradio Interface ----
+# ---- Gradio UI ----
 demo = gr.ChatInterface(
-    respond,
+    fn=respond,
+    additional_inputs=[gr.State(value=None)],
     title="BoundrAI",
 )