import os
import streamlit as st
import google.generativeai as genai

# TERMINATOR

error_flag = False  # Global variable to track error display

def generate_response(cleaned_input, model):
    global error_flag  # Use the global error_flag variable

    try:
        # Generate response using the model
        response = model.generate_content(cleaned_input, stream=True)

        # Ensure response iteration is complete before accessing accumulated attributes
        response.resolve()

        # Display the generated response
        full_response = ""
        for part in response.parts:  # Iterate through parts in multipart response
            full_response += part.text  # Use part.text to access the text of each part

        return full_response

    except Exception as e:
        error_message = str(e)
        if "text must be a valid text with a maximum of 5000 characters" in error_message and not error_flag:
            error_response = ("The question you are asking may go against Google GEMINI policies: WiseOracle"
                              "Please reformulate your question without forbidden topics or ask something else. "
                              "For more information, see: https://policies.google.com/terms/generative-ai/use-policy "
                             )
            st.error(error_response)
            error_flag = True  # Set the error_flag to True after displaying the error message
            return error_response
        else:
            error_response = f"Error: {error_message}\nSorry, I am an artificial intelligence that is still in development and is in alpha phase. At the moment, I cannot answer your question properly, but in the future, I will be able to do so."
            st.error(error_response)
            return error_response

# Aplicación Streamlit
def principal():
    st.title("Formateador de Entrevistas")

    genai.configure(api_key='AIzaSyA4k6JoFNZsf8L1ixLMMRjEMoBPns5SHZk')  # Replace with your Google GEMINI API key

    # Choose the Gemini model
    model = genai.GenerativeModel('gemini-pro')

    # Cargador de archivos para el archivo .txt
    archivo_cargado = st.file_uploader("Cargar un archivo .txt", type=["txt"])

    if archivo_cargado is not None:
        # Lee el contenido del archivo cargado
        contenido = archivo_cargado.read()

        # Botón para iniciar el formateo
        if st.button("Iniciar Formateo"):
            st.info("Formateo en progreso...")

            # Divide el contenido en chunks de 1500 palabras
            tamano_chunk = 1500
            chunks = [contenido[i:i + tamano_chunk] for i in range(0, len(contenido), tamano_chunk)]

            # Inicializa la barra de progreso
            barra_progreso_chunks = st.progress(0)

            # Procesa cada chunk y muestra la salida formateada
            salida_formateada = ""
            for i, chunk in enumerate(chunks):
                respuesta_bot = generate_response(
                    f"  Formatea el texto de la entrevista como un segmento, colocando etiquetas *Entrevistador:* y *Entrevistado:* antes de cada segmento identificado. Mantén el orden original y agrega un salto de línea entre etiquetas y segmentos. Corrige palabras inentendibles según el contexto global. Procede sistemáticamente para lograr coherencia en los resultados al procesar cada chunk.: {chunk}",
                    model
                )

                # Agrega la respuesta formateada a la salida
                salida_formateada += respuesta_bot + "\n"

                # Actualiza la barra de progreso
                barra_progreso_chunks.progress((i + 1) / len(chunks))

            # Muestra la salida formateada
            st.success("Formateo completado")
            st.text_area("Segmento de Entrevista Formateado:", salida_formateada)

            # Botón para descargar la respuesta formateada
            download_button = st.download_button(
                label="Descargar Respuesta",
                data=salida_formateada,
                file_name="respuesta_formateada.txt",
                key="download_button"
            )

if __name__ == "__main__":
    principal()