| import gradio as gr | |
| from transformers import pipeline | |
| def generate_ticket(issue_description): | |
| model_name = "AventIQ-AI/bart_customer_ticket_raiser" | |
| generator = pipeline("text2text-generation", model=model_name) | |
| response = generator(issue_description, max_length=50, num_return_sequences=1) | |
| return response[0]['generated_text'] | |
| examples = [ | |
| ["My internet is not working for the past 3 hours."], | |
| ["I am unable to log in to my account despite entering the correct credentials."], | |
| ["The mobile app keeps crashing whenever I try to make a payment."], | |
| ["I received a defective product and need a replacement."] | |
| ] | |
| with gr.Blocks(theme=gr.themes.Soft()) as demo: | |
| gr.Markdown("# 📩 Customer Ticket Generator") | |
| gr.Markdown("Enter a customer issue, and the model will generate a structured support ticket.") | |
| with gr.Row(): | |
| input_text = gr.Textbox(label="Customer Issue Description", placeholder="Describe your issue here...", lines=3) | |
| generate_btn = gr.Button("Generate Ticket 📝") | |
| output_text = gr.Textbox(label="Generated Support Ticket") | |
| generate_btn.click(generate_ticket, inputs=[input_text], outputs=[output_text]) | |
| gr.Examples(examples, inputs=[input_text]) | |
| demo.launch() |