app.py

import gradio as gr
import os
from transformers import AutoTokenizer
from get_loss.get_loss_hf import run_get_loss
import pdb
from types import SimpleNamespace
import pandas as pd
import plotly.express as px
# os.system('git clone https://github.com/EleutherAI/lm-evaluation-harness')
# os.system('cd lm-evaluation-harness')
# os.system('pip install -e .')
# -i https://pypi.tuna.tsinghua.edu.cn/simple
# 第一个功能：基于输入文本和对应的损失值对文本进行着色展示

def color_text(text_list=["hi", "FreshEval","!"], loss_list=[0.1,0.7]):
    """
    根据损失值为文本着色。
    """
    highlighted_text = []
    # print('loss_list',loss_list)
    # ndarray to list
    loss_list = loss_list.tolist()
    loss_list=[0]+loss_list
    # print('loss_list',loss_list)
    # print('text_list',text_list)
    # pdb.set_trace()
    for text, loss in zip(text_list, loss_list):
        # color = "#FF0000" if float(loss) > 0.5 else "#00FF00"
        color=loss/20#TODO rescale
        # highlighted_text.append({"text": text, "bg_color": color})
        highlighted_text.append((text, color))
        print('highlighted_text',highlighted_text)
    return highlighted_text

# 第二个功能：根据 ID 列表和 tokenizer 将 ID 转换为文本，并展示
def get_text(ids_list=[0.1,0.7], tokenizer=None):
    """
    给定一个 ID 列表和 tokenizer 名称，将这些 ID 转换成文本。
    """
    # return ['Hi', 'Adam']
    # tokenizer = AutoTokenizer.from_pretrained(tokenizer)
    # print('ids_list',ids_list)
    # pdb.set_trace()
    text=[]
    for id in ids_list:
        text.append(  tokenizer.decode(id, skip_special_tokens=True))
    # 这里只是简单地返回文本，但是可以根据实际需求添加颜色或其他样式
    print(f'L41:{text}')
    return text


# def get_ids_loss(text, tokenizer, model):
#     """
#     给定一个文本，model and its tokenizer,返回其对应的 IDs 和损失值。
#     """
#     # tokenizer = AutoTokenizer.from_pretrained(tokenizer_name)
#     # model = AutoModelForCausalLM.from_pretrained(model_name)
#     # 这里只是简单地返回 IDs 和损失值，但是可以根据实际需求添加颜色或其他样式
#     return [1, 2], [0.1, 0.7]

def harness_eval(question, choices, answer_index, model=None,tokenizer=None):
    '''
    use harness to test one question,  can specify the model, (extract or ppl)
    '''
    # TODO add the model and its score
    # torch.nn.functional.softmax(output.logits, dim=0)
    # topk = torch.topk(output.logits, 5)
    
    return {'A':0.5, 'B':0.3, 'C':0.1, 'D':0.1}




def plotly_plot():#(df, x, y, color,title, x_title, y_title):
        # plotly_plot(sample_df, 'date', 'loss_mean_at_1000', 'model','ppl with time', 'time', 'ppl')
    df=pd.read_csv('./data/tmp.csv')
    df['date'] = pd.to_datetime(df['date'])
    # sort by date
    df.sort_values(by='date', inplace=True)

    # use a dic to filter the dataframe
    df = df[df['file_name'] == 'arxiv_computer_science']

    x,y,color,title, x_title, y_title='date', 'loss_mean_at_1000', 'model','ppl with time', 'time', 'ppl'

    fig = px.line(df, x=x, y=y, color=color,title=title)
    fig.update_xaxes(title_text=x_title)
    fig.update_yaxes(title_text=y_title)
    # fig.update_layout()
    return fig

# def plotly_plot(df, x, y, color, title, x_title, y_title):
#     fig = px.line(df, x=x, y=y, color=color, title=title)
#     fig.update_xaxes(title_text=x_title)
#     fig.update_yaxes(title_text=y_title)
#     return fig




def color_pipeline(texts=["Hi","FreshEval","!"],  model=None):
    """
    给定一个文本，返回其对应的着色文本。
    """
    print('text,model',texts,model)
    args=SimpleNamespace(texts=texts,model=model)
    print(f'L60,text:{texts}')
    rtn_dic=run_get_loss(args)
    # print(rtn_dic)
    # pdb.set_trace()
    # {'logit':logit,'input_ids':input_chunk,'tokenizer':tokenizer,'neg_log_prob_temp':neg_log_prob_temp}
    ids, loss =rtn_dic['input_ids'],rtn_dic['loss']#= get_ids_loss(text, tokenizer, model)
    # notice here is numpy ndarray
    tokenizer=rtn_dic['tokenizer'] # get tokenizer
    text = get_text(ids, tokenizer)
    # print('ids, loss ,text',ids, loss ,text)
    return color_text(text, loss)


# TODO can this be global ? maybe need session to store info of the user

