Update app.py

app.py

@@ -5,7 +5,7 @@ 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) # Asegúrate de que este archivo existe y contiene las funciones
+from gemini_utility import (load_gemini_pro, gemini_pro_vision_responce)
 from PIL import Image
 
 # Setting the page config
@@ -16,7 +16,35 @@ st.set_page_config(
     initial_sidebar_state="expanded",
 )
 
-# Función para guardar el historial en cookies (sin cambios)
+CONFIG_FILE = "config.json"
+
+# Función para cargar configuraciones desde un archivo JSON
+def load_config():
+    if os.path.exists(CONFIG_FILE):
+        try:
+            with open(CONFIG_FILE, "r") as file:
+                return json.load(file)
+        except Exception as e:
+            st.error(f"Error cargando configuraciones: {e}")
+    return {}
+
+# Función para guardar configuraciones en un archivo JSON
+def save_config(config):
+    try:
+        with open(CONFIG_FILE, "w") as file:
+            json.dump(config, file, indent=4)
+    except Exception as e:
+        st.error(f"Error guardando configuraciones: {e}")
+
+# Cargar configuraciones al inicio
+config = load_config()
+if "system_prompt" not in config:
+    config["system_prompt"] = ""
+
+# Inicializar estados
+if "system_prompt" not in st.session_state:
+    st.session_state.system_prompt = config["system_prompt"]
+
 def save_chat_history(history):
     serializable_history = []
     for message in history:
@@ -26,7 +54,6 @@ def save_chat_history(history):
         })
     st.session_state.cookie_chat_history = json.dumps(serializable_history)
 
-# Función para cargar el historial desde cookies (sin cambios)
 def load_chat_history():
     if 'cookie_chat_history' in st.session_state:
         try:
@@ -43,10 +70,6 @@ def load_chat_history():
             st.error(f"Error cargando el historial: {e}")
     return None
 
-# Inicializar estados (sin cambios)
-if "system_prompt" not in st.session_state:
-    st.session_state.system_prompt = st.session_state.get('cookie_system_prompt', "")
-
 with st.sidebar:
     selected = option_menu(
         "GD AI",
@@ -69,7 +92,6 @@ 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)
@@ -77,7 +99,7 @@ def extract_urls(text):
 
 def fetch_url_content(url):
     try:
-        response = requests.get(url, timeout=10)  # Agregar timeout para evitar bloqueos
+        response = requests.get(url, timeout=10)
         response.raise_for_status()
         return response.text
     except requests.exceptions.RequestException as e:
@@ -86,19 +108,17 @@ def fetch_url_content(url):
 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
+        if content.startswith("Error"):
             new_prompt = new_prompt.replace(url, content)
         else:
             processed_content = process_url_content(content)
@@ -117,7 +137,8 @@ if selected == "System Prompt":
     if st.button("Guardar System Prompt"):
         processed_prompt = process_urls_in_prompt(new_system_prompt)
         st.session_state.system_prompt = processed_prompt
-        st.session_state.cookie_system_prompt = processed_prompt
+        config["system_prompt"] = processed_prompt
+        save_config(config)
         if "chat_session" in st.session_state:
             del st.session_state.chat_session
         st.success("System Prompt actualizado con éxito!")
@@ -126,7 +147,6 @@ if selected == "System Prompt":
         st.markdown("### System Prompt Actual:")
         st.info(st.session_state.system_prompt)
 
-
 elif selected == "Chatbot":
     model = load_gemini_pro()
     if "chat_session" not in st.session_state:
@@ -146,7 +166,7 @@ elif selected == "Chatbot":
         with st.chat_message(translate_role_to_streamlit(message.role)):
             st.markdown(message.parts[0].text)
 
-    user_prompt = st.chat_input("Preguntame algo...")
+    user_prompt = st.chat_input("Pregúntame algo...")
     if user_prompt:
         processed_user_prompt = process_urls_in_prompt(user_prompt)
         st.chat_message("user").markdown(processed_user_prompt)
@@ -155,16 +175,30 @@ elif selected == "Chatbot":
             st.markdown(gemini_response.text)
         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)
-        with col1:
-            st.image(image, caption="Uploaded Image", use_column_width=True)
-        default_prompt = "Write a caption for this image"
-        caption = gemini_pro_vision_responce(default_prompt, image)
-        with col2:
-            st.info(caption)
+    upload_image = st.file_uploader("Sube una imagen...", type=["jpg", "jpeg", "png"])
+    
+    if upload_image and st.button("Generar"):
+        try:
+            # Cargar y mostrar la imagen
+            image = Image.open(upload_image)
+            col1, col2 = st.columns(2)
+            with col1:
+                st.image(image, caption="Imagen subida", use_column_width=True)
+
+            # Generar un subtítulo usando una función de visión AI
+            default_prompt = "Escribe un subtítulo para esta imagen"
+            caption = gemini_pro_vision_responce(default_prompt, image)
+            
+                        # Mostrar el resultado
+            with col2:
+                st.info(caption)
+        except Exception as e:
+            st.error(f"Error procesando la imagen: {e}")
+
+# Mensaje por defecto si ninguna sección está seleccionada
+else:
+    st.write("Selecciona una opción en el menú para comenzar.")
+
+