import random
import gradio as gr
from huggingface_hub import InferenceClient

client = InferenceClient("microsoft/phi-4")

def respond(message,history):
    messages = [{"role": "system", "content": "You are a friendly chatbot."}]
    if history:
        messages.extend(history)
    messages.append({"role": "user", "content": message})
    response = client.chat_completion(messages, max_tokens = 100)
    return response['choices'][0]['message']['content'].strip()

def magic_ball(message, history):
    return random.choice(["Yes, definitely.",
       "As I see it, yes.",
       "Reply hazy, try again.",
       "Cannot predict now.",
       "Do not count on it.",
       "My sources say no.",
       "Outlook not so good.",
       "Very doubtful."])

print("Hello, world!")

chatbot = gr.ChatInterface(respond, type="messages", description="Ask me any question!", title="Magic 8 Ball")

chatbot.launch()