import gradio as gr
from transformers import AutoModelForSeq2SeqLM, AutoTokenizer

# Load model from Hugging Face Model Hub
model_path = "asritha22bce/bart-positive-tone"  # Ensure this model exists and is public
model = AutoModelForSeq2SeqLM.from_pretrained(model_path)
tokenizer = AutoTokenizer.from_pretrained(model_path)

def generate_positive_headline(text):
    inputs = tokenizer(text, return_tensors="pt", truncation=True, padding=True)
    outputs = model.generate(**inputs, max_length=50)
    return tokenizer.decode(outputs[0], skip_special_tokens=True)

interface = gr.Interface(
    fn=generate_positive_headline,
    inputs=gr.Textbox(label="Enter a Headline"),
    outputs=gr.Textbox(label="Positive Headline"),
    title="Positive Headline Generator",
    description="This app converts headlines into a more positive tone using a fine-tuned BART model.",
    allow_flagging="never"  # 🚀 Makes API accessible without UI issues
)

interface.launch(share=True)