import gradio as gr
from groq import Groq

client = Groq(api_key="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")

def chat(message, history):
    messages = []
    for user_msg, assistant_msg in history:
        messages.append({"role": "user", "content": user_msg})
        messages.append({"role": "assistant", "content": assistant_msg})
    messages.append({"role": "user", "content": message})
    
    completion = client.chat.completions.create(model="llama-3.3-70b-versatile", messages=messages)
    return completion.choices[0].message.content

gr.ChatInterface(fn=chat).launch()