Update app.py

app.py

@@ -4,8 +4,9 @@ import json
 from streamlit_option_menu import option_menu
 from gemini_utility import (load_gemini_pro, gemini_pro_vision_responce)
 from PIL import Image
+import time
 
-# Setting the page config
+# Configuración de la página
 st.set_page_config(
     page_title="GnosticDev AI",
     page_icon="🤖",
@@ -13,123 +14,127 @@ st.set_page_config(
     initial_sidebar_state="expanded",
 )
 
-# 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)
+# Archivo para almacenamiento permanente
+PROMPTS_FILE = "training_prompts.json"
 
-# 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
+# Funciones para manejar el almacenamiento permanente
+def load_prompts():
+    try:
+        if os.path.exists(PROMPTS_FILE):
+            with open(PROMPTS_FILE, 'r', encoding='utf-8') as file:
+                return json.load(file)
+    except Exception as e:
+        st.error(f"Error cargando prompts: {e}")
+    return {"prompts": [], "current_prompt": ""}
 
-# Inicializar estados
-if "system_prompt" not in st.session_state:
-    st.session_state.system_prompt = st.session_state.get('cookie_system_prompt', "")
+def save_prompts(prompts_data):
+    try:
+        with open(PROMPTS_FILE, 'w', encoding='utf-8') as file:
+            json.dump(prompts_data, file, ensure_ascii=False, indent=2)
+    except Exception as e:
+        st.error(f"Error guardando prompts: {e}")
+
+# Inicialización de datos
+if 'prompts_data' not in st.session_state:
+    st.session_state.prompts_data = load_prompts()
 
 with st.sidebar:
     selected = option_menu(
         "GD AI",
-        ["System Prompt", "Chatbot", "Image Captioning"],
+        ["System Prompts", "Chatbot", "Image Captioning"],
         menu_icon="robot",
         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":
-        return "assistant"
-    else:
-        return user_role
 
-if selected == "System Prompt":
-    st.title("Configuración del System Prompt")
+if selected == "System Prompts":
+    st.title("Gestión de System Prompts")
     
-    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"
+    # Área para nuevo prompt
+    new_prompt = st.text_area(
+        "Nuevo System Prompt",
+        value="",
+        height=200,
+        help="Escribe aquí las nuevas instrucciones para el 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)
+    col1, col2 = st.columns([1, 2])
+    with col1:
+        if st.button("Añadir Nuevo Prompt"):
+            if new_prompt and new_prompt not in st.session_state.prompts_data["prompts"]:
+                st.session_state.prompts_data["prompts"].append(new_prompt)
+                save_prompts(st.session_state.prompts_data)
+                st.success("Nuevo prompt añadido!")
+                time.sleep(1)
+                st.rerun()
+    
+    # Lista de prompts guardados
+    st.markdown("### Prompts Guardados")
+    for i, prompt in enumerate(st.session_state.prompts_data["prompts"]):
+        with st.expander(f"Prompt {i+1}"):
+            st.text_area("", prompt, height=100, key=f"prompt_{i}", disabled=True)
+            col1, col2, col3 = st.columns([1, 1, 1])
+            with col1:
+                if st.button("Usar este prompt", key=f"use_{i}"):
+                    st.session_state.prompts_data["current_prompt"] = prompt
+                    save_prompts(st.session_state.prompts_data)
+                    if "chat_session" in st.session_state:
+                        del st.session_state.chat_session
+                    st.success("Prompt activado!")
+            with col2:
+                if st.button("Editar", key=f"edit_{i}"):
+                    st.session_state.editing_prompt = i
+                    st.session_state.editing_text = prompt
+            with col3:
+                if st.button("Eliminar", key=f"delete_{i}"):
+                    st.session_state.prompts_data["prompts"].pop(i)
+                    save_prompts(st.session_state.prompts_data)
+                    st.success("Prompt eliminado!")
+                    time.sleep(1)
+                    st.rerun()
+    
+    # Mostrar prompt actual
+    st.markdown("### Prompt Actual")
+    current_prompt = st.session_state.prompts_data.get("current_prompt", "")
+    if current_prompt:
+        st.info(current_prompt)
+    else:
+        st.warning("No hay prompt activo")
 
 elif selected == "Chatbot":
     model = load_gemini_pro()
     
-    # Inicializar o cargar sesión de chat
+    # Initialize chat session with current prompt
     if "chat_session" not in st.session_state:
-        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.session_state.chat_session = model.start_chat(history=[])
+        current_prompt = st.session_state.prompts_data.get("current_prompt")
+        if current_prompt:
+            st.session_state.chat_session.send_message(current_prompt)
 
     st.title("Gnosticdev Chatbot")
     
-    if st.session_state.system_prompt:
+    # Mostrar prompt actual
+    current_prompt = st.session_state.prompts_data.get("current_prompt")
+    if current_prompt:
         with st.expander("Ver System Prompt actual"):
-            st.info(st.session_state.system_prompt)
+            st.info(current_prompt)
     
-    # Mostrar historial
+    # Display chat history
     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)
 
-    # Campo de entrada
+    # Chat input
     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📸")
     upload_image = st.file_uploader("Upload an image...", type=["jpg", "jpeg", "png"])
-    
     if upload_image and st.button("Generate"):
         image = Image.open(upload_image)
         col1, col2 = st.columns(2)