'''
Author: i316984 daisy.mao@sap.com
Date: 2024-02-16 12:02:26
LastEditors: i316984 daisy.mao@sap.com
LastEditTime: 2024-02-16 12:30:20
FilePath: /undefined/Users/i316984_1/hackaithon101/app.py
Description: 这是默认设置,请设置`customMade`, 打开koroFileHeader查看配置 进行设置: https://github.com/OBKoro1/koro1FileHeader/wiki/%E9%85%8D%E7%BD%AE
'''

import gradio as gr
import requests
import os
import json

API_URL = "https://api-inference.huggingface.co/models/cardiffnlp/twitter-roberta-base-sentiment"

bt = os.environ['HACKAITHONBEARERTOKEN']
headers = {"Authorization": bt }
key_mapping = {'LABEL_0':'negative','LABEL_1':'neutral','LABEL_2':'positive'}

def query(data):
    response = requests.post(API_URL, headers=headers, json=data)
    list_json = response.json()
    json_resp = list_json[0]
    return "V2"+json.dumps([{k: key_mapping.get(v, v) for k, v in temp_dict.items()} for temp_dict in json_resp], indent=4)


def greet(howareyoufeeling):
    output = query({"inputs":howareyoufeeling})
    print (str(output))
    return str(output)

iface = gr.Interface(
  fn=greet, inputs=["text"], outputs="text",  allow_flagging="never")
iface.launch()