Update app.py

app.py

@@ -20,12 +20,17 @@ def generate_gemini_content(prompt, model_name='gemini-pro-vision', image=None):
 
 # Streamlit app
 def main():
-    st.set_page_config(page_title="Gemini Chatbot", page_icon="🤖")
-    st.title("Gemini Chatbot")
-    st.sidebar.title("Configuración de Gemini")
+    st.set_page_config(page_title="Laura Chatbot - INIF", page_icon="🤖")
 
     # Configurar la API key de Gemini (reemplazar con tu clave de API de Gemini)
-    genai.configure(api_key='TU_CLAVE_API_DE_GEMINI')
+    genai.configure(api_key='AIzaSyA4k6JoFNZsf8L1ixLMMRjEMoBPns5SHZk')
+
+    st.title("Laura Chatbot - INIF")
+    st.sidebar.title("Configuración de Laura Chatbot")
+
+    # Configurar la API key de INIF
+    inif_api_key = 'AIzaSyA4k6JoFNZsf8L1ixLMMRjEMoBPns5SHZk'
+    genai.configure(api_key=inif_api_key)
 
     # Seleccionar el modelo Gemini
     select_model = st.sidebar.selectbox("Selecciona el modelo", ["gemini-pro", "gemini-pro-vision"])
@@ -56,6 +61,22 @@ def main():
     # Entrada del usuario
     user_input = st.text_area("Tú:")
 
+    # Agregar contexto del INIF al input del usuario
+    inif_context = (
+        "I am an informative data analyst chatbot named TERMINATOR, working for the National Institute of Fraud Research and Prevention (INIF), dedicated to fraud prevention and mitigation."
+        " If you have questions related to fraud or prevention, feel free to ask. For inquiries about other topics, I'll redirect you to the fraud prevention context."
+        "\n\nContact Information for INIF:"
+        "\nPhone: +57 317 638 94 71"
+        "\nEmail: atencionalcliente@inif.com.co"
+        "\n\nOur Mission:"
+        "\nTo be the most reliable engine of knowledge, research, and information in Colombia, capable of preventing and combating fraud through synergy between our team and companies."
+        "\n\nOur Vision:"
+        "\nTo lead the construction of a more honest culture, allowing us to progress as a society."
+    )
+
+    # Concatenar el contexto del INIF al input del usuario
+    user_input_with_context = f"{user_input}\n\n{inif_context}"
+
     # Get optional image input if the model selected is 'gemini-pro-vision'
     image_file = None
     if select_model == 'gemini-pro-vision':
@@ -75,7 +96,7 @@ def main():
                     st.warning("Por favor, proporciona una imagen para el modelo gemini-pro-vision.")
                 else:
                     image = Image.open(image_file)
-                    response = generate_gemini_content(user_input, model_name=select_model, image=image)
+                    response = generate_gemini_content(user_input_with_context, model_name=select_model, image=image)
                     if response:
                         if response.candidates:
                             parts = response.candidates[0].content.parts
@@ -86,7 +107,7 @@ def main():
                             st.warning("No se encontraron candidatos en la respuesta.")
             else:
                 # Otros modelos Gemini seleccionados
-                response = get_response(user_input)
+                response = get_response(user_input_with_context)
 
                 # Mostrar respuesta del modelo solo una vez
                 res_text = ""