app.py

import gradio as gr
import model_wrapper



model = model_wrapper.PredictionModel()

def pretty_print_opinion(opinion_dict):
    res = []
    maxlen = max([len(key) for key in opinion_dict.keys()]) + 2
    maxlen = 0
    for key, value in opinion_dict.items():
        if key == 'Polarity':
            res.append(f'{(key + ":").ljust(maxlen)} {value}')
        else:
            res.append(f'{(key + ":").ljust(maxlen)} \'{" ".join(value[0])}\'')
    return '\n'.join(res) + '\n'


def predict(text):
    predictions = model.predict([text])
    prediction = predictions[0]
    results = []
    if not prediction['opinions']:
        return 'No opinions detected'
    for opinion in prediction['opinions']:
        results.append(pretty_print_opinion(opinion))

    return '\n'.join(results)


iface = gr.Interface(fn=predict, inputs="text", outputs="text")
iface.launch()