Improve chat initial experience

app/lib/backend/http/api/messages.dart

@@ -23,6 +23,16 @@ Future<List<ServerMessage>> getMessagesServer() async {
   return [];
 }
 
+Future<List<ServerMessage>> clearChatServer() async {
+  var response = await makeApiCall(url: '${Env.apiBaseUrl}v1/clear-chat', headers: {}, method: 'DELETE', body: '');
+  if (response == null) throw Exception('Failed to delete chat');
+  if (response.statusCode == 200) {
+    return [ServerMessage.fromJson(jsonDecode(response.body))];
+  } else {
+    throw Exception('Failed to delete chat');
+  }
+}
+
 Future<ServerMessage> sendMessageServer(String text, {String? pluginId}) {
   return makeApiCall(
     url: '${Env.apiBaseUrl}v1/messages?plugin_id=$pluginId',

app/lib/main.dart

@@ -274,8 +274,10 @@ class _DeciderWidgetState extends State<DeciderWidget> {
       if (context.read<ConnectivityProvider>().isConnected) {
         NotificationService.instance.saveNotificationToken();
       }
+
       if (context.read<AuthenticationProvider>().user != null) {
-        context.read<HomeProvider>().setupHasSpeakerProfile();
+        context.read<MessageProvider>().setMessagesFromCache();
+        context.read<MessageProvider>().refreshMessages();
       }
     });
     super.initState();

app/lib/pages/chat/widgets/ai_message.dart

@@ -9,7 +9,9 @@ import 'package:friend_private/backend/preferences.dart';
 import 'package:friend_private/backend/schema/memory.dart';
 import 'package:friend_private/backend/schema/message.dart';
 import 'package:friend_private/backend/schema/plugin.dart';
+import 'package:friend_private/pages/memory_detail/memory_detail_provider.dart';
 import 'package:friend_private/pages/memory_detail/page.dart';
+import 'package:friend_private/providers/memory_provider.dart';
 import 'package:friend_private/utils/analytics/mixpanel.dart';
 import 'package:friend_private/providers/connectivity_provider.dart';
 import 'package:friend_private/utils/other/temp.dart';
@@ -116,8 +118,8 @@ class _AIMessageState extends State<AIMessage> {
                 style: TextStyle(fontSize: 15.0, fontWeight: FontWeight.w500, color: Colors.grey.shade300),
               )),
               if (widget.message.id != 1) _getCopyButton(context), // RESTORE ME
