Update app.py

app.py

@@ -5,11 +5,20 @@ from streamlit_option_menu import option_menu
 from gemini_utility import (load_gemini_pro, gemini_pro_vision_responce)
 from PIL import Image
 
+# Función para traducir el rol a un formato que Streamlit entienda
+def translate_role_to_streamlit(role):
+    if role == "user":
+        return "user"
+    elif role == "assistant":
+        return "assistant"
+    else:
+        return "default"  # Manejar otros roles si es necesario
+
 # Configuración de la página
 st.set_page_config(
     page_title="GnosticDev AI",
     page_icon="🤖",
-    layout="wide",  # Asegúrate de que el layout sea 'wide'
+    layout="wide",
     initial_sidebar_state="expanded",
 )
 
@@ -19,7 +28,7 @@ selected = option_menu(
     options=["System Prompt", "Chatbot", "Image Captioning"],
     icons=["gear", "chat", "camera"],
     default_index=0,
-    orientation="vertical"  # Asegúrate de que esté en vertical
+    orientation="vertical"
 )
 
 # Inicializar el estado de la sesión
@@ -70,8 +79,8 @@ def download_chat_history(history):
 def fetch_url_content(url):
     try:
         response = requests.get(url)
-        response.raise_for_status()  # Lanza un error si la respuesta es un código de error
-        return response.text  # Devuelve el contenido de la página
+        response.raise_for_status()
+        return response.text
     except requests.RequestException as e:
         st.error(f"Error al acceder a {url}: {e}")
         return None
@@ -86,7 +95,6 @@ if selected == "System Prompt":
         help="Escribe aquí las instrucciones que definirán el comportamiento del AI"
     )
     
-    # Campo para ingresar URLs
     urls_input = st.text_area(
         "Ingresa URLs de información y documentos (separadas por comas)",
         value=", ".join(st.session_state.cookie_urls),
@@ -96,7 +104,7 @@ if selected == "System Prompt":
     
     if st.button("Guardar System Prompt y URLs"):
         st.session_state.system_prompt = new_system_prompt
-        st.session_state.cookie_urls = [url.strip() for url in urls_input.split(",") if url.strip()]  # Guardar las URLs en una lista
+        st.session_state.cookie_urls = [url.strip() for url in urls_input.split(",") if url.strip()]
         if "chat_session" in st.session_state:
             del st.session_state.chat_session
         st.success("System Prompt y URLs actualizados con éxito!")
@@ -112,8 +120,7 @@ if selected == "System Prompt":
 elif selected == "Chatbot":
     model = load_gemini_pro()
     
-    # Inicializar o cargar sesión de chat
-    if "chat_session" not in st.session_state:
+        if "chat_session" not in st.session_state:
         loaded_chat = load_chat_history()
         if loaded_chat:
             st.session_state.chat_session = loaded_chat
@@ -150,7 +157,6 @@ elif selected == "Chatbot":
                     fetched_contents.append(content)
 
             # Aquí puedes procesar el contenido obtenido de las URLs
-            # Por ejemplo, podrías resumirlo o extraer información relevante
             combined_content = "\n\n".join(fetched_contents)
             user_prompt += f"\n\nInformación adicional de URLs:\n{combined_content}"
 
@@ -186,5 +192,3 @@ elif selected == "Image Captioning":
         with col2:
             st.info(caption)
 
-# Fin del script
-