Update app.py

app.py

@@ -49,7 +49,7 @@ def chatgpt_clone(input, history):
     return history, history
 
 
-# Open AI Translation Model
+# Open AI Chinese Translation Model
 def translate_to_chinese(text_to_translate):
     response = openai.Completion.create(
       model="text-davinci-003",
@@ -62,6 +62,20 @@ def translate_to_chinese(text_to_translate):
     )
     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 sentence into English:\n\n{text_to_translate}\n\n1.",
+      temperature=0.3,
+      max_tokens=1024,
+      top_p=1.0,
+      frequency_penalty=0.0,
+      presence_penalty=0.0
+    )
+    return response.choices[0].text.strip()
+
+
     
 # Gradio Output Model
 with gr.Blocks() as demo:
@@ -72,17 +86,21 @@ with gr.Blocks() as demo:
         state = gr.State()
         submit = gr.Button("SEND")
         submit.click(chatgpt_clone, inputs=[message, state], outputs=[chatbot, state])
-    with gr.Tab("Translation to Chinese"):
-        inputs = gr.Textbox(placeholder="Enter an 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)
     with gr.Tab("Sentiment Analysis"):
         inputs = gr.Textbox(placeholder="Enter a Chinese positive or negative sentence here")
         outputs = gr.Textbox(label="Sentiment Analysis")
         proceed_button = gr.Button("proceed")           
         proceed_button.click(fn=sentiment_analysis, inputs=inputs, outputs=outputs)
-
+    with gr.Tab("Translation to Chinese"):
+        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)
+    with gr.Tab("Translation to English"):
+        inputs = gr.Textbox(placeholder="Enter a Short sentence to translate to English here.")
+        outputs = gr.Textbox(label="Translation Result")
+        proceed_button = gr.Button("Translate")
+        proceed_button.click(fn=translate_to_chinese, inputs=inputs, outputs=outputs)