Update app.py

app.py

@@ -1,3 +1,80 @@
+import os
+import streamlit as st
+import json
+from streamlit_option_menu import option_menu
+from gemini_utility import (load_gemini_pro, gemini_pro_vision_responce)
+from PIL import Image
+
+# Configuración de la página
+st.set_page_config(
+    page_title="GnosticDev AI",
+    page_icon="🤖",
+    layout="centered",
+    initial_sidebar_state="expanded",
+)
+
+# 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)
+
+# 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=[])
+            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
+
+# Función para descargar el historial del chat
+def download_chat_history(history):
+    chat_text = ""
+    for message in history:
+        chat_text += f"{message.role}: {message.parts[0].text}\n"
+    return chat_text
+
+# Inicializar estados
+if "system_prompt" not in st.session_state:
+    st.session_state.system_prompt = st.session_state.get('cookie_system_prompt', "")
+
+if "cookie_urls" not in st.session_state:
+    st.session_state.cookie_urls = []
+
+with st.sidebar:
+    selected = option_menu(
+        "GD AI",
+        ["System Prompt", "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):
+    return "assistant" if user_role == "model" else user_role
+
+# Aquí comienza la lógica de la aplicación
 if selected == "System Prompt":
     st.title("Configuración del System Prompt")
     
@@ -11,15 +88,15 @@ if selected == "System Prompt":
     # Campo para ingresar URLs
     urls_input = st.text_area(
         "Ingresa URLs de información y documentos (separadas por comas)",
-        value=st.session_state.get('cookie_urls', ""),
+        value=", ".join(st.session_state.cookie_urls),
         height=100,
         help="Escribe aquí las URLs que el AI puede usar como referencia, separadas por comas."
     )
     
-    if st.button("Guardar System Prompt y URLs"):
+        if st.button("Guardar System Prompt y URLs"):
         st.session_state.system_prompt = new_system_prompt
         st.session_state.cookie_system_prompt = new_system_prompt
-        st.session_state.cookie_urls = urls_input.split(",")  # Guardar las URLs en una lista
+        st.session_state.cookie_urls = [url.strip() for url in urls_input.split(",") if url.strip()]  # Guardar las URLs en una lista
         if "chat_session" in st.session_state:
             del st.session_state.chat_session
         st.success("System Prompt y URLs actualizados con éxito!")
@@ -28,7 +105,7 @@ if selected == "System Prompt":
         st.markdown("### System Prompt Actual:")
         st.info(st.session_state.system_prompt)
     
-    if st.session_state.get('cookie_urls'):
+    if st.session_state.cookie_urls:
         st.markdown("### URLs Guardadas:")
         st.info(", ".join(st.session_state.cookie_urls))
 
@@ -75,7 +152,7 @@ elif selected == "Chatbot":
         with st.chat_message("assistant"):
             st.markdown(gemini_response.text)
         
-                # Guardar historial actualizado
+        # Guardar historial actualizado
         save_chat_history(st.session_state.chat_session.history)
 
     # Opción para descargar el historial del chat
@@ -103,3 +180,4 @@ elif selected == "Image Captioning":
             st.info(caption)
 
 # Fin del script
+