add session id

app.py

@@ -17,35 +17,16 @@ with open("system_prompt.txt", "r") as f:
 client = InferenceClient(MODEL_NAME)
 api = HfApi()
 
-
-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"
-
-    row = {
-        "timestamp": datetime.now().isoformat(),
-        "user": user_message,
-        "assistant": assistant_message,
-        "system_prompt": SYSTEM_PROMPT,
-        "session_id": session_id,
-    }
-    with open(local_path, "a", encoding="utf-8") as f:
-        f.write(json.dumps(row) + "\n")
-
-    api.upload_file(
-        path_or_fileobj=local_path,
-        path_in_repo=remote_path,
-        repo_id=DATASET_REPO,
-        repo_type="dataset",
-        token=HF_TOKEN
-    )
-
-
+# ---- Chatbot function with per-user session ID ----
 def respond(message, history, session_id):
     if session_id is None:
-        session_id = str(uuid.uuid4())  # unique per user session
+        session_id = str(uuid.uuid4())  # Generate a new unique ID
+
+    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"
 
+    # Construct conversation history
     messages = [{"role": "system", "content": SYSTEM_PROMPT}]
     for user_msg, bot_msg in history:
         if user_msg:
@@ -54,6 +35,7 @@ def respond(message, history, session_id):
             messages.append({"role": "assistant", "content": bot_msg})
     messages.append({"role": "user", "content": message})
 
+    # Generate response
     response = ""
     for chunk in client.chat_completion(
         messages,
@@ -65,15 +47,32 @@ def respond(message, history, session_id):
         token = chunk.choices[0].delta.content
         if token:
             response += token
-            yield response, session_id  # return state too
+            yield response, session_id  # Yield response and session ID state
 
-    append_to_session_log(session_id, message, response)
+    # ---- Logging ----
+    row = {
+        "timestamp": datetime.now().isoformat(),
+        "user": message,
+        "assistant": response,
+        "system_prompt": SYSTEM_PROMPT,
+        "session_id": session_id
+    }
 
+    with open(local_log_path, "a", encoding="utf-8") as f:
+        f.write(json.dumps(row) + "\n")
+
+    api.upload_file(
+        path_or_fileobj=local_log_path,
+        path_in_repo=remote_log_path,
+        repo_id=DATASET_REPO,
+        repo_type="dataset",
+        token=HF_TOKEN
+    )
 
-# ---- Gradio UI ----
+# ---- Gradio Interface with session state ----
 demo = gr.ChatInterface(
     fn=respond,
-    additional_inputs=[gr.State(value=None)],
+    additional_inputs=[gr.State(None)],
     title="BoundrAI",
 )