import streamlit as st
from PIL import Image
from groq import Groq
import os


#st.image('calamo.png', caption="", use_column_width=False)

client = Groq(
    api_key=os.environ.get("GROQ_API_KEY"),
)
# Rest of your Streamlit app
st.write("# Inicio")

# Other content of your app
#st.title("plAIn")
# Add more components here
# Create a text input widget
user_input = st.text_input('Pega tu texto:', '')

# Define a function to process the input
def process_text(input_text):
    prompt = '''
    Eres un experto en lenguaje claro. Las pautas básicas para lenguaje claro son:
    - Expresar una idea por oración.
    - Utilizar oraciones de treinta palabras o menos.
    - Evitar la jerga.
    - Seguir el orden sujeto, verbo y predicado.
    - Utilizar una estructura lógica, organizando la información de manera clara y coherente.
    Evalúa la calidad del lenguaje de este texto y sugiere las correcciones oportunas:"
    '''
    input = prompt + input_text

    chat_completion = client.chat.completions.create(
    messages=[
        {
            "role": "user",
            "content": input,
        }
    ],
    model="mixtral-8x7b-32768",
)
    return (chat_completion.choices[0].message.content)

# Call the function with the user input
processed_output = process_text(user_input)

# Display the processed output
st.write('Texto procesado:', processed_output)