Update app.py

app.py

@@ -8,6 +8,21 @@ from streamlit_option_menu import option_menu
 from gemini_utility import (load_gemini_pro, gemini_pro_vision_responce)
 from PIL import Image
 
+# Ruta para guardar el System Prompt
+SYSTEM_PROMPT_FILE = "system_prompt.json"
+
+# Función para cargar el System Prompt desde el archivo
+def load_system_prompt():
+    if os.path.exists(SYSTEM_PROMPT_FILE):
+        with open(SYSTEM_PROMPT_FILE, 'r') as f:
+            return json.load(f).get("system_prompt", "")
+    return ""
+
+# Función para guardar el System Prompt en el archivo
+def save_system_prompt(system_prompt):
+    with open(SYSTEM_PROMPT_FILE, 'w') as f:
+        json.dump({"system_prompt": system_prompt}, f)
+
 # Setting the page config
 st.set_page_config(
     page_title="GnosticDev AI",
@@ -16,35 +31,11 @@ st.set_page_config(
     initial_sidebar_state="expanded",
 )
 
-CONFIG_FILE = "config.json"
-
-# Función para cargar configuraciones desde un archivo JSON
-def load_config():
-    if os.path.exists(CONFIG_FILE):
-        try:
-            with open(CONFIG_FILE, "r") as file:
-                return json.load(file)
-        except Exception as e:
-            st.error(f"Error cargando configuraciones: {e}")
-    return {}
-
-# Función para guardar configuraciones en un archivo JSON
-def save_config(config):
-    try:
-        with open(CONFIG_FILE, "w") as file:
-            json.dump(config, file, indent=4)
-    except Exception as e:
-        st.error(f"Error guardando configuraciones: {e}")
-
-# Cargar configuraciones al inicio
-config = load_config()
-if "system_prompt" not in config:
-    config["system_prompt"] = ""
-
-# Inicializar estados
+# Cargar el System Prompt guardado
 if "system_prompt" not in st.session_state:
-    st.session_state.system_prompt = config["system_prompt"]
+    st.session_state.system_prompt = load_system_prompt()
 
+# Función para guardar el historial en cookies
 def save_chat_history(history):
     serializable_history = []
     for message in history:
@@ -54,6 +45,7 @@ def save_chat_history(history):
         })
     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:
@@ -99,7 +91,7 @@ def extract_urls(text):
 
 def fetch_url_content(url):
     try:
-        response = requests.get(url, timeout=10)
+        response = requests.get(url, timeout=10)  # Agregar timeout para evitar bloqueos
         response.raise_for_status()
         return response.text
     except requests.exceptions.RequestException as e:
@@ -108,6 +100,7 @@ def fetch_url_content(url):
 def process_url_content(content):
     try:
         soup = BeautifulSoup(content, "html.parser")
+        # Extrae solo el texto del cuerpo principal, ignorando etiquetas de scripts y estilos
         text = soup.get_text(" ", strip=True)
         return text
     except Exception as e:
@@ -118,7 +111,7 @@ def process_urls_in_prompt(prompt):
     new_prompt = prompt
     for url in urls:
         content = fetch_url_content(url)
-        if content.startswith("Error"):
+        if content.startswith("Error"):  # Gestion de errores al obtener el contenido de la URL
             new_prompt = new_prompt.replace(url, content)
         else:
             processed_content = process_url_content(content)
@@ -137,13 +130,14 @@ if selected == "System Prompt":
     if st.button("Guardar System Prompt"):
         processed_prompt = process_urls_in_prompt(new_system_prompt)
         st.session_state.system_prompt = processed_prompt
-        config["system_prompt"] = processed_prompt
-        save_config(config)
+        save_system_prompt(processed_prompt)  # Guardar en archivo JSON
         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()
@@ -152,18 +146,11 @@ elif selected == "Chatbot":
         if loaded_chat:
             st.session_state.chat_session = loaded_chat
         else:
-            # Crea una nueva sesión de chat sin enviar el system_prompt como mensaje inicial
             st.session_state.chat_session = model.start_chat(history=[])
+            # No enviar automáticamente el system_prompt aquí
 
     st.title("Gnosticdev Chatbot")
-
-    # Muestra solo el historial del chat, si existe
-    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)
-
-    # Entrada del usuario
-    user_prompt = st.chat_input("Pregúntame algo...")
+    user_prompt = st.chat_input("Preguntame algo...")
     if user_prompt:
         processed_user_prompt = process_urls_in_prompt(user_prompt)
         st.chat_message("user").markdown(processed_user_prompt)