import os

from groq import Groq

import gradio as gr

client = Groq(
    api_key=os.environ.get("GROQ_API_KEY"),
)

def eval_text (text):
  prompt = "Eres un experto en lenguaje claro. Evalúa la calidad del lenguaje de este texto:"
  input = prompt + text

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

  return (chat_completion.choices[0].message.content)

custom_image_url = "https://www.traductores.org.ar/wp-content/uploads/2021/09/CalamoCran_logo-web-.jpg"

custom_image = gr.Image(custom_image_url)

demo = gr.Interface(fn=eval_text, inputs="text", outputs="text", title="plAIn", image=custom_image)

demo.launch()