Update app.py

app.py

@@ -1,9 +1,11 @@
 import streamlit as st
 from deep_translator import GoogleTranslator
-import google.generativeai as palm
+import google.generativeai as genai
 import re
+import textwrap
+from IPython.display import display, Markdown
 
-# L.A.U.R.A: Lingüista Artificial Unificada de Respuestas Automaticas
+# TERMINATOR
 
 error_flag = False  # Global variable to track error display
 
@@ -22,15 +24,15 @@ def generate_response(cleaned_input, model):
 
     try:
         # Generate response using the model
-        completion = palm.generate_text(
-            model=model,
-            prompt=cleaned_input,
-            temperature=0,
-            max_output_tokens=1000,
-        )
+        response = model.generate_content(cleaned_input, stream=True)
+
+        # Display the generated response
+        full_response = ""
+        for chunk in response:
+            full_response += chunk.text
 
         # Translate the response to Spanish without modifying it
-        translated_output = translate_text(completion.result, target_language='es')
+        translated_output = translate_text(full_response, target_language='es')
 
         return translated_output
 
@@ -50,17 +52,11 @@ def generate_response(cleaned_input, model):
             return error_response
 
 def main():
-    st.image("https://i.ibb.co/2j2gGW1/logo-inif.png", use_column_width=True)
-    st.title("Chatbot INIF - L.A.U.R.A.")
-    palm.configure(api_key='AIzaSyCezVerubEzQc9JHz3V8hofpAlSIJXGxFQ')
-
-    # Select the model that supports text generation
-    models = [m for m in palm.list_models() if 'generateText' in m.supported_generation_methods]
-    if not models:
-        st.error("No hay modelos disponibles para la generación de texto. Verifica tu configuración o inténtalo más tarde.")
-        st.stop()
+    st.title("Chatbot INIF - TERMINATOR")
+    genai.configure(api_key='AIzaSyCezVerubEzQc9JHz3V8hofpAlSIJXGxFQ')  # Replace with your Gemini API key
 
-    model = models[0].name
+    # Choose the Gemini model
+    model = genai.GenerativeModel('gemini-pro')
 
     st.write("Bienvenido al Chatbot informativo del Instituto Nacional de Investigación y Prevención del Fraude (INIF).")
 
@@ -83,13 +79,12 @@ def main():
         additional_info = (
             "Trabajamos para empresarios y colaboradores de cualquier industria que buscan combatir el fraude que impacta negativamente la reputación y la economía de sus organizaciones y del país.\n"
             "Somos una empresa privada que trabaja con la industria bancaria y aseguradora."
-            "no somos una empresa publica ni guibernamental, somos una empresa privada"
         )
 
         # Add the command to act as an INIF informative chatbot
         bot_command = (
-            "I am an informative data analyst chatbot named L.A.U.R.A, 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. when you refer to INIF, refer to it as INIF, not mention THE INIF. treat INIF as a name, not as a Institution. For inquiries about other topics, I'll redirect you to the fraud prevention 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."
             f"\n\n{additional_info}"
             "\n\nContact Information for INIF:"
             "\nPhone: +57 317 638 94 71"