Update app.py

app.py

@@ -1,5 +1,6 @@
 import os
 import streamlit as st
+import json
 from streamlit_option_menu import option_menu
 from gemini_utility import (load_gemini_pro, gemini_pro_vision_responce)
 from PIL import Image
@@ -12,9 +13,39 @@ st.set_page_config(
     initial_sidebar_state="expanded",
 )
 
-# Inicializar el system prompt en session state si no existe
+# Función para guardar el historial en cookies
+def save_chat_history(history):
+    # Convertir el historial a un formato serializable
+    serializable_history = []
+    for message in history:
+        serializable_history.append({
+            "role": message.role,
+            "text": message.parts[0].text
+        })
+    # Guardar en cookie
+    st.session_state.cookie_chat_history = json.dumps(serializable_history)
+
+# Función para cargar el historial desde cookies
+def load_chat_history():
+    if 'cookie_chat_history' in st.session_state:
+        try:
+            history = json.loads(st.session_state.cookie_chat_history)
+            model = load_gemini_pro()
+            chat = model.start_chat(history=[])
+            # Reconstruir el historial
+            if st.session_state.system_prompt:
+                chat.send_message(st.session_state.system_prompt)
+            for message in history:
+                if message["role"] != "model" or not message["text"].startswith(st.session_state.system_prompt):
+                    chat.send_message(message["text"])
+            return chat
+        except Exception as e:
+            st.error(f"Error cargando el historial: {e}")
+    return None
+
+# Inicializar estados
 if "system_prompt" not in st.session_state:
-    st.session_state.system_prompt = ""
+    st.session_state.system_prompt = st.session_state.get('cookie_system_prompt', "")
 
 with st.sidebar:
     selected = option_menu(
@@ -24,6 +55,14 @@ with st.sidebar:
         icons=['gear', 'chat-dots-fill', 'image-fill'],
         default_index=0
     )
+    
+    # Botón para borrar historial
+    if st.button("Borrar Historial"):
+        if 'cookie_chat_history' in st.session_state:
+            del st.session_state.cookie_chat_history
+        if 'chat_session' in st.session_state:
+            del st.session_state.chat_session
+        st.success("Historial borrado!")
 
 def translate_role_to_streamlit(user_role):
     if user_role == "model":
@@ -34,7 +73,6 @@ def translate_role_to_streamlit(user_role):
 if selected == "System Prompt":
     st.title("Configuración del System Prompt")
     
-    # Área para editar el system prompt
     new_system_prompt = st.text_area(
         "Ingresa las instrucciones para el AI (System Prompt)",
         value=st.session_state.system_prompt,
@@ -44,11 +82,11 @@ if selected == "System Prompt":
     
     if st.button("Guardar System Prompt"):
         st.session_state.system_prompt = new_system_prompt
+        st.session_state.cookie_system_prompt = new_system_prompt  # Guardar en cookie
         if "chat_session" in st.session_state:
-            del st.session_state.chat_session  # Reset chat session
+            del st.session_state.chat_session
         st.success("System Prompt actualizado con éxito!")
         
-    # Mostrar el prompt actual
     if st.session_state.system_prompt:
         st.markdown("### System Prompt Actual:")
         st.info(st.session_state.system_prompt)
@@ -56,31 +94,37 @@ if selected == "System Prompt":
 elif selected == "Chatbot":
     model = load_gemini_pro()
     
-    # Initialize chat session with system prompt
+    # Inicializar o cargar sesión de chat
     if "chat_session" not in st.session_state:
-        st.session_state.chat_session = model.start_chat(history=[])
-        if st.session_state.system_prompt:
-            st.session_state.chat_session.send_message(st.session_state.system_prompt)
+        loaded_chat = load_chat_history()
+        if loaded_chat:
+            st.session_state.chat_session = loaded_chat
+        else:
+            st.session_state.chat_session = model.start_chat(history=[])
+            if st.session_state.system_prompt:
+                st.session_state.chat_session.send_message(st.session_state.system_prompt)
 
     st.title("Gnosticdev Chatbot")
     
-    # Mostrar el system prompt actual en un expander
     if st.session_state.system_prompt:
         with st.expander("Ver System Prompt actual"):
             st.info(st.session_state.system_prompt)
     
-    # Display the chatbot history
+    # Mostrar historial
     for message in st.session_state.chat_session.history:
         with st.chat_message(translate_role_to_streamlit(message.role)):
             st.markdown(message.parts[0].text)
 
-    # Input field for user's message
+    # Campo de entrada
     user_prompt = st.chat_input("Preguntame algo...")
     if user_prompt:
         st.chat_message("user").markdown(user_prompt)
         gemini_response = st.session_state.chat_session.send_message(user_prompt)
         with st.chat_message("assistant"):
             st.markdown(gemini_response.text)
+        
+        # Guardar historial actualizado
+        save_chat_history(st.session_state.chat_session.history)
 
 elif selected == "Image Captioning":
     st.title("Image Caption Generation📸")
@@ -94,4 +138,4 @@ elif selected == "Image Captioning":
         default_prompt = "Write a caption for this image"
         caption = gemini_pro_vision_responce(default_prompt, image)
         with col2:
-            st.info(caption)
+            st.info(caption)