Update app.py

app.py

@@ -1,6 +1,7 @@
 import streamlit as st
 import google.generativeai as genai
 import os
+import time
 from dotenv import load_dotenv
 from styles import get_custom_css
 from formulas import offer_formulas
@@ -24,6 +25,10 @@ if 'submitted' not in st.session_state:
     st.session_state.submitted = False
 if 'offer_result' not in st.session_state:
     st.session_state.offer_result = ""
+if 'success_message' not in st.session_state:
+    st.session_state.success_message = None
+if 'success_time' not in st.session_state:
+    st.session_state.success_time = None
 
 # Hide Streamlit menu and footer
 st.markdown("""
@@ -61,6 +66,19 @@ with col1:
         # Add file/image uploader here
         uploaded_file = st.file_uploader("📄 Sube un archivo o imagen", 
                                         type=['txt', 'pdf', 'docx', 'jpg', 'jpeg', 'png'])
+        
+        # Display success message if it exists and hasn't timed out
+        if st.session_state.success_message:
+            current_time = time.time()
+            # Show message for 3 seconds
+            if current_time - st.session_state.success_time < 3:
+                st.success(st.session_state.success_message)
+            else:
+                # Clear the message after timeout
+                st.session_state.success_message = None
+                st.session_state.success_time = None
+                st.rerun()
+        
         if uploaded_file is not None:
             file_type = uploaded_file.name.split('.')[-1].lower()
             
@@ -68,7 +86,9 @@ with col1:
             if file_type in ['txt', 'pdf', 'docx']:
                 if file_type == 'txt':
                     file_content = uploaded_file.read().decode('utf-8')
-                    st.success(f"Archivo de texto cargado correctamente: {uploaded_file.name}")
+                    # Set success message and time instead of showing directly
+                    st.session_state.success_message = f"Archivo de texto cargado correctamente: {uploaded_file.name}"
+                    st.session_state.success_time = time.time()
                     
                 elif file_type == 'pdf':
                     try:
@@ -77,7 +97,9 @@ with col1:
                         file_content = ""
                         for page in pdf_reader.pages:
                             file_content += page.extract_text() + "\n"
-                        st.success(f"Archivo PDF cargado correctamente: {uploaded_file.name}")
+                        # Set success message and time instead of showing directly
+                        st.session_state.success_message = f"Archivo PDF cargado correctamente: {uploaded_file.name}"
+                        st.session_state.success_time = time.time()
                     except Exception as e:
                         st.error(f"Error al leer el archivo PDF: {str(e)}")
                         file_content = ""
@@ -87,7 +109,9 @@ with col1:
                         import docx
                         doc = docx.Document(uploaded_file)
                         file_content = "\n".join([para.text for para in doc.paragraphs])
-                        st.success(f"Archivo DOCX cargado correctamente: {uploaded_file.name}")
+                        # Set success message and time instead of showing directly
+                        st.session_state.success_message = f"Archivo DOCX cargado correctamente: {uploaded_file.name}"
+                        st.session_state.success_time = time.time()
                     except Exception as e:
                         st.error(f"Error al leer el archivo DOCX: {str(e)}")
                         file_content = ""
@@ -111,7 +135,9 @@ with col1:
                         "data": image_bytes
                     }
                 ]
-                st.success(f"Imagen cargada correctamente: {uploaded_file.name}")
+                # Set success message and time instead of showing directly
+                st.session_state.success_message = f"Imagen cargada correctamente: {uploaded_file.name}"
+                st.session_state.success_time = time.time()
                 
                 # Set file type flag
                 is_image = True