import gradio as gr
from transformers import pipeline

# Load the fine-tuned model
generator = pipeline("text-generation", model="shibly100/gpt2_corporate_leadership")

# Function to generate responses
def chat_with_ai(prompt, history=[]):
    response = generator(prompt, max_length=200, do_sample=True, top_p=0.95, top_k=50, temperature=0.7)
    return response[0]["generated_text"]

# Fast and simple chat interface
with gr.Blocks() as demo:
    gr.Markdown("# SPARKY Corporate Leadership AI")
    gr.Markdown("Ask anything about corporate leadership and get AI-powered responses!")

    chatbot = gr.ChatInterface(chat_with_ai)

demo.launch()