Update app.py

app.py

@@ -11,6 +11,8 @@ import torch
 import numpy as np
 import networkx as nx
 from collections import Counter
+import json
+from datetime import datetime
 
 @dataclass
 class ChatMessage:
@@ -93,6 +95,11 @@ class XylariaChat:
             "photosynthesis": "process used by plants to convert light to energy"
         }
 
+        # Initialize chat history file
+        self.chat_history_file = "chat_history.json"
+        self.load_all_chats()
+        
+
     def update_internal_state(self, emotion_deltas, cognitive_load_deltas, introspection_delta, engagement_delta):
         for emotion, delta in emotion_deltas.items():
             if emotion in self.internal_state["emotions"]:
@@ -559,6 +566,52 @@ class XylariaChat:
         prompt += "<|assistant|>\n"
         return prompt
 
+    def save_chat(self):
+        chat_data = {
+            "timestamp": datetime.now().strftime("%Y-%m-%d %H:%M:%S"),
+            "conversation": self.conversation_history
+        }
+        try:
+            with open(self.chat_history_file, "r") as f:
+                all_chats = json.load(f)
+        except (FileNotFoundError, json.JSONDecodeError):
+            all_chats = []
+
+        all_chats.append(chat_data)
+
+        with open(self.chat_history_file, "w") as f:
+            json.dump(all_chats, f)
+
+    def load_all_chats(self):
+        try:
+            with open(self.chat_history_file, "r") as f:
+                all_chats = json.load(f)
+        except (FileNotFoundError, json.JSONDecodeError):
+            all_chats = []
+        return all_chats
+
+    def load_chat(self, chat_index):
+        all_chats = self.load_all_chats()
+        if 0 <= chat_index < len(all_chats):
+            self.conversation_history = all_chats[chat_index]["conversation"]
+            self.reset_conversation()
+            for msg in self.conversation_history:
+                if msg['role'] == 'user':
+                    self.dynamic_belief_update(msg['content'])
+            return self.conversation_history
+        else:
+            raise ValueError("Invalid chat index")
+            
+    def delete_chat(self, chat_index):
+        all_chats = self.load_all_chats()
+        if 0 <= chat_index < len(all_chats):
+            del all_chats[chat_index]
+            with open(self.chat_history_file, "w") as f:
+                json.dump(all_chats, f)
+            return self.load_all_chats()
+        else:
+            raise ValueError("Invalid chat index")
+
     def create_interface(self):
         loading_svg = """<svg width="256" height="256" viewBox="0 0 256 256" xmlns="http://www.w3.org/2000/svg">
             <style>
@@ -580,7 +633,7 @@ class XylariaChat:
           <text x="50%" y="50%" dominant-baseline="middle" text-anchor="middle" font-size="24" fill="white" opacity="0.8">
             <tspan>creating your image</tspan>
             <tspan x="50%" dy="1.2em">with xylaria iris</tspan>
-          </text>
+          </text>
         </svg>"""
             
         def streaming_response(message, chat_history, image_filepath, math_ocr_image_path):
@@ -628,7 +681,7 @@ class XylariaChat:
 
             if image_filepath:
                 response_stream = self.get_response(message, image_filepath)
-            else:
+                        else:
                 response_stream = self.get_response(message)
                 
             if isinstance(response_stream, str):
@@ -697,6 +750,26 @@ class XylariaChat:
             if len(self.conversation_history) > 10:
                 self.conversation_history = self.conversation_history[-10:]
 
+            self.save_chat()
+            all_chats = self.load_all_chats()
+            chat_titles = [f"{chat['timestamp']}: {chat['conversation'][0]['content'][:30]}..." if len(chat['conversation']) > 0 and chat['conversation'][0]['content'] else f"{chat['timestamp']}: Empty Chat" for chat in all_chats]
+            yield "", updated_history, None, None, gr.update(choices=chat_titles)
+
+        def load_selected_chat(chat_index, evt: gr.SelectData):
+            if chat_index is not None:
+                loaded_chat = self.load_chat(evt.index)
+                return loaded_chat
+            else:
+                return []
+        
+        def delete_selected_chat(chat_index, evt: gr.SelectData):
+            if chat_index is not None:
+                all_chats = self.delete_chat(evt.index)
+                chat_titles = [f"{chat['timestamp']}: {chat['conversation'][0]['content'][:30]}..." if len(chat['conversation']) > 0 and chat['conversation'][0]['content'] else f"{chat['timestamp']}: Empty Chat" for chat in all_chats]
+                return gr.update(choices=chat_titles)
+            else:
+                return gr.update()
+
         custom_css = """
         @import url('https://fonts.googleapis.com/css2?family=Inter:wght@300;400;500;600;700&display=swap');
         body, .gradio-container {
@@ -771,72 +844,90 @@ class XylariaChat:
             display: flex;
             flex-direction: column-reverse;
         }
+        #chat_list {
+            height: 500px;
+            overflow-y: auto;
+        }
         """
 
         with gr.Blocks(theme='soft', css=custom_css) as demo:
