import gradio as gr
from openai import OpenAI

# Konfigurasi OpenRouter
client = OpenAI(
    base_url="https://openrouter.ai/api/v1",
    api_key="your-openrouter-api-key",  # Ganti dengan API key OpenRouter Anda
)

def respond(message, history):
    # Kirim pesan ke model DeepSeek R1 Zero
    completion = client.chat.completions.create(
        extra_headers={
            "HTTP-Referer": "https://huggingface.co/spaces",  # Optional
            "X-Title": "DeepSeek Chatbot",  # Optional
        },
        model="deepseek/deepseek-r1-zero:free",
        messages=[
            {"role": "system", "content": "Anda adalah asisten AI yang membantu. Berikan jawaban yang singkat dan jelas."},
            *[{"role": "user" if i % 2 == 0 else "assistant", "content": h} for i, h in enumerate(history)],
            {"role": "user", "content": message}
        ]
    )
    
    return completion.choices[0].message.content

# Buat antarmuka chatbot
demo = gr.ChatInterface(
    respond,
    title="DeepSeek R1 Zero Chatbot",
    description="Chatbot menggunakan model DeepSeek R1 Zero melalui OpenRouter",
    theme="soft",
    examples=["Apa itu AI?", "Buatkan puisi pendek", "Jelaskan teori relativitas dengan sederhana"]
)

demo.launch()