Update app.py

app.py

@@ -19,62 +19,70 @@ def sentiment_analysis(text):
 # Open AI Chatbot Model
 openai.api_key = os.environ['openai_api']
 
-start_sequence = "\nAI:"
-restart_sequence = "\nHuman: "
+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=5000,
+    top_p=1,
+    frequency_penalty=0,
+    presence_penalty=0.6
+    
+    )
 
-prompt = "You can discuss any topic with the Chinese Chatbot assistant by typing Chinese in here"
+    return response.choices[0].message.content
 
-def openai_create(prompt):
+def openai_translation_ec(prompt):
 
-    response = openai.Completion.create(
-    model="text-davinci-003",
-    prompt=prompt,
-    temperature=0.9,
-    max_tokens=2048,
+    response = openai.ChatCompletion.create(
+    model="gpt-3.5-turbo",
+    messages=[
+        {"role":"system","content":"You are a sentense translator which translate article from English to Chinese. Don't do any task other than translation"},
+        {"role":"user","content":prompt}
+    ],
+    temperature=0.8,
+    max_tokens=5000,
     top_p=1,
     frequency_penalty=0,
-    presence_penalty=0.6,
-    stop=[" Human:", " AI:"]
+    presence_penalty=0.6
+    
     )
 
-    return response.choices[0].text
+    return response.choices[0].message.content
+
+def openai_translation_ce(prompt):
+
+    response = openai.ChatCompletion.create(
+    model="gpt-3.5-turbo",
+    messages=[
+        {"role":"system","content":"You are a sentense translator which translate article from Chinese to English. Don't do any task other than translation"},
+        {"role":"user","content":prompt}
+    ],
+    temperature=0.8,
+    max_tokens=5000,
+    top_p=1,
+    frequency_penalty=0,
+    presence_penalty=0.6
+    
+    )
+
+    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_create(inp)
+    output = openai_chatbot(inp)
     history.append((input, output))
     return history, history
 
-
-# Open AI Chinese Translation Model
-def translate_to_chinese(text_to_translate):
-    response = openai.Completion.create(
-      model="text-davinci-003",
-      prompt=f"Translate this short English sentence into Chinese:\n\n{text_to_translate}\n\n1.",
-      temperature=0.3,
-      max_tokens=2048,
-      top_p=1.0,
-      frequency_penalty=0.0,
-      presence_penalty=0.0
-    )
-    return response.choices[0].text.strip()
-
-# Open AI English Translation Model
-def translate_to_english(text_to_translate):
-    response = openai.Completion.create(
-      model="text-davinci-003",
-      prompt=f"Translate this short Chinese sentence into English:\n\n{text_to_translate}\n\n1.",
-      temperature=0.3,
-      max_tokens=2048,
-      top_p=1.0,
-      frequency_penalty=0.0,
-      presence_penalty=0.0
-    )
-    return response.choices[0].text.strip()
-
+"""# **Gradio Model**"""
 
 # Gradio Output Model
 with gr.Blocks() as demo:
@@ -82,7 +90,7 @@ with gr.Blocks() as demo:
     with gr.Tab("🗣️Chatbot"):
         gr.Markdown("This is a Chinese chatbot powered by the OpenAI language model. Enter your message below in Chinese and the chatbot will respond.")
         chatbot = gr.Chatbot()
-        message = gr.Textbox(placeholder=prompt)
+        message = gr.Textbox(placeholder="You can discuss any topic with the Chinese Chatbot assistant by typing Chinese in here")
         state = gr.State()
         submit = gr.Button("Send")
         submit.click(chatgpt_clone, inputs=[message, state], outputs=[chatbot, state])
@@ -97,18 +105,16 @@ with gr.Blocks() as demo:
         inputs = gr.Textbox(placeholder="Enter a short English sentence to translate to Chinese here.")
         outputs = gr.Textbox(label="Translation Result")
         proceed_button = gr.Button("Translate")
-        proceed_button.click(fn=translate_to_chinese, inputs=inputs, outputs=outputs)
+        proceed_button.click(fn=openai_translation_ec, inputs=inputs, outputs=outputs)
     with gr.Tab("🔤English Translation"):
         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.")
         inputs = gr.Textbox(placeholder="Enter a short Chinese sentence to translate to English here.")
         outputs = gr.Textbox(label="Translation Result")
         proceed_button = gr.Button("Translate")
-        proceed_button.click(fn=translate_to_english, inputs=inputs, outputs=outputs)
+        proceed_button.click(fn=openai_translation_ce, inputs=inputs, outputs=outputs)
     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.
     ''')