Update app.py

app.py

@@ -14,6 +14,25 @@ st.set_page_config(
     initial_sidebar_state="expanded",
 )
 
+# Menú de opciones
+selected = option_menu(
+    menu_title="Menú",
+    options=["System Prompt", "Chatbot", "Image Captioning"],
+    icons=["gear", "chat", "camera"],
+    default_index=0,
+    orientation="horizontal"
+)
+
+# Inicializar el estado de la sesión
+if 'cookie_chat_history' not in st.session_state:
+    st.session_state.cookie_chat_history = json.dumps([])
+
+if 'cookie_urls' not in st.session_state:
+    st.session_state.cookie_urls = []
+
+if 'system_prompt' not in st.session_state:
+    st.session_state.system_prompt = ""
+
 # Función para guardar el historial en cookies
 def save_chat_history(history):
     serializable_history = []
@@ -78,7 +97,6 @@ if selected == "System Prompt":
     
     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 = [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
@@ -95,7 +113,7 @@ if selected == "System Prompt":
 elif selected == "Chatbot":
     model = load_gemini_pro()
     
-    # Inicializar o cargar sesión de chat
+        # Inicializar o cargar sesión de chat
     if "chat_session" not in st.session_state:
         loaded_chat = load_chat_history()
         if loaded_chat:
@@ -116,7 +134,7 @@ elif selected == "Chatbot":
         with st.chat_message(translate_role_to_streamlit(message.role)):
             st.markdown(message.parts[0].text)
 
-        # Campo de entrada
+    # Campo de entrada
     user_prompt = st.chat_input("Preguntame algo...")
     if user_prompt:
         st.chat_message("user").markdown(user_prompt)
@@ -134,7 +152,6 @@ elif selected == "Chatbot":
 
             # Aquí puedes procesar el contenido obtenido de las URLs
             # Por ejemplo, podrías resumirlo o extraer información relevante
-            # Para este ejemplo, simplemente unimos el contenido
             combined_content = "\n\n".join(fetched_contents)
             user_prompt += f"\n\nInformación adicional de URLs:\n{combined_content}"
 
@@ -172,6 +189,3 @@ elif selected == "Image Captioning":
 
 # Fin del script
 
-       
-
-