Update app.py

app.py

@@ -40,6 +40,13 @@ def load_chat_history():
             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', "")
@@ -109,7 +116,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)
@@ -120,6 +127,16 @@ elif selected == "Chatbot":
         # Guardar historial actualizado
         save_chat_history(st.session_state.chat_session.history)
 
+    # Opción para descargar el historial del chat
+    if st.button("Descargar Historial del Chat"):
+        chat_history = download_chat_history(st.session_state.chat_session.history)
+        st.download_button(
+            label="Descargar",
+            data=chat_history,
+            file_name="historial_chat.txt",
+            mime="text/plain"
+        )
+
 elif selected == "Image Captioning":
     st.title("Image Caption Generation📸")
     upload_image = st.file_uploader("Upload an image...", type=["jpg", "jpeg", "png"])
@@ -132,4 +149,6 @@ elif selected == "Image Captioning":
         default_prompt = "Write a caption for this image"
         caption = gemini_pro_vision_responce(default_prompt, image)
         with col2:
-            st.info(caption)
+            st.info(caption)
+
+# Fin del script