app.py

from transformers import BertTokenizerFast,TFBertForSequenceClassification,TextClassificationPipeline
import numpy as np
import tensorflow as tf
import gradio as gr 
import openai
import os

# Sentiment Analysis Pre-Trained Model
model_path = "leadingbridge/sentiment-analysis"
tokenizer = BertTokenizerFast.from_pretrained(model_path)
model = TFBertForSequenceClassification.from_pretrained(model_path, id2label={0: 'negative', 1: 'positive'} )

def sentiment_analysis(text):
  pipe = TextClassificationPipeline(model=model, tokenizer=tokenizer)
  result = pipe(text)
  return result


# Open AI Chatbot Model
openai.api_key = os.environ['openai_api']

def openai_chatbot(prompt):

    response = openai.ChatCompletion.create(
    model="gpt-3.5-turbo",
    messages=[
        {"role":"system","content":"You are a general chatbot that can answer anything"},
        {"role":"user","content":prompt}
    ],
    temperature=0.8,
    max_tokens=3000,
    top_p=1,
    frequency_penalty=0,
    presence_penalty=0.6
    
    )

    return response.choices[0].message.content

def openai_translation_ec(prompt):

    response = openai.ChatCompletion.create(
    model="gpt-3.5-turbo",
    messages=[
        {"role":"system","content":"Translate the article to Chinese:"},
        {"role":"user","content":prompt}
    ],
    temperature=0.8,
    max_tokens=3000,
    top_p=1,
    frequency_penalty=0,
    presence_penalty=1
    
    )

    return response.choices[0].message.content

def openai_translation_ce(prompt):

    response = openai.ChatCompletion.create(
    model="gpt-3.5-turbo",
    messages=[
        {"role":"system","content":"Translate the article to English:"},
        {"role":"user","content":prompt}
    ],
    temperature=0.8,
    max_tokens=3000,
    top_p=1,
    frequency_penalty=0,
    presence_penalty=1
    
    )

    return response.choices[0].message.content

def chatgpt_clone(input, history):
    history = history or []
    s = list(sum(history, ()))
    s.append(input)
    inp = ' '.join(s)
    output = openai_chatbot(inp)
    history.append((input, output))
    return history, history

"""# **Gradio Model**"""

# Gradio Output Model
with gr.Blocks() as demo:
    gr.Markdown('Welcome to the Chinese NLP Demo! Please select a model tab to interact with:')
    with gr.Tab("🤖Chatbot"):
        gr.Markdown("""<h4><center>🤖Chatbot</center></h4>""")
        chatbot = gr.Chatbot()
        message = gr.Textbox(placeholder="You can discuss any topic with the Chinese Chatbot assistant by typing Chinese in here",lines=3)
        state = gr.State()
        submit = gr.Button("Send")
        submit.click(chatgpt_clone, inputs=[message, state], outputs=[chatbot, state])
        gr.Markdown("This is a Chinese chatbot powered by the OpenAI language model. Enter your message above in Chinese and the chatbot will respond.")   
    with gr.Tab("🤗Sentiment Analysis"):
        gr.Markdown("""<h4><center>🤗Sentiment Analysis</center></h4>""")
        inputs = gr.Textbox(placeholder="Type a Chinese sentence here, either positive or negative in sentiment.",lines=3)
        outputs = gr.Textbox(label="Sentiment Analysis")
        proceed_button = gr.Button("Proceed")           
        proceed_button.click(fn=sentiment_analysis, inputs=inputs, outputs=outputs)
        gr.Markdown("This is a self-trained fine-tuned model using Chinese BERT for sentiment analysis. Enter a sentence in Chinese in the input box and click the 'proceed' button to get the sentiment analysis result.")    
    with gr.Tab("🀄Chinese Translation"):
        gr.Markdown("""<h4><center>🀄Chinese Translation</center></h4>""")
        inputs = gr.Textbox(placeholder="Enter a short English sentence to translate to Chinese here.",lines=3)
        outputs = gr.Textbox(label="Translation Result")
        proceed_button = gr.Button("Translate")
        proceed_button.click(fn=openai_translation_ec, inputs=inputs, outputs=outputs)
        gr.Markdown("This model translate an English sentence to Chinese using the OpenAI engine. Enter an English short sentence in the input box and click the 'Translate' button to get the translation result in Chinese.")  
    with gr.Tab("🔤English Translation"):
        gr.Markdown("""<h4><center>🔤English Translation</center></h4>""")
        inputs = gr.Textbox(placeholder="Enter a short Chinese sentence to translate to English here.",lines=3)
        outputs = gr.Textbox(label="Translation Result")
        proceed_button = gr.Button("Translate")
        proceed_button.click(fn=openai_translation_ce, inputs=inputs, outputs=outputs)
        gr.Markdown("This model translate a Chinese sentence to English using the OpenAI engine. Enter a Chinese short sentence in the input box and click the 'Translate' button to get the translation result in English.")
    gr.Markdown('''
    We are happy to share with you some Chinese language models that we've made using NLP. When we looked online, we noticed that there weren't many resources available for Chinese NLP, so we hope that our models can be useful to you.
    We want to mention that these models aren't perfect and there is still room for improvement. Because of limited resources, there might be some mistakes or limitations in the models.
    However, We hope that you find them helpful and that you can help make them even better.
    ''')



demo.launch(inline=False)