# 创建 Gradio 界面
with gr.Blocks() as demo:
    with gr.Tab("color your text"):
        with gr.Row():
            text_input = gr.Textbox(label="input text", placeholder="input your text here...")
            # file_input = gr.File(file_count="multiple",label='to add content')#
            # TODO craw and drop the file

            # loss_input = gr.Number(label="loss")
            model_input = gr.Textbox(label="model name", placeholder="input your model name here... now I am trying phi-2...")#TODO make a choice here
        output_box=gr.HighlightedText(label="colored text")#,interactive=True
        
        gr.Examples(
                [
                    ["Hi FreshEval !", "microsoft/phi-2"],
                    ["Hello FreshBench !", "/home/sribd/chenghao/models/phi-2"],
                ],
                [text_input, model_input],)
            #     cache_examples=True,
            #     # cache_examples=False,
            #     fn=color_pipeline,
            #     outputs=output_box
            # )
            # TODO select models that can be used online
            # TODO maybe add our own models


        color_text_output = gr.HTML(label="colored text")
        color_text_button = gr.Button("color the text").click(color_pipeline, inputs=[text_input, model_input], outputs=output_box)

        # markdown
        gr.Markdown('### How to use this app')
        




        date_time_input = gr.Textbox(label="the date when the text is generated")#TODO add date time input
        description_input = gr.Textbox(label="description of the text")
        submit_button = gr.Button("submit a post or record").click()
        #TODO add model and its score

    with gr.Tab('test your qeustion'):
        '''
        use extract, or use ppl
        '''
        question=gr.Textbox(label="input question", placeholder='input your question here...')
        answer_index=gr.Textbox(label="right answer index", placeholder='index for right anser here, start with 0')#TODO add multiple choices, 
        choices=gr.Textbox(placeholder='input your other choices here...')
        
        # test_button=gr.Button('test').click(harness_eval())# TODO figure out the input and output

        answer_type=gr.Dropdown(label="answer type", choices=['extract', 'ppl'])
        #TODO add the model and its score
        answer_label=gr.Label('the answers\'s detail')# RETURN the answer and its score,in the form of dic{str: float}

        test_question_button=gr.Button('test question').click(harness_eval,inputs=[question, choices, answer_index ,answer_type],outputs=[answer_label])

        forecast_q='A Ukrainian counteroffensive began in 2023, though territorial gains by November 2023 were limited (Economist, BBC, Newsweek). The question will be suspended on 31 July 2024 and the outcome determined using data as reported in the Brookings Institution\'s "Ukraine Index" (Brookings Institution - Ukraine Index, see "Percentage of Ukraine held by Russia" chart). If there is a discrepancy between the chart data and the downloaded data (see "Get the data" within the "NET TERRITORIAL GAINS" chart border), the downloaded data will be used for resolution.'
        answer_list=['Less than 5%','At least 5%, but less than 10%','At least 10%, but less than 15%','At least 15%, but less than 20%','20% or more' ]
         
        gr.Examples([
                    [forecast_q, '&&&&&&'.join(answer_list), '0']
                     ],
                    [question, choices, answer_index])

        date_time_input = gr.Textbox(label="the date when the text is generated")#TODO add date time input
        description_input = gr.Textbox(label="description of the text")
        submit_button = gr.Button("submit a post or record").click()

        #TODO add the model and its score

        def test_question(question, answer, other_choices):
            '''
            use extract, or use ppl
            '''
            answer_ppl, other_choices_ppl =  (question, answer, other_choices)
            return answer_ppl, other_choices_ppl



    with gr.Tab("model text ppl with time"):
        '''
        see the matplotlib example, to see ppl with time, select the models
        '''
        # load the json file with time,

        # sample_df=pd.DataFrame({'time':pd.date_range('2021-01-01', periods=6), 'ppl': [1,2,3,4,5,6]})
        pd_df=pd.read_csv('./data/tmp.csv')
        pd_df['date'] = pd.to_datetime(pd_df['date'])
        print(pd_df.head)
        # gr_df=gr.Dataframe(pd_df)
        gr_df=pd_df


        # print(gr_df.head)
        print('done')
        # sample
        plot=gr.Plot(label='model text ppl')
        # plotly_plot(gr_df, 'date', 'loss_mean_at_1000', 'model','ppl with time', 'time', 'ppl')
        # draw_pic_button=gr.Button('draw the pic').click(plotly_plot,inputs=['gr_df', 'date', 'loss_mean_at_1000', 'model','ppl with time', 'time', 'ppl'],outputs=[plot])
        draw_pic_button=gr.Button('draw the pic').click(plotly_plot,inputs=[],outputs=[plot])


    
    with gr.Tab("model quesion acc with time"):
        '''
        see the matplotlib example, to see ppl with time, select the models
        ''' 
        #


    with gr.Tab("hot questions"):
        '''
        see the questions and answers
        '''
        with gr.Tab("ppl"):
            '''
            see the questions
            '''
        

demo.launch(debug=True)





# import gradio as gr
# import os
# os.system('python -m spacy download en_core_web_sm')
# import spacy
# from spacy import displacy

# nlp = spacy.load("en_core_web_sm")

# def text_analysis(text):
#     doc = nlp(text)
#     html = displacy.render(doc, style="dep", page=True)
#     html = (
#         "<div style='max-width:100%; max-height:360px; overflow:auto'>"
#         + html
#         + "</div>"
#     )
#     pos_count = {
#         "char_count": len(text),
#         "token_count": 0,
#     }
#     pos_tokens = []

#     for token in doc:
#         pos_tokens.extend([(token.text, token.pos_), (" ", None)])

#     return pos_tokens, pos_count, html

# demo = gr.Interface(
#     text_analysis,
#     gr.Textbox(placeholder="Enter sentence here..."),
#     ["highlight", "json", "html"],
#     examples=[
#         ["What a beautiful morning for a walk!"],
#         ["It was the best of times, it was the worst of times."],
#     ],
# )

# demo.launch()



# # lm-eval 
# # lm-evaluation-harness