Update app.py

app.py

@@ -1,12 +1,29 @@
 import os
 import streamlit as st
 import json
+import requests
+import re
+from bs4 import BeautifulSoup
 from streamlit_option_menu import option_menu
 from gemini_utility import (load_gemini_pro, gemini_pro_vision_responce)
 from PIL import Image
-import time
 
-# Configuración de la página
+# 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",
     page_icon="🤖",
@@ -14,203 +31,144 @@ st.set_page_config(
     initial_sidebar_state="expanded",
 )
 
-# Add the missing translation function
-def translate_role_to_streamlit(role):
-    """
-    Translate Gemini message roles to Streamlit chat roles
-    """
-    role_mapping = {
-        'model': 'assistant',
-        'user': 'user'
-    }
-    return role_mapping.get(role, 'assistant')
-
-# Archivo para almacenamiento permanente
-PROMPTS_FILE = "training_prompts.json"
-
-# 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": ""}
+# Cargar el System Prompt guardado
+if "system_prompt" not in st.session_state:
+    st.session_state.system_prompt = load_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}")
+# Función para guardar el historial en cookies
+def save_chat_history(history):
+    serializable_history = []
+    for message in history:
+        serializable_history.append({
+            "role": message.role,
+            "text": message.parts[0].text
+        })
+    st.session_state.cookie_chat_history = json.dumps(serializable_history)
 
-# Inicialización de datos
-if 'prompts_data' not in st.session_state:
-    st.session_state.prompts_data = load_prompts()
+# 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=[])
+            # NO enviar el system_prompt al inicio aquí
+            for message in history:
+                if message["role"] != "model":
+                    chat.send_message(message["text"])
+            return chat
+        except Exception as e:
+            st.error(f"Error cargando el historial: {e}")
+    return None
 
 with st.sidebar:
     selected = option_menu(
         "GD AI",
-        ["System Prompts", "Chatbot", "Image Captioning"],
+        ["System Prompt", "Chatbot", "Image Captioning"],
         menu_icon="robot",
         icons=['gear', 'chat-dots-fill', 'image-fill'],
         default_index=0
     )
 
-if selected == "System Prompts":
-    st.title("Gestión de System Prompts")
-    
-    # Área para nuevo prompt
-    new_prompt = st.text_area(
-        "Nuevo System Prompt",
-        value="",
-        height=200,
-        help="Escribe aquí las nuevas instrucciones para el AI"
-    )
-    
-    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)
+    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:
-        st.warning("No hay prompt activo")
+        return user_role
+
+def extract_urls(text):
+    url_pattern = r"(https?://\S+)"
+    urls = re.findall(url_pattern, text)
+    return urls
+
+def fetch_url_content(url):
+    try:
+        response = requests.get(url, timeout=10)  # Agregar timeout para evitar bloqueos
+        response.raise_for_status()
+        return response.text
+    except requests.exceptions.RequestException as e:
+        return f"Error al acceder a la URL '{url}': {e}"
+
+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:
+        return f"Error al procesar el contenido HTML: {e}"
+
+def process_urls_in_prompt(prompt):
+    urls = extract_urls(prompt)
+    new_prompt = prompt
+    for url in urls:
+        content = fetch_url_content(url)
+        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)
+            new_prompt = new_prompt.replace(url, processed_content)
+    return new_prompt
+
+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), incluyendo URLs",
+        value=st.session_state.system_prompt,
+        height=300,
+        help="Escribe aquí las instrucciones que definirán el comportamiento del AI. Puedes incluir URLs."
+    )
+
+    if st.button("Guardar System Prompt"):
+        processed_prompt = process_urls_in_prompt(new_system_prompt)
+        st.session_state.system_prompt = processed_prompt
+        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()
-    
-    # Initialize chat session with current prompt
     if "chat_session" not in st.session_state:
-        st.session_state.chat_session = model.start_chat(history=[])
-        # Enviar el prompt actual de manera silenciosa (sin mostrarlo en la interfaz)
-        current_prompt = st.session_state.prompts_data.get("current_prompt")
-        if current_prompt:
-            st.session_state.chat_session.send_message(current_prompt)
-            # Agregar mensaje de bienvenida
-            try:
-                welcome_message = "¡Hola! Soy GnosticDev AI. ¿En qué puedo ayudarte hoy?"
-                with st.chat_message("assistant"):
-                    st.markdown(welcome_message)
-            except Exception as e:
-                st.error(f"Error al mostrar mensaje de bienvenida: {str(e)}")
+        loaded_chat = load_chat_history()
+        if loaded_chat:
+            st.session_state.chat_session = loaded_chat
+        else:
+            # Iniciar el chat con un saludo o mensaje de bienvenida simple
+            st.session_state.chat_session = model.start_chat(history=[])
+            st.chat_message("assistant").markdown("¡Hola! ¿En qué puedo ayudarte hoy?")
+            # Solo se configura el behavior con el System Prompt, pero no se muestra en el inicio
 
     st.title("Gnosticdev Chatbot")
