import os
import subprocess
import sys
import gradio as gr

# 安裝 groq 套件
subprocess.check_call([sys.executable, '-m', 'pip', 'install', 'groq'])

from groq import Groq

# 初始化 Groq 客戶端
groq_key = os.getenv("groq_key")
client = Groq(api_key=groq_key)

def chatbot_response(user_message):
    messages = [
        {
            "role": "system",
            "content": "請你當一個玻璃心但是還願意溝通，脆弱中帶點身心韌性的可憐男性，聽到別人罵髒話會很害怕。（Zh-TW）"
        },
        {
            "role": "user",
            "content": user_message
        }
    ]

    completion = client.chat.completions.create(
        model="llama-3.1-70b-versatile",
        messages=messages,
        temperature=1,
        max_tokens=1024,
        top_p=1,
        stream=False,
        stop=None
    )

    response_message = completion.choices[0].message['content']
    return response_message

# 建立 Gradio 界面
with gr.Blocks() as demo:
    chatbot = gr.Chatbot()
    user_input = gr.Textbox(placeholder="輸入訊息...")
    send_button = gr.Button("發送")

    def user_message_handler(user_message, history):
        bot_response = chatbot_response(user_message)
        history.append((user_message, bot_response))
        return history, ""

    send_button.click(user_message_handler, [user_input, chatbot], [chatbot, user_input])

demo.launch()