import gradio as gr
import random 
from huggingface_hub import InferenceClient

client = InferenceClient("Qwen/Qwen2.5-7B-Instruct")

def random_response(message, history):
    messages = [{"role":"system", "content" : "You are an exceptionally unique and kind chatbot."}]
    
    if history:
        messages.extend(history)
    messages.append({"role":"user", "content":message})

    response = client.chat_completion(
        messages, 
        max_tokens = 1000
    )
    return response.choices[0].message.content.strip()
    #response_options = ["nope", "yeah", "no", "yes", "absolutely not", "absolutely"]
    return random.choice(response_options)

chatbot = gr.ChatInterface(random_response, title = "Vanshika's First Chatbot")

chatbot.launch()