-    
-    # Display chat history
-    if hasattr(st.session_state.chat_session, 'history'):
-        for message in st.session_state.chat_session.history:
-            # Skip displaying the system prompt message
-            if message.role == 'model' and message.parts[0].text == st.session_state.prompts_data.get("current_prompt"):
-                continue
-            role = translate_role_to_streamlit(message.role)
-            with st.chat_message(role):
-                st.markdown(message.parts[0].text)
-
-    # Chat input
     user_prompt = st.chat_input("Preguntame algo...")
     if user_prompt:
-        st.chat_message("user").markdown(user_prompt)
-        try:
-            gemini_response = st.session_state.chat_session.send_message(user_prompt)
-            with st.chat_message("assistant"):
-                st.markdown(gemini_response.text)
-        except Exception as e:
-            st.error(f"Error en la respuesta: {str(e)}")
+        processed_user_prompt = process_urls_in_prompt(user_prompt)
+        st.chat_message("user").markdown(processed_user_prompt)
+        gemini_response = st.session_state.chat_session.send_message(processed_user_prompt)
+        with st.chat_message("assistant"):
+            st.markdown(gemini_response.text)
+        save_chat_history(st.session_state.chat_session.history)
 
 elif selected == "Image Captioning":
     st.title("Image Caption Generation📸")
-    
-    # Add custom prompt option
-    use_custom_prompt = st.checkbox("Usar prompt personalizado")
-    if use_custom_prompt:
-        custom_prompt = st.text_area(
-            "Escribe tu prompt personalizado",
-            value="Write a caption for this image",
-            help="Puedes personalizar las instrucciones para el análisis de la imagen"
-        )
-    
     upload_image = st.file_uploader("Upload an image...", type=["jpg", "jpeg", "png"])
-    
-    if upload_image:
-        try:
-            image = Image.open(upload_image)
-            col1, col2 = st.columns(2)
-            
-            with col1:
-                st.image(image, caption="Uploaded Image", use_column_width=True)
-                image_info = f"Dimensiones: {image.size[0]}x{image.size[1]}"
-                st.text(image_info)
-            
-            if st.button("Generate", key="generate_caption"):
-                with st.spinner("Generando descripción..."):
-                    prompt = custom_prompt if use_custom_prompt else "Write a caption for this image"
-                    try:
-                        caption = gemini_pro_vision_responce(prompt, image)
-                        with col2:
-                            st.success("¡Descripción generada!")
-                            st.info(caption)
-                            
-                            # Opción para guardar la descripción
-                            if st.button("Guardar descripción"):
-                                timestamp = time.strftime("%Y%m%d-%H%M%S")
-                                filename = f"caption_{timestamp}.txt"
-                                try:
-                                    with open(filename, "w", encoding="utf-8") as f:
-                                        f.write(caption)
-                                    st.success(f"Descripción guardada en {filename}")
-                                except Exception as e:
-                                    st.error(f"Error al guardar la descripción: {str(e)}")
-                    
-                    except Exception as e:
-                        st.error(f"Error al generar la descripción: {str(e)}")
-                        st.error("Por favor, intenta con otra imagen o revisa tu conexión")
-        
-        except Exception as e:
-            st.error(f"Error al procesar la imagen: {str(e)}")
-            st.error("Por favor, asegúrate de que el archivo es una imagen válida")
-
-# Añadir footer
-st.markdown("---")
-col1, col2, col3 = st.columns(3)
-with col1:
-    st.markdown("**GnosticDev AI**")
-with col2:
-    st.markdown("Versión 1.0.0")
-with col3:
-    st.markdown("Made with ❤️ by GnosticDev")
+    if upload_image and st.button("Generate"):
+        image = Image.open(upload_image)
+        col1, col2 = st.columns(2)
+        with col1:
+            st.image(image, caption="Uploaded Image", use_column_width=True)
+        default_prompt = "Write a caption for this image"
+        caption = gemini_pro_vision_responce(default_prompt, image)
+        with col2:
+            st.info(caption)