Update auth.py

auth.py

@@ -24,10 +24,15 @@ def register_user(username, password, logged_in_username, auth_globals):
         except:
             pass
 
+        # Create default conversation
         user_data = {
             "password": generate_password_hash(password),
-            "history": []
+            "conversations": {
+                "default": []
+            },
+            "current_conversation": "default"
         }
+
         hf_api.upload_file(
             path_or_fileobj=json.dumps(user_data).encode('utf-8'),
             path_in_repo=f"{username}.json",
@@ -35,8 +40,9 @@ def register_user(username, password, logged_in_username, auth_globals):
             repo_type="dataset",
             token=hf_token
         )
-        # Auto-login
+
         return username, "Registration successful! Logged in.", gr.update(visible=True)
+
     except Exception as e:
         return logged_in_username, f"Registration failed: {str(e)}", gr.update(visible=bool(logged_in_username))
 
@@ -46,7 +52,7 @@ def login_user(username, password, logged_in_username, auth_globals):
     hf_api = HfApi(token=hf_token)
 
     if not username or not password:
-        return [], logged_in_username, "Please enter username and password.", gr.update(visible=bool(logged_in_username))
+        return [], "default", logged_in_username, "Please enter username and password.", gr.update(visible=bool(logged_in_username))
 
     try:
         user_file = hf_hub_download(
@@ -55,24 +61,34 @@ def login_user(username, password, logged_in_username, auth_globals):
             repo_type="dataset",
             token=hf_token
         )
+
         with open(user_file, 'r') as f:
             data = json.load(f)
+
         if check_password_hash(data["password"], password):
-            return data["history"], username, "Login successful!", gr.update(visible=True)
+            current_conv = data.get("current_conversation", "default")
+            return (data["conversations"].get(current_conv, []),
+                    current_conv,
+                    username,
+                    "Login successful!",
+                    gr.update(visible=True))
         else:
-            return [], logged_in_username, "Incorrect password.", gr.update(visible=bool(logged_in_username))
+            return [], "default", logged_in_username, "Incorrect password.", gr.update(visible=bool(logged_in_username))
+
     except Exception as e:
-        return [], logged_in_username, f"Login failed: {str(e)}", gr.update(visible=bool(logged_in_username))
+        return [], "default", logged_in_username, f"Login failed: {str(e)}", gr.update(visible=bool(logged_in_username))
 
 def logout_user():
-    return [], None, "Logged out successfully!", gr.update(visible=False)
+    return [], "default", None, "Logged out successfully!", gr.update(visible=False)
 
-def save_history(username, history, auth_globals):
+def save_history(username, history, current_conversation, auth_globals):
     if not username:
         return "No user logged in!"
+
     hf_token = auth_globals["hf_token"]
     user_repo = auth_globals["user_repo"]
     hf_api = HfApi(token=hf_token)
+
     try:
         user_file = hf_hub_download(
             repo_id=user_repo,
@@ -80,9 +96,12 @@ def save_history(username, history, auth_globals):
             repo_type="dataset",
             token=hf_token
         )
+
         with open(user_file, 'r') as f:
             data = json.load(f)
-        data["history"] = history
+
+        data["conversations"][current_conversation] = history
+
         hf_api.upload_file(
             path_or_fileobj=json.dumps(data).encode('utf-8'),
             path_in_repo=f"{username}.json",
@@ -92,4 +111,102 @@ def save_history(username, history, auth_globals):
         )
         return "History saved!"
     except Exception as e:
-        return f"Failed to save history: {str(e)}"
+        return f"Failed to save history: {str(e)}"
+
+def create_new_conversation(username, current_conversation, new_conversation_name, auth_globals):
+    if not username:
+        return current_conversation, "No user logged in!", []
+
+    hf_token = auth_globals["hf_token"]
+    user_repo = auth_globals["user_repo"]
+    hf_api = HfApi(token=hf_token)
+
+    try:
+        user_file = hf_hub_download(
+            repo_id=user_repo,
+            filename=f"{username}.json",
+            repo_type="dataset",
+            token=hf_token
+        )
+
+        with open(user_file, 'r') as f:
+            data = json.load(f)
+
+        if new_conversation_name in data["conversations"]:
+            return current_conversation, f"Conversation '{new_conversation_name}' already exists!", data["conversations"][current_conversation]
+
+        data["conversations"][new_conversation_name] = []
+        data["current_conversation"] = new_conversation_name
+
+        hf_api.upload_file(
+            path_or_fileobj=json.dumps(data).encode('utf-8'),
+            path_in_repo=f"{username}.json",
+            repo_id=user_repo,
+            repo_type="dataset",
+            token=hf_token
+        )
+
+        return new_conversation_name, f"Created new conversation: {new_conversation_name}", []
+
+    except Exception as e:
+        return current_conversation, f"Failed to create conversation: {str(e)}", data["conversations"][current_conversation]
+
+def switch_conversation(username, current_conversation, target_conversation, auth_globals):
+    if not username:
+        return current_conversation, "No user logged in!", []
+
+    hf_token = auth_globals["hf_token"]
+    user_repo = auth_globals["user_repo"]
+    hf_api = HfApi(token=hf_token)
+
+    try:
+        user_file = hf_hub_download(
+            repo_id=user_repo,
+            filename=f"{username}.json",
+            repo_type="dataset",
+            token=hf_token
+        )
+
+        with open(user_file, 'r') as f:
+            data = json.load(f)
+
+        if target_conversation not in data["conversations"]:
+            return current_conversation, f"Conversation '{target_conversation}' doesn't exist!", data["conversations"][current_conversation]
+
+        data["current_conversation"] = target_conversation
+
+        hf_api.upload_file(
+            path_or_fileobj=json.dumps(data).encode('utf-8'),
+            path_in_repo=f"{username}.json",
+            repo_id=user_repo,
+            repo_type="dataset",
+            token=hf_token
+        )
+
+        return target_conversation, f"Switched to conversation: {target_conversation}", data["conversations"][target_conversation]
+
+    except Exception as e:
+        return current_conversation, f"Failed to switch conversation: {str(e)}", data["conversations"][current_conversation]
+
+def get_conversation_list(username, auth_globals):
+    if not username:
+        return []
+
+    hf_token = auth_globals["hf_token"]
+    user_repo = auth_globals["user_repo"]
+
+    try:
+        user_file = hf_hub_download(
+            repo_id=user_repo,
+            filename=f"{username}.json",
+            repo_type="dataset",
+            token=hf_token
+        )
+
+        with open(user_file, 'r') as f:
+            data = json.load(f)
+
+        return list(data["conversations"].keys())
+
+    except Exception:
+        return []