Update app.py

app.py

@@ -1,8 +1,11 @@
 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 gemini_utility import (load_gemini_pro, gemini_pro_vision_responce) # Asegúrate de que este archivo existe y contiene las funciones
 from PIL import Image
 
 # Setting the page config
@@ -13,26 +16,23 @@ st.set_page_config(
     initial_sidebar_state="expanded",
 )
 
-# Función para guardar el historial en cookies
+# Función para guardar el historial en cookies (sin cambios)
 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
+# Función para cargar el historial desde cookies (sin cambios)
 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:
@@ -43,7 +43,7 @@ def load_chat_history():
             st.error(f"Error cargando el historial: {e}")
     return None
 
-# Inicializar estados
+# Inicializar estados (sin cambios)
 if "system_prompt" not in st.session_state:
     st.session_state.system_prompt = st.session_state.get('cookie_system_prompt', "")
 
@@ -55,8 +55,7 @@ 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
@@ -70,31 +69,66 @@ def translate_role_to_streamlit(user_role):
     else:
         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)",
+        "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"
+        help="Escribe aquí las instrucciones que definirán el comportamiento del AI. Puedes incluir URLs."
     )
-    
+
     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
+        processed_prompt = process_urls_in_prompt(new_system_prompt)
+        st.session_state.system_prompt = processed_prompt
+        st.session_state.cookie_system_prompt = processed_prompt
         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
     if "chat_session" not in st.session_state:
         loaded_chat = load_chat_history()
         if loaded_chat:
@@ -105,31 +139,26 @@ elif selected == "Chatbot":
                 st.session_state.chat_session.send_message(st.session_state.system_prompt)
 
     st.title("Gnosticdev Chatbot")
-    
     if st.session_state.system_prompt:
         with st.expander("Ver System Prompt actual"):
             st.info(st.session_state.system_prompt)
-    
-    # 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)
 
-    # 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)
+        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)
-        
-        # 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)