history display

storage.py

@@ -10,9 +10,15 @@ import os
 from datetime import datetime
 from huggingface_hub import HfApi, hf_hub_download
 
+
+# --- CONFIGURATION ---
 REPO_ID = "prashantmatlani/chathistorycoderg"
 HISTORY_DIR = "./chathistory"
 
+# Internal Session Authentication Toggle
+_SESSION_UNLOCKED = False
+
+# Initialize the API with your token
 api = HfApi(token=os.getenv("HF_TOKEN"))
 
 def get_secret_password():
@@ -23,13 +29,30 @@ def get_secret_password():
 
     return str(val).strip() if val is not None else ""
 
+def verify_and_unlock(user_password_input):
+    """Validates input credentials and flips the internal tracking state variable."""
+    global _SESSION_UNLOCKED
+    target_password = get_secret_password()
+    clean_input = str(user_password_input).strip()
+    
+    # If no master environment password exists, unlock access automatically
+    if not target_password:
+        _SESSION_UNLOCKED = True
+        return True
+        
+    if clean_input == target_password:
+        _SESSION_UNLOCKED = True
+        return True
+        
+    return False
+
 def load_history(user_password=""):
     """Retrieves list of chat IDs from the Hub ONLY if the password matches the secret."""
     target_password = get_secret_password()
     clean_input = str(user_password).strip()
 
     # ====================================================================
-    # 🔐 YOUR NEW IF / ELSE SECURITY CONFIGURATION
+    # 🔐 IF / ELSE SECURITY CONFIGURATION
     # ====================================================================
     
     # Condition 1: If the password field is left blank OR if it doesn't match the secret...
@@ -65,4 +88,54 @@ def load_history(user_password=""):
         print(f"[!!] Critical Storage Interface Breakdown: {fallback_error}")
         return []
 
-# Keep your existing save_chat and get_chat_content below unchanged...
+def save_chat(chat_id, history):
+    """Saves chat to local subdirectory and syncs to Hugging Face Dataset."""
+    if not os.path.exists(HISTORY_DIR):
+        os.makedirs(HISTORY_DIR)
+    
+    if not chat_id:
+        chat_id = datetime.now().strftime("%m%d%Y_%H%M%S")
+    
+    filename = f"{chat_id}.json"
+    local_path = os.path.join(HISTORY_DIR, filename)
+    
+    with open(local_path, "w", encoding="utf-8") as f:
+        json.dump(history, f, indent=4)
+    
+    try:
+        api.upload_file(
+            path_or_fileobj=local_path,
+            path_in_repo=f"chats/{filename}",
+            repo_id=REPO_ID,
+            repo_type="dataset"
+        )
+    except Exception as e:
+        print(f"Cloud Sync Warning: {e}")
+        
+    return chat_id
+
+def get_chat_content(chat_id):
+    """Loads a specific chat's content from the Hub or local cache."""
+    filename = f"chats/{chat_id}.json"
+    local_path = os.path.join(HISTORY_DIR, f"{chat_id}.json")
+    
+    try:
+        if not os.path.exists(HISTORY_DIR):
+            os.makedirs(HISTORY_DIR)
+            
+        downloaded_path = hf_hub_download(
+            repo_id=REPO_ID, 
+            repo_type="dataset", 
+            filename=filename,
+            token=os.getenv("HF_TOKEN")
+        )
+        with open(downloaded_path, "r", encoding="utf-8") as f:
+            return json.load(f)
+    except Exception:
+        if os.path.exists(local_path):
+            with open(local_path, "r", encoding="utf-8") as f:
+                return json.load(f)
+    return []
+
+
+