Update app.py

app.py

@@ -5,7 +5,7 @@ from streamlit_option_menu import option_menu
 from gemini_utility import (load_gemini_pro, gemini_pro_vision_responce)
 from PIL import Image
 
-# Configuración de la página
+# Setting the page config
 st.set_page_config(
     page_title="GnosticDev AI",
     page_icon="🤖",
@@ -13,23 +13,26 @@ st.set_page_config(
     initial_sidebar_state="expanded",
 )
 
-# Función para guardar el historial en el estado de la sesión
+# 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
         })
-    st.session_state.chat_history = serializable_history
+    # Guardar en cookie
+    st.session_state.cookie_chat_history = json.dumps(serializable_history)
 
-# Función para cargar el historial desde el estado de la sesión
+# Función para cargar el historial desde cookies
 def load_chat_history():
-    if 'chat_history' in st.session_state:
+    if 'cookie_chat_history' in st.session_state:
         try:
-            history = st.session_state.chat_history
+            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:
@@ -40,26 +43,32 @@ def load_chat_history():
             st.error(f"Error cargando el historial: {e}")
     return None
 
+# Función para descargar el historial del chat
+def download_chat_history(history):
+    chat_text = ""
+    for message in history:
+        chat_text += f"{message.role}: {message.parts[0].text}\n"
+    return chat_text
+
 # Inicializar estados
 if "system_prompt" not in st.session_state:
-    st.session_state.system_prompt = ""
-
-if "chat_history" not in st.session_state:
-    st.session_state.chat_history = []
+    st.session_state.system_prompt = st.session_state.get('cookie_system_prompt', "")
 
 with st.sidebar:
     selected = option_menu(
         "GD AI",
-        ["Chatbot", "Image Captioning"],  # Eliminamos "System Prompt"
+        ["System Prompt", "Chatbot", "Image Captioning"],
         menu_icon="robot",
-        icons=['chat-dots-fill', 'image-fill'],
+        icons=['gear', 'chat-dots-fill', 'image-fill'],
         default_index=0
     )
     
     # Botón para borrar historial
     if st.button("Borrar Historial"):
-        st.session_state.chat_history = []
-        st.session_state.chat_session = None
+        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):
@@ -68,7 +77,28 @@ def translate_role_to_streamlit(user_role):
     else:
         return user_role
 
-if selected == "Chatbot":
+if selected == "System Prompt":
+    st.title("Configuración del System Prompt")
+    
+    new_system_prompt = st.text_area(
+        "Ingresa las instrucciones para el AI (System Prompt)",
+        value=st.session_state.system_prompt,
+        height=300,
+        help="Escribe aquí las instrucciones que definirán el comportamiento del AI"
+    )
+    
+    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
+        st.success("System Prompt actualizado con éxito!")
+        
+    if st.session_state.system_prompt:
+        st.markdown("### System Prompt Actual:")
+        st.info(st.session_state.system_prompt)
+
+elif selected == "Chatbot":
     model = load_gemini_pro()
     
     # Inicializar o cargar sesión de chat
@@ -92,7 +122,7 @@ if selected == "Chatbot":
         with st.chat_message(translate_role_to_streamlit(message.role)):
             st.markdown(message.parts[0].text)
 
-    # Campo de entrada
+        # Campo de entrada
     user_prompt = st.chat_input("Preguntame algo...")
     if user_prompt:
         st.chat_message("user").markdown(user_prompt)
@@ -105,7 +135,7 @@ if selected == "Chatbot":
 
     # Opción para descargar el historial del chat
     if st.button("Descargar Historial del Chat"):
-        chat_history = "\n".join([f"{msg['role']}: {msg['text']}" for msg in st.session_state.chat_history])
+        chat_history = download_chat_history(st.session_state.chat_session.history)
         st.download_button(
             label="Descargar",
             data=chat_history,