-              // if (message.id == 1 && displayOptions) const SizedBox(height: 8),
-              // if (message.id == 1 && displayOptions) ..._getInitialOptions(context),
+              if (widget.displayOptions) const SizedBox(height: 8),
+              if (widget.displayOptions) ..._getInitialOptions(context),
               if (messageMemories.isNotEmpty) ...[
                 const SizedBox(height: 16),
                 for (var data in messageMemories.indexed) ...[
@@ -127,30 +129,58 @@ class _AIMessageState extends State<AIMessage> {
                       onTap: () async {
                         final connectivityProvider = Provider.of<ConnectivityProvider>(context, listen: false);
                         if (connectivityProvider.isConnected) {
-                          if (memoryDetailLoading[data.$1]) return;
-                          setState(() => memoryDetailLoading[data.$1] = true);
+                          var memProvider = Provider.of<MemoryProvider>(context, listen: false);
+                          var idx = memProvider.memoriesWithDates.indexWhere((e) {
+                            if (e.runtimeType == ServerMemory) {
+                              return e.id == data.$2.id;
+                            }
+                            return false;
+                          });
 
-                          ServerMemory? m = await getMemoryById(data.$2.id);
-                          if (m == null) return;
-                          MixpanelManager().chatMessageMemoryClicked(m);
-                          setState(() => memoryDetailLoading[data.$1] = false);
-                          await Navigator.of(context)
-                              .push(MaterialPageRoute(builder: (c) => MemoryDetailPage(memory: m)));
-                          if (SharedPreferencesUtil().modifiedMemoryDetails?.id == m.id) {
-                            ServerMemory modifiedDetails = SharedPreferencesUtil().modifiedMemoryDetails!;
-                            widget.updateMemory(SharedPreferencesUtil().modifiedMemoryDetails!);
-                            var copy = List<MessageMemory>.from(widget.message.memories);
-                            copy[data.$1] = MessageMemory(
-                                modifiedDetails.id,
-                                modifiedDetails.createdAt,
-                                MessageMemoryStructured(
-                                  modifiedDetails.structured.title,
-                                  modifiedDetails.structured.emoji,
-                                ));
-                            widget.message.memories.clear();
-                            widget.message.memories.addAll(copy);
-                            SharedPreferencesUtil().modifiedMemoryDetails = null;
-                            setState(() {});
+                          if (idx != -1) {
+                            context.read<MemoryDetailProvider>().updateMemory(idx);
+                            var m = memProvider.memoriesWithDates[idx];
+                            MixpanelManager().chatMessageMemoryClicked(m);
+                            await Navigator.of(context).push(
+                              MaterialPageRoute(
+                                builder: (c) => MemoryDetailPage(
+                                  memory: m,
+                                ),
+                              ),
+                            );
+                          } else {
+                            if (memoryDetailLoading[data.$1]) return;
+                            setState(() => memoryDetailLoading[data.$1] = true);
+                            ServerMemory? m = await getMemoryById(data.$2.id);
+                            if (m == null) return;
+                            idx = memProvider.addMemoryWithDate(m);
+                            MixpanelManager().chatMessageMemoryClicked(m);
+                            setState(() => memoryDetailLoading[data.$1] = false);
+                            context.read<MemoryDetailProvider>().updateMemory(idx);
+                            await Navigator.of(context).push(
+                              MaterialPageRoute(
+                                builder: (c) => MemoryDetailPage(
+                                  memory: m,
+                                ),
+                              ),
+                            );
+                            //TODO: Not needed anymore I guess because memories are stored in provider and read from there only
+                            if (SharedPreferencesUtil().modifiedMemoryDetails?.id == m.id) {
+                              ServerMemory modifiedDetails = SharedPreferencesUtil().modifiedMemoryDetails!;
+                              widget.updateMemory(SharedPreferencesUtil().modifiedMemoryDetails!);
+                              var copy = List<MessageMemory>.from(widget.message.memories);
+                              copy[data.$1] = MessageMemory(
+                                  modifiedDetails.id,
+                                  modifiedDetails.createdAt,
+                                  MessageMemoryStructured(
+                                    modifiedDetails.structured.title,
+                                    modifiedDetails.structured.emoji,
+                                  ));
+                              widget.message.memories.clear();
+                              widget.message.memories.addAll(copy);
+                              SharedPreferencesUtil().modifiedMemoryDetails = null;
+                              setState(() {});
+                            }
                           }
                         } else {
                           ScaffoldMessenger.of(context).showSnackBar(
@@ -256,7 +286,7 @@ class _AIMessageState extends State<AIMessage> {
   _getInitialOption(BuildContext context, String optionText) {
     return GestureDetector(
       child: Container(
-        padding: const EdgeInsets.symmetric(horizontal: 12.0, vertical: 8),
+        padding: const EdgeInsets.symmetric(horizontal: 12.0, vertical: 10),
         width: double.maxFinite,
         decoration: BoxDecoration(
           color: Colors.grey.shade900,
@@ -273,11 +303,11 @@ class _AIMessageState extends State<AIMessage> {
   _getInitialOptions(BuildContext context) {
     return [
       const SizedBox(height: 8),
-      _getInitialOption(context, 'What tasks do I have from yesterday?'),
+      _getInitialOption(context, 'What\'s been on my mind a lot?'),
       const SizedBox(height: 8),
-      _getInitialOption(context, 'What conversations did I have with John?'),
+      _getInitialOption(context, 'Did I forget to follow up on something?'),
       const SizedBox(height: 8),
-      _getInitialOption(context, 'What advise have I received about entrepreneurship?'),
+      _getInitialOption(context, 'What\'s the funniest thing I\'ve said lately?'),
     ];
   }
 }

app/lib/pages/chat/widgets/animated_mini_banner.dart

@@ -0,0 +1,20 @@
+import 'package:flutter/material.dart';
+
+class AnimatedMiniBanner extends StatelessWidget implements PreferredSizeWidget {
+  const AnimatedMiniBanner({super.key, required this.showAppBar, required this.child});
+
+  final bool showAppBar;
+  final Widget child;
+
+  @override
+  Widget build(BuildContext context) {
+    return AnimatedContainer(
+      height: showAppBar ? kToolbarHeight : 0,
+      duration: const Duration(milliseconds: 400),
+      child: child,
+    );
+  }
+
+  @override
+  Size get preferredSize => const Size.fromHeight(30);
+}

app/lib/providers/memory_provider.dart

@@ -106,8 +106,34 @@ class MemoryProvider extends ChangeNotifier {
 
   void addMemory(ServerMemory memory) {
     memories.insert(0, memory);
-    filterMemories('');
+    initFilteredMemories();
+    notifyListeners();
+  }
+
+  int addMemoryWithDate(ServerMemory memory) {
+    int idx;
+    var date = memoriesWithDates.indexWhere((element) =>
+        element is DateTime &&
+        element.day == memory.createdAt.day &&
+        element.month == memory.createdAt.month &&
+        element.year == memory.createdAt.year);
+    if (date != -1) {
+      var hour = memoriesWithDates[date + 1].createdAt.hour;
+      var newHour = memory.createdAt.hour;
+      if (newHour > hour) {
+        memoriesWithDates.insert(date + 1, memory);
+        idx = date + 1;
+      } else {
+        memoriesWithDates.insert(date + 2, memory);
+        idx = date + 2;
+      }
+    } else {
+      memoriesWithDates.add(memory.createdAt);
+      memoriesWithDates.add(memory);
+      idx = memoriesWithDates.length - 1;
+    }
     notifyListeners();
+    return idx;
   }
 
   void updateMemory(ServerMemory memory, [int? index]) {

app/lib/providers/message_provider.dart

@@ -10,11 +10,25 @@ class MessageProvider extends ChangeNotifier {
   List<ServerMessage> messages = [];
 
   bool isLoadingMessages = false;
+  bool hasCachedMessages = false;
+  bool isClearingChat = false;
+
+  String firstTimeLoadingText = '';
 
   void updatePluginProvider(PluginProvider p) {
     pluginProvider = p;
   }
 
+  void setHasCachedMessages(bool value) {
+    hasCachedMessages = value;
+    notifyListeners();
+  }
+
+  void setClearingChat(bool value) {
+    isClearingChat = value;
+    notifyListeners();
+  }
+
   void setLoadingMessages(bool value) {
     isLoadingMessages = value;
     notifyListeners();
@@ -32,15 +46,39 @@ class MessageProvider extends ChangeNotifier {
     notifyListeners();
   }
 
+  void setMessagesFromCache() {
+    if (SharedPreferencesUtil().cachedMessages.isNotEmpty) {
+      setHasCachedMessages(true);
+      messages = SharedPreferencesUtil().cachedMessages;
+    }
+    notifyListeners();
+  }
+
   Future<List<ServerMessage>> getMessagesFromServer() async {
+    if (!hasCachedMessages) {
+      firstTimeLoadingText = 'Reading your memories...';
+      notifyListeners();
+    }
     setLoadingMessages(true);
     var mes = await getMessagesServer();
+    if (!hasCachedMessages) {
+      firstTimeLoadingText = 'Learning from your memories...';
+      notifyListeners();
+    }
     messages = mes;
     setLoadingMessages(false);
     notifyListeners();
     return messages;
   }
 
+  Future clearChat() async {
+    setClearingChat(true);
+    var mes = await clearChatServer();
+    messages = mes;
+    setClearingChat(false);
+    notifyListeners();
+  }
+
   void addMessage(ServerMessage message) {
     messages.insert(0, message);
     notifyListeners();

backend/database/chat.py

@@ -2,6 +2,7 @@
 from datetime import datetime, timezone
 from typing import Optional
 
+from fastapi import HTTPException
 from google.cloud import firestore
 
 from models.chat import Message
@@ -82,3 +83,27 @@ def get_messages(uid: str, limit: int = 20, offset: int = 0, include_memories: b
         ]
 
     return messages
+
+
+async def batch_delete_messages(parent_doc_ref, batch_size=450):
+    # batch size is 450 because firebase can perform upto 500 operations in a batch
+    messages_ref = parent_doc_ref.collection('messages')
+    while True:
+        docs = messages_ref.limit(batch_size).stream()
+        docs_list = list(docs)
+        if not docs_list:
+            break
+        batch = db.batch()
+        for doc in docs_list:
+            batch.delete(doc.reference)
+        batch.commit()
+
+
+async def clear_chat( uid: str):
+    try:
+        user_ref = db.collection('users').document(uid)
+        if not user_ref.get().exists:
+            raise HTTPException(status_code=404, detail="User not found")
+        await batch_delete_messages(user_ref)
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=f"Error deleting messages: {str(e)}")

backend/routers/chat.py

@@ -55,6 +55,12 @@ def send_message(
     ai_message.memories = memories if len(memories) < 5 else memories[:5]
     return ai_message
 
+@router.delete('/v1/clear-chat', tags=['chat'], response_model=Message)
+def clear_chat(uid: str = Depends(auth.get_current_user_uid)):
+    chat_db.clear_chat(uid)
+    return initial_message_util(uid)
+
+
 
 def initial_message_util(uid: str, plugin_id: Optional[str] = None):
     plugin = get_plugin_by_id(plugin_id)