Update app.py

app.py

@@ -47,34 +47,6 @@ def download_chat_history(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")
     
@@ -93,7 +65,7 @@ if selected == "System Prompt":
         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 = [url.strip() for url in urls_input.split(",") if url.strip()]  # Guardar las URLs en una lista
@@ -180,4 +152,3 @@ elif selected == "Image Captioning":
             st.info(caption)
 
 # Fin del script
-