#pip install transformers
#pip install gradio
import gradio as gr

from transformers import AutoTokenizer, AutoModelForCausalLM
from transformers import pipeline

tokenizer = AutoTokenizer.from_pretrained('gpt2')
model = AutoModelForCausalLM.from_pretrained('gpt2')

#model = AutoModelForSeq2SeqLM.from_pretrained("t5-base")

#set theme 
import gradio as gr
from gradio.themes.base import Base
from gradio.themes.utils import colors, fonts, sizes

theme = gr.themes.Base(
    primary_hue="pink",
    secondary_hue="pink",
    neutral_hue="pink",
    font=[gr.themes.GoogleFont('Merriweather'), 'ui-sans-serif', 'system-ui', 'sans-serif'],
).set(
    body_text_color='*primary_500',
    body_text_color_dark='*secondary_600',
    body_text_weight='500',
    background_fill_primary='*primary_50',
    background_fill_secondary='*primary_100',
    border_color_accent='*primary_400',
    button_border_width='*block_label_border_width'
)


def prompt_fx(text):
    translator = pipeline(
        model=model,
        tokenizer=tokenizer
    )
    response = translator(text)
    return response

def clear_out():
    return ""


with gr.Blocks(theme=theme) as demo:

    gr.Markdown(
    """
    # AI for all
    """)


    with gr.Tab("Home"):
      prompt = gr.Textbox(label="Enter prompt here:")
      output = gr.Textbox(label="Response")
    with gr.Row():
            button = gr.Button("Submit", variant="primary")
            clear = gr.Button("Clear")
            button.click(fn=prompt_fx, inputs=prompt, outputs=output)
            clear.click(fn=clear_out, outputs=prompt)

    with gr.Tab("About"):
        gr.Markdown(
            """
            AI for All is a fine-tuned version of Open AI's GPT2 model. It was 
            fine-tuned using readings from important scholars on power dynamics 
            and how they affect oppressed people. Examples include Walter Rodney,
            Angela Davis, and Audre Lorde. Technical specifics can be found on 
            the [HuggingFace page](https://link-url-here.org).
            
            The goal of this project is to center marginalized voices. Because 
            of the way our society is structured, we are never considered
            first. Efforts towards the advancement of technology frequently 
            ignore the struggles of oppressed people, so in this project I am 
            **finally** creating something for us. 
            """)

        
    with gr.Accordion("Quick Links"):
        gr.Markdown("""
          [HuggingFace page](https://link-url-here.org)
          [GitHub](https://link-url-here.org)
          """)
        

def greet(name):
    return "Hello " + name + "!!"

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