Update app.py

app.py

@@ -8,22 +8,7 @@ from streamlit_option_menu import option_menu
 from gemini_utility import (load_gemini_pro, gemini_pro_vision_responce)
 from PIL import Image
 
-# Ruta para guardar el System Prompt
-SYSTEM_PROMPT_FILE = "system_prompt.json"
-
-# Función para cargar el System Prompt desde el archivo
-def load_system_prompt():
-    if os.path.exists(SYSTEM_PROMPT_FILE):
-        with open(SYSTEM_PROMPT_FILE, 'r') as f:
-            return json.load(f).get("system_prompt", "")
-    return ""
-
-# Función para guardar el System Prompt en el archivo
-def save_system_prompt(system_prompt):
-    with open(SYSTEM_PROMPT_FILE, 'w') as f:
-        json.dump({"system_prompt": system_prompt}, f)
-
-# Setting the page config
+# Configuración de la página
 st.set_page_config(
     page_title="GnosticDev AI",
     page_icon="🤖",
@@ -31,10 +16,6 @@ st.set_page_config(
     initial_sidebar_state="expanded",
 )
 
-# Cargar el System Prompt guardado
-if "system_prompt" not in st.session_state:
-    st.session_state.system_prompt = load_system_prompt()
-
 # Función para guardar el historial en cookies
 def save_chat_history(history):
     serializable_history = []
@@ -62,6 +43,10 @@ def load_chat_history():
             st.error(f"Error cargando el historial: {e}")
     return None
 
+# Inicializar estados
+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",
@@ -79,10 +64,7 @@ with st.sidebar:
         st.success("Historial borrado!")
 
 def translate_role_to_streamlit(user_role):
-    if user_role == "model":
-        return "assistant"
-    else:
-        return user_role
+    return "assistant" if user_role == "model" else user_role
 
 def extract_urls(text):
     url_pattern = r"(https?://\S+)"
@@ -91,7 +73,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:
@@ -100,7 +82,6 @@ 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:
@@ -111,26 +92,17 @@ def process_urls_in_prompt(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)
             new_prompt = new_prompt.replace(url, processed_content)
     return new_prompt
 
-if selected == "System Prompt":
-    st.title("Configuración del System Prompt")
-    new_system_prompt = st.text_area(
-        "Ingresa las instrucciones para el AI (System Prompt), incluyendo URLs",
-        value=st.session_state.system_prompt,
-        height=300,
-        help="Escribe aquí las instrucciones que definirán el comportamiento del AI. Puedes incluir URLs."
-    )
-
     if st.button("Guardar System Prompt"):
         processed_prompt = process_urls_in_prompt(new_system_prompt)
         st.session_state.system_prompt = processed_prompt
-        save_system_prompt(processed_prompt)  # Guardar en archivo JSON
+        st.session_state.cookie_system_prompt = processed_prompt
         if "chat_session" in st.session_state:
             del st.session_state.chat_session
         st.success("System Prompt actualizado con éxito!")
@@ -147,43 +119,46 @@ elif selected == "Chatbot":
             st.session_state.chat_session = loaded_chat
         else:
             st.session_state.chat_session = model.start_chat(history=[])
-            # No enviar automáticamente el system_prompt aquí
+            if st.session_state.system_prompt:
+                st.session_state.chat_session.send_message(st.session_state.system_prompt)
 
     st.title("Gnosticdev Chatbot")
-    user_prompt = st.chat_input("Preguntame algo...")
+    if st.session_state.system_prompt:
+        with st.expander("Ver System Prompt actual"):
+            st.info(st.session_state.system_prompt)
+    
+    for message in st.session_state.chat_session.history:
+        with st.chat_message(translate_role_to_streamlit(message.role)):
+            st.markdown(message.parts[0].text)
+
+    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)
-        gemini_response = st.session_state.chat_session.send_message(processed_user_prompt)
+        
+        # Combina el System Prompt con la entrada del usuario
+        full_prompt = f"{st.session_state.system_prompt}\nUser: {processed_user_prompt}\nAI:"
+        gemini_response = st.session_state.chat_session.send_message(full_prompt)
+        
         with st.chat_message("assistant"):
             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("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.")
-
-
+    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)
+        
+        # Generar un subtítulo para la imagen
+        default_prompt = "Write a caption for this image"
+        caption = gemini_pro_vision_responce(default_prompt, image)
+        
+        with col2:
+            st.info(caption)
+
+# Fin del archivo