app.py

import gradio as gr
from utils import T5Generator


# Preprocessing function that modifies the user input.
def preprocess_input(user_input):
    bos_instruction_id = """Definition: The output will be the aspects (both implicit and explicit) and the aspect's sentiment polarity. In cases where there are no aspects the output should be noaspectterm:none.
            Examples:
            1.
            
            input: The hotel was beautiful, with friendly staff and a delicious breakfast buffet. However, the Wi-Fi was slow, the bathroom wasn't clean, and the room had a weird smell.
            output: hotel\:positive, staff\:positive, breakfast buffet\:positive, Wi-Fi\:negative, bathroom\:negative, room smell\:negative
            
            2.
            
            input: This coat looks stylish and fits perfectly. The fabric feels high-quality, and the color is vibrant. But the buttons fall off easily, and it's not warm enough for winter.
            output: style\:positive, fit\:positive, fabric\:positive, color\:positive, buttons\:negative, warmth\:negative
            
            3.
            
            input: The online course was affordable and the instructor explained topics clearly. However, the video quality was poor, the quizzes were too simple, and the pacing was inconsistent.
            output: cost\:positive, instructor clarity\:positive, video quality\:negative, quizzes\:negative, pacing\:negative
            
            4.
            
            input: My experience with this bank has been mostly good. The mobile app is user-friendly, staff are polite, and transactions are smooth. But the customer support takes too long to respond, and hidden fees are frustrating.
            output: mobile app\:positive, staff\:positive, transactions\:positive, customer support\:negative, hidden fees\:negative
            
            5.
            
            input: I enjoyed the plot and the character development in this novel. The writing style was smooth, and the dialogue felt real. But the ending was disappointing and the pacing was uneven in the middle chapters.
            output: plot\:positive, character development\:positive, writing style\:positive, dialogue\:positive, ending\:negative, pacing\:negative
            
            6.
            
            input: The doctor was kind and explained everything well. The clinic was clean, and the staff was professional. But the waiting time was too long, and I felt rushed during the consultation.
            output: doctor\:positive, clinic cleanliness\:positive, staff professionalism\:positive, waiting time\:negative, consultation time\:negative
            
            7.
            
            input: The sound quality of these headphones is amazing, and the battery lasts long. The design is sleek, and the noise cancellation works well. But they’re uncomfortable after long use and overpriced.
            output: sound quality\:positive, battery life\:positive, design\:positive, noise cancellation\:positive, comfort\:negative, price\:negative
            
            8.
            
            input: This vacuum cleaner has strong suction power and is easy to maneuver. It’s lightweight and the attachments are useful. However, it's very loud and the dustbin fills up quickly.
            output: suction power\:positive, maneuverability\:positive, weight\:positive, attachments\:positive, noise\:negative, dustbin size\:negative
            
            9.
            
            input: The recipe was easy to follow, and the dish turned out flavorful. The ingredients were accessible, and the cooking time was just right. But the portion size was small and cleanup took a while.
            output: instructions\:positive, taste\:positive, ingredients\:positive, cooking time\:positive, portion size\:negative, cleanup\:negative
            
            10.
            
            input: This gym is well-equipped and spacious. The trainers are knowledgeable and the membership is reasonably priced. However, the showers are often dirty and there’s limited parking.
            output: equipment\:positive, space\:positive, trainers\:positive, price\:positive, showers\:negative, parking\:negative
            
            11.
            
            input: The flight was on time, and the in-flight entertainment was great. Seats were clean and staff were courteous. But the legroom was terrible and the food was tasteless.
            output: punctuality\:positive, entertainment\:positive, cleanliness\:positive, staff\:positive, legroom\:negative, food\:negative
            
            12.
            
            input: The mobile phone's camera is sharp, the screen is vibrant, and the battery life is impressive. However, the device heats up quickly and the UI is cluttered.
            output: camera\:positive, screen\:positive, battery\:positive, heating\:negative, UI\:negative
            
            13.
            
            input: The tour guide was informative, and the destinations were breathtaking. The transportation was comfortable and well organized. But meals were mediocre and the schedule was too packed.
            output: guide\:positive, destinations\:positive, transportation\:positive, meals\:negative, schedule\:negative
            
            14.
            
            input: The jeans fit perfectly, are very comfortable, and the material feels premium. However, the dye runs in the wash and delivery took too long.
            output: fit\:positive, comfort\:positive, material\:positive, dye quality\:negative, delivery\:negative
            
            15.
            
            input: This smartwatch tracks fitness accurately, has a responsive touch screen, and looks stylish. But syncing with the phone is inconsistent and charging takes too long.
            output: fitness tracking\:positive, touchscreen\:positive, design\:positive, syncing\:negative, charging\:negative
            
            16.
            
            input: The therapist was compassionate and helpful. Sessions were structured and impactful. However, the cost was high, and appointment availability was limited.
            output: therapist\:positive, session structure\:positive, impact\:positive, cost\:negative, appointment availability\:negative
            
            17.
            
            input: The gaming console runs smoothly and the graphics are top-notch. The game library is vast, and loading times are short. But the controllers feel cheap and it's very bulky.
            output: performance\:positive, graphics\:positive, game library\:positive, loading time\:positive, controller quality\:negative, size\:negative
            
            18.
            
            input: This online store has a great variety of products and fast delivery. Prices are competitive and the website is easy to navigate. But the return process is tedious and customer support is unhelpful.
            output: product variety\:positive, delivery\:positive, pricing\:positive, website UX\:positive, return process\:negative, customer support\:negative
            
            19.
            
            input: The air conditioner cools quickly, has a quiet mode, and looks modern. But the installation service was poor and the remote is hard to use.
            output: cooling\:positive, noise\:positive, design\:positive, installation\:negative, remote\:negative
            
            20.
            
            input: The blender is powerful, easy to clean, and comes with useful attachments. But it's very noisy and the lid is hard to remove.
            output: power\:positive, cleaning\:positive, attachments\:positive, noise\:negative, lid\:negative
            
            21.
            
            input: The drama had amazing cinematography and a compelling storyline. The acting was strong and the soundtrack was fitting. But the pacing was off and the ending felt rushed.
            output: cinematography\:positive, storyline\:positive, acting\:positive, soundtrack\:positive, pacing\:negative, ending\:negative
            
            22.
            
            input: The mattress is very comfortable and supports my back well. It’s breathable and has no chemical smell. However, delivery was delayed and the warranty terms are confusing.
            output: comfort\:positive, support\:positive, breathability\:positive, smell\:positive, delivery\:negative, warranty\:negative
            
            23.
            
            input: The mobile app is intuitive and has many features. Login is quick and data syncs well. However, it crashes often and has annoying ads.
            output: usability\:positive, features\:positive, login\:positive, sync\:positive, stability\:negative, ads\:negative
            
            24.
            
            input: The watch looks elegant, keeps time perfectly, and feels solid. But the strap is uncomfortable and it scratches too easily.
            output: design\:positive, accuracy\:positive, build\:positive, strap comfort\:negative, durability\:negative
            
            25.
            
            input: The refrigerator is spacious and energy-efficient. It’s quiet and cools evenly. But the freezer section is small and the shelves feel flimsy.
            output: capacity\:positive, efficiency\:positive, noise\:positive, cooling\:positive, freezer\:negative, shelf quality\:negative
            
            26.
            
            input: The coffee maker brews quickly and the coffee is hot. The design is sleek, and it's easy to clean. However, it leaks when brewing and is louder than expected.
            output: brewing time\:positive, coffee temperature\:positive, design\:positive, cleaning\:positive, leakage\:negative, noise\:negative
            
            27.
            
            input: The sports shoes are comfortable, durable, and stylish. The grip is excellent, and they are lightweight. But the stitching came undone after a few uses, and the price is too high.
            output: comfort\:positive, durability\:positive, style\:positive, grip\:positive, stitching\:negative, price\:negative
            
            28.
            
            input: The smartwatch tracks steps accurately and has a great battery life. It's sleek and easy to use. But the heart rate monitor is inaccurate and the band is uncomfortable.
            output: step tracking\:positive, battery life\:positive, design\:positive, ease of use\:positive, heart rate monitor\:negative, band comfort\:negative
            
            29.
            
            input: The pizza crust was crispy, and the toppings were fresh. The sauce was flavorful, and the cheese was perfectly melted. However, it was too greasy and the delivery was late.
            output: crust\:positive, toppings\:positive, sauce\:positive, cheese\:positive, grease\:negative, delivery time\:negative
            
            30.
            
            input: The tablet has a sharp display and the interface is responsive. It’s lightweight and portable. But the battery drains too quickly, and the camera quality is poor.
            output: display\:positive, interface\:positive, portability\:positive, battery life\:negative, camera\:negative
            
            31.
            
            input: The hotel room had a great view and was spacious. The bed was comfortable, and the bathroom was well-equipped. However, the room service was slow, and the TV had limited channels.
            output: view\:positive, space\:positive, bed\:positive, bathroom\:positive, room service\:negative, TV channels\:negative
            
            32.
            
            input: The coffee shop has a cozy atmosphere and friendly staff. The coffee was great, and the pastries were fresh. However, the seating was uncomfortable and the place was too noisy.
            output: atmosphere\:positive, staff\:positive, coffee\:positive, pastries\:positive, seating\:negative, noise\:negative
            
            33.
            
            input: The museum had a variety of exhibits and was well-maintained. The staff was informative, and the layout was easy to navigate. However, the entrance fee was too high, and the gift shop had limited items.
            output: exhibits\:positive, maintenance\:positive, staff\:positive, layout\:positive, entrance fee\:negative, gift shop\:negative
            
            34.
            
            input: The restaurant's ambiance was elegant, and the staff was attentive. The appetizers were great, and the main course was delicious. However, the portions were small, and the dessert was disappointing.
            output: ambiance\:positive, staff\:positive, appetizers\:positive, main course\:positive, portion size\:negative, dessert\:negative
            
            35.
            
            input: The online customer support was quick and efficient. The website was easy to navigate, and the checkout process was smooth. However, the product packaging was damaged, and the shipping took longer than expected.
            output: customer support\:positive, website\:positive, checkout\:positive, packaging\:negative, shipping\:negative
            
            36.
            
            input: The book was an interesting read, with great character development and a unique storyline. However, the pacing was slow, and the ending was predictable.
            output: character development\:positive, storyline\:positive, pacing\:negative, ending\:negative
            
            37.
            
            input: The video game has great graphics and a compelling story. The gameplay mechanics are smooth, and the sound design is immersive. However, there are frequent bugs, and the loading time is long.
            output: graphics\:positive, story\:positive, gameplay\:positive, sound design\:positive, bugs\:negative, loading time\:negative
            
            38.
            
            input: The skincare product is gentle on my skin and leaves it feeling smooth. The scent is pleasant

            38. input: The battery life is amazing and the screen is crystal clear.
               output: battery life:positive, screen:positive
            
            40. input: The receptionist was rude, but the room was clean.
               output: receptionist:negative, room:positive
            
            41. input: This shirt fits well but the fabric feels cheap.
               output: fit:positive, fabric:negative
            
            42. input: The plot of the book was a bit slow but very emotional.
               output: plot:negative, emotional content:positive
            
            43. input: The Wi-Fi in the hotel barely worked and the bathroom was dirty.
               output: Wi-Fi:negative, bathroom:negative
            
            44. input: Their customer support team responded quickly and solved my problem.
               output: customer support:positive, response time:positive
            
            45. input: I like the color and design, but the stitching came loose after one wash.
               output: color:positive, design:positive, stitching:negative
            
            46. input: The professor explains concepts very clearly but gives very tough exams.
               output: explanations:positive, exams:negative
            
            47. input: The application interface is clean and intuitive, but it crashes too often.
                output: interface:positive, usability:positive, stability:negative
            
            48. input: The museum had a good collection, but the lighting was poor.
                output: collection:positive, lighting:negative
            
            49. input: The speaker has good bass but lacks clarity at high volumes.
                output: bass:positive, clarity:negative
            
            50. input: Parking was easy, but getting out took forever due to traffic.
                output: parking:positive, traffic:negative
                
            51. input: The gameplay is addictive but the graphics are outdated.
                output: gameplay:positive, graphics:negative
            
  
            Now complete the following example-
            input: """
    eos_instruction = ' \noutput:'
    # Append and prepend the text to the user's input.
    modified_input = bos_instruction_id + user_input + eos_instruction
    return modified_input


# Assuming you have a function `model_predict` that takes the processed text and returns a prediction.
def model_predict(processed_text):
    # Here you would include the logic to make a prediction based on the processed_text.
    # This example simply returns the processed text.
    # tokenize input
    input_ids = t5_exp.tokenizer(processed_text, return_tensors="pt").input_ids
    # generate output
    outputs = t5_exp.model.generate(input_ids, max_length=128)
    return t5_exp.tokenizer.decode(outputs[0], skip_special_tokens=True)


# Create T5 model object along with instructions
model_checkpoint = "Homeskills/mt_instruct_absa"
t5_exp = T5Generator(model_checkpoint)

iface = gr.Interface(fn=lambda x: model_predict(preprocess_input(x)),
                     inputs=[gr.Textbox(label="Input review text here")],
                     outputs=gr.Textbox(label="Extracted aspects & sentiment"),
                     title="Aspect-Based Sentiment Analysis (ABSA)",
                     description="This app will take a sample review segment as input and return the extracted aspects and the corresponding sentiments.",
                     )
iface.launch()