-            with gr.Column():
-                chatbot = gr.Chatbot(
-                    label="Xylaria 1.5 Senoa",
-                    height=500,
-                    show_copy_button=True,
-                )
-
-                with gr.Accordion("Image Input", open=False, elem_classes="gr-accordion"):
-                    with gr.Row(elem_classes="image-container"):
-                        with gr.Column(elem_classes="image-upload"):
-                            img = gr.Image(
-                                sources=["upload", "webcam"],
-                                type="filepath",
-                                label="Upload Image",
-                                elem_classes="image-preview"
-                            )
-                        with gr.Column(elem_classes="image-upload"):
-                            math_ocr_img = gr.Image(
-                                sources=["upload", "webcam"],
-                                type="filepath",
-                                label="Upload Image for Math OCR",
-                                elem_classes="image-preview"
+            with gr.Row():
+                with gr.Column(scale=1):
+                    all_chats = self.load_all_chats()
+                    chat_titles = [f"{chat['timestamp']}: {chat['conversation'][0]['content'][:30]}..." if len(chat['conversation']) > 0 and chat['conversation'][0]['content'] else f"{chat['timestamp']}: Empty Chat" for chat in all_chats]
+
+                    chat_list = gr.Radio(label="Chat History", choices=chat_titles, type="index", elem_id="chat_list")
+                    
+                    load_button = gr.Button("Load Selected Chat")
+                    delete_button = gr.Button("Delete Selected Chat")
+                with gr.Column(scale=4):
+                    chatbot = gr.Chatbot(
+                        label="Xylaria 1.5 Senoa",
+                        height=500,
+                        show_copy_button=True,
+                    )
+
+                    with gr.Accordion("Image Input", open=False, elem_classes="gr-accordion"):
+                        with gr.Row(elem_classes="image-container"):
+                            with gr.Column(elem_classes="image-upload"):
+                                img = gr.Image(
+                                    sources=["upload", "webcam"],
+                                    type="filepath",
+                                    label="Upload Image",
+                                    elem_classes="image-preview"
+                                )
+                            with gr.Column(elem_classes="image-upload"):
+                                math_ocr_img = gr.Image(
+                                    sources=["upload", "webcam"],
+                                    type="filepath",
+                                    label="Upload Image for Math OCR",
+                                    elem_classes="image-preview"
+                                )
+
+                    with gr.Row():
+                        with gr.Column(scale=4):
+                            txt = gr.Textbox(
+                                show_label=False,
+                                placeholder="Type your message...",
+                                container=False
                             )
+                        btn = gr.Button("Send", scale=1)
+
+                    with gr.Row():
+                        clear = gr.Button("Clear Conversation")
+                        clear_memory = gr.Button("Clear Memory")
+            
+            load_button.click(fn=load_selected_chat, inputs=[chat_list], outputs=[chatbot])
+            delete_button.click(fn=delete_selected_chat, inputs=[chat_list], outputs=[chat_list])
+
+            btn.click(
+                fn=streaming_response,
+                inputs=[txt, chatbot, img, math_ocr_img],
+                outputs=[txt, chatbot, img, math_ocr_img, chat_list]
+            )
+            txt.submit(
+                fn=streaming_response,
+                inputs=[txt, chatbot, img, math_ocr_img],
+                outputs=[txt, chatbot, img, math_ocr_img, chat_list]
+            )
+
+            clear.click(
+                fn=lambda: None,
+                inputs=None,
+                outputs=[chatbot],
+                queue=False
+            )
+
+            clear_memory.click(
+                fn=self.reset_conversation,
+                inputs=None,
+                outputs=[chatbot],
+                queue=False
+            )
+            
+            chat_list.select(fn=load_selected_chat, inputs=[chat_list], outputs=[chatbot])
 
-                with gr.Row():
-                    with gr.Column(scale=4):
-                        txt = gr.Textbox(
-                            show_label=False,
-                            placeholder="Type your message...",
-                            container=False
-                        )
-                    btn = gr.Button("Send", scale=1)
-
-                with gr.Row():
-                    clear = gr.Button("Clear Conversation")
-                    clear_memory = gr.Button("Clear Memory")
-
-                btn.click(
-                    fn=streaming_response,
-                    inputs=[txt, chatbot, img, math_ocr_img],
-                    outputs=[txt, chatbot, img, math_ocr_img]
-                )
-                txt.submit(
-                    fn=streaming_response,
-                    inputs=[txt, chatbot, img, math_ocr_img],
-                    outputs=[txt, chatbot, img, math_ocr_img]
-                )
-
-                clear.click(
-                    fn=lambda: None,
-                    inputs=None,
-                    outputs=[chatbot],
-                    queue=False
-                )
-
-                clear_memory.click(
-                    fn=self.reset_conversation,
-                    inputs=None,
-                    outputs=[chatbot],
-                    queue=False
-                )
-
-                demo.load(self.reset_conversation, None, None)
+            demo.load(self.reset_conversation, None, None)
 
         return demo