Update app.py

app.py

@@ -42,6 +42,8 @@ LOG_DIR        = pathlib.Path("logs")             # each session: logs/<uuid>.js
 IMG_DIR        = LOG_DIR / "imgs"                 # cached PNGs (optional)
 LOG_DIR.mkdir(exist_ok=True)
 IMG_DIR.mkdir(exist_ok=True)
+HF_TOKEN = os.getenv("HF_TOKEN")  # Set this in your Space's secrets
+DATASET_NAME = "collinear-ai/amazon-external-premier-chat-logs"  # Create this dataset on HF
 
 # ====== Bedrock client ====== #
 bedrock = boto3.client(
@@ -99,7 +101,35 @@ def cache_image(session_id: str, pil_img: Image.Image) -> str:
     pil_img.save(fpath, format="PNG")
     return str(fpath)
 
+
+
+def upload_to_hf_dataset(session_id: str, user_text: str, assistant_text: str, img_path: Optional[str] = None):
+    """Upload conversation log to HF Dataset"""
+    try:
+        record = {
+            "session_id": session_id,
+            "timestamp": datetime.datetime.utcnow().isoformat() + "Z",
+            "user_message": user_text,
+            "assistant_message": assistant_text,
+            "has_image": img_path is not None,
+            "image_path": img_path if img_path else ""
+        }
+        
+        # Create dataset from single record
+        dataset = Dataset.from_list([record])
+        
+        # Push to hub (append mode)
+        dataset.push_to_hub(
+            DATASET_NAME,
+            token=HF_TOKEN,
+            private=True  # Keep logs private
+        )
+    except Exception as e:
+        print(f"Failed to upload log: {e}")
+
+# Replace your append_log function with:
 def append_log(session_id: str, user_text: str, assistant_text: str, img_path: Optional[str] = None):
+    # Keep local logging for immediate access
     record = {
         "ts": datetime.datetime.utcnow().isoformat(timespec="seconds") + "Z",
         "user": user_text,
@@ -110,6 +140,9 @@ def append_log(session_id: str, user_text: str, assistant_text: str, img_path: O
     path = LOG_DIR / f"{session_id}.jsonl"
     with path.open("a", encoding="utf-8") as f:
         f.write(json.dumps(record, ensure_ascii=False) + "\n")
+    
+    # Also upload to HF Dataset
+    upload_to_hf_dataset(session_id, user_text, assistant_text, img_path)
 
 # ====== Gradio UI ====== #
 with gr.Blocks(title="Multimodal Chat") as demo: