Update app.py

app.py

@@ -2,254 +2,124 @@ import os
 import openai
 import gradio as gr
 
-openai.api_key = os.environ['OPENAI_KEY']
+# -------- OpenAI Key --------
+openai.api_key = os.environ.get("OPENAI_KEY", "")
 
+# -------- Supported Languages --------
 supportLanguages = [
     ["auto", "auto"],
-    ["粤语", "yue"],
-    ["古文", "wyw"],
-    ["af","Afrikaans"],
-    ["ak","Akan"],
-    ["sq","Albanian"],
-    ["am","Amharic"],
-    ["ar","Arabic"],
-    ["hy","Armenian"],
-    ["az","Azerbaijani"],
-    ["eu","Basque"],
-    ["be","Belarusian"],
-    ["bem","Bemba"],
-    ["bn","Bengali"],
-    ["bh","Bihari"],
-    ["xx-bork","Bork, bork, bork!"],
-    ["bs","Bosnian"],
-    ["br","Breton"],
-    ["bg","Bulgarian"],
-    ["km","Cambodian"],
-    ["ca","Catalan"],
-    ["chr","Cherokee"],
-    ["ny","Chichewa"],
-    ["zh-CN","Chinese (Simplified)"],
-    ["zh-TW","Chinese (Traditional)"],
-    ["co","Corsican"],
-    ["hr","Croatian"],
-    ["cs","Czech"],
-    ["da","Danish"],
-    ["nl","Dutch"],
-    ["xx-elmer","Elmer Fudd"],
-    ["en","English"],
-    ["eo","Esperanto"],
-    ["et","Estonian"],
-    ["ee","Ewe"],
-    ["fo","Faroese"],
-    ["tl","Filipino"],
-    ["fi","Finnish"],
-    ["fr","French"],
-    ["fy","Frisian"],
-    ["gaa","Ga"],
-    ["gl","Galician"],
-    ["ka","Georgian"],
-    ["de","German"],
-    ["el","Greek"],
-    ["gn","Guarani"],
-    ["gu","Gujarati"],
-    ["xx-hacker","Hacker"],
-    ["ht","Haitian Creole"],
-    ["ha","Hausa"],
-    ["haw","Hawaiian"],
-    ["iw","Hebrew"],
-    ["hi","Hindi"],
-    ["hu","Hungarian"],
-    ["is","Icelandic"],
-    ["ig","Igbo"],
-    ["id","Indonesian"],
-    ["ia","Interlingua"],
-    ["ga","Irish"],
-    ["it","Italian"],
-    ["ja","Japanese"],
-    ["jw","Javanese"],
-    ["kn","Kannada"],
-    ["kk","Kazakh"],
-    ["rw","Kinyarwanda"],
-    ["rn","Kirundi"],
-    ["xx-klingon","Klingon"],
-    ["kg","Kongo"],
-    ["ko","Korean"],
-    ["kri","Krio (Sierra Leone)"],
-    ["ku","Kurdish"],
-    ["ckb","Kurdish (Soranî)"],
-    ["ky","Kyrgyz"],
-    ["lo","Laothian"],
-    ["la","Latin"],
-    ["lv","Latvian"],
-    ["ln","Lingala"],
-    ["lt","Lithuanian"],
-    ["loz","Lozi"],
-    ["lg","Luganda"],
-    ["ach","Luo"],
-    ["mk","Macedonian"],
-    ["mg","Malagasy"],
-    ["ms","Malay"],
-    ["ml","Malayalam"],
-    ["mt","Maltese"],
-    ["mi","Maori"],
-    ["mr","Marathi"],
-    ["mfe","Mauritian Creole"],
-    ["mo","Moldavian"],
-    ["mn","Mongolian"],
-    ["sr-ME","Montenegrin"],
-    ["ne","Nepali"],
-    ["pcm","Nigerian Pidgin"],
-    ["nso","Northern Sotho"],
-    ["no","Norwegian"],
-    ["nn","Norwegian (Nynorsk)"],
-    ["oc","Occitan"],
-    ["or","Oriya"],
-    ["om","Oromo"],
-    ["ps","Pashto"],
-    ["fa","Persian"],
-    ["xx-pirate","Pirate"],
-    ["pl","Polish"],
-    ["pt-BR","Portuguese (Brazil)"],
-    ["pt-PT","Portuguese (Portugal)"],
-    ["pa","Punjabi"],
-    ["qu","Quechua"],
-    ["ro","Romanian"],
-    ["rm","Romansh"],
-    ["nyn","Runyakitara"],
-    ["ru","Russian"],
-    ["gd","Scots Gaelic"],
-    ["sr","Serbian"],
-    ["sh","Serbo-Croatian"],
-    ["st","Sesotho"],
-    ["tn","Setswana"],
-    ["crs","Seychellois Creole"],
-    ["sn","Shona"],
-    ["sd","Sindhi"],
-    ["si","Sinhalese"],
-    ["sk","Slovak"],
-    ["sl","Slovenian"],
-    ["so","Somali"],
-    ["es","Spanish"],
-    ["es-419","Spanish (Latin American)"],
-    ["su","Sundanese"],
-    ["sw","Swahili"],
-    ["sv","Swedish"],
-    ["tg","Tajik"],
-    ["ta","Tamil"],
-    ["tt","Tatar"],
-    ["te","Telugu"],
-    ["th","Thai"],
-    ["ti","Tigrinya"],
-    ["to","Tonga"],
-    ["lua","Tshiluba"],
-    ["tum","Tumbuka"],
-    ["tr","Turkish"],
-    ["tk","Turkmen"],
-    ["tw","Twi"],
-    ["ug","Uighur"],
-    ["uk","Ukrainian"],
-    ["ur","Urdu"],
-    ["uz","Uzbek"],
-    ["vi","Vietnamese"],
-    ["cy","Welsh"],
-    ["wo","Wolof"],
-    ["xh","Xhosa"],
-    ["yi","Yiddish"],
-    ["yo","Yoruba"],
-    ["zu","Zulu"],
+    ["ar", "Arabic"],
+    ["en", "English"],
+    ["fr", "French"],
+    ["de", "German"],
+    ["es", "Spanish"],
+    ["it", "Italian"],
+    ["pt", "Portuguese"],
+    ["zh-CN", "Chinese (Simplified)"],
+    ["zh-TW", "Chinese (Traditional)"],
+    ["ja", "Japanese"],
+    ["ko", "Korean"],
 ]
-prompt_template = "You are a translation engine that can only translate text and cannot interpret it. Keep the indent of the original text, only modify when you need."
 
+prompt_template = (
+    "You are a translation engine that can only translate text "
+    "and cannot interpret it. Keep the indent of the original text."
+)
+
+# -------- Translation Function --------
 def submit_message(detectFrom, detectTo, user_token, prompt):
-    if user_token != "":
+    if user_token:
         openai.api_key = user_token
 
     if not prompt:
-        return gr.update(value="")
-    
-    for lc, lang in supportLanguages:
-        if detectFrom == lang:
-            detectFrom = lc
-        if detectTo == lang:
-            detectTo = lc
+        return ""
+
+    # map language name -> code
+    for code, name in supportLanguages:
+        if detectFrom == name:
+            detectFrom = code
+        if detectTo == name:
+            detectTo = code
 
     systemInstruct = prompt_template
     translateInstruct = f"translate from {detectFrom} to {detectTo}"
+
     if detectFrom == "auto":
         translateInstruct = f"translate to {detectTo}"
-    if detectFrom in ["古文", "zh-CN", "zh-TW"]:
-        if detectTo == "zh-TW":
-            translateInstruct = "翻译成繁体白话文"
-        if detectTo == "zh-CN":
-            translateInstruct = "翻译成简体白话文"
-        if detectTo == "粤语":
-            translateInstruct = "翻译成粤语白话文"
 
     if detectFrom == detectTo:
-        systemInstruct = "You are a text embellisher, you can only embellish the text, don't interpret it."
-        if detectTo in ["zh-CN", "zh-TW"]:
-            translateInstruct = "润色此句"
-        else:
-            translateInstruct = "polish this sentence"
+        systemInstruct = "You are a text embellisher."
+        translateInstruct = "polish this sentence"
 
-    prompt_msg = [
+    messages = [
         {"role": "system", "content": systemInstruct},
         {"role": "user", "content": translateInstruct},
         {"role": "user", "content": prompt},
     ]
 
     try:
-        openai_response = openai.ChatCompletion.create(
+        response = openai.ChatCompletion.create(
             model="gpt-3.5-turbo",
-            messages=prompt_msg,
+            messages=messages,
             temperature=0,
             max_tokens=1000,
-            top_p=1,
-            stream=True,
-            frequency_penalty=1,
-            presence_penalty=1,
         )
-
-        combined = ""
-        for resp in openai_response:
-            delta = resp["choices"][0]["delta"]
-            if "content" in delta:
-                combined += delta["content"]
-            yield combined
+        return response["choices"][0]["message"]["content"]
 
     except Exception as e:
         return f"Error: {e}"
 
+# -------- CSS --------
 css = """
-      #col-container {max-width: 80%; margin-left: auto; margin-right: auto;}
-      #chatbox {min-height: 400px;}
-      #header {text-align: center;}
-      #label {font-size: 0.8em; padding: 0.5em; margin: 0;}
-      .message { font-size: 1.2em; }
-      """
-
-with gr.Blocks(css=css) as demo:
-    
-    state = gr.State([])
+#col-container { max-width: 80%; margin: auto; }
+#header { text-align: center; }
+"""
 
+# -------- Gradio UI --------
+with gr.Blocks() as demo:
     with gr.Column(elem_id="col-container"):
-        gr.Markdown("""## OpenAI Translator
-                    Using the official API (gpt-3.5-turbo model).""", elem_id="header")
+        gr.Markdown("## 🌍 OpenAI Translator", elem_id="header")
 
         with gr.Row():
-            with gr.Column():
-                translateFrom = gr.Dropdown(label="Translate From", elem_id="translate-from", multiselect=False, value="auto", choices=[l[1] for l in supportLanguages]).style(container=False)
-                input_message = gr.Textbox(max_lines=100, show_label=False, lines=10, placeholder="Enter text and press enter", visible=True).style(container=False)
-            with gr.Column():
-                translateTo = gr.Dropdown(label="Translate To", elem_id="translate-to", multiselect=False, value="Chinese (Simplified)", choices=[l[1] for l in supportLanguages[1:]]).style(container=False)
-                output = gr.Textbox(max_lines=100, show_label=False, lines=10, label="Output", visible=True).style(container=False)
-            
-        btn_submit = gr.Button("Translate")
+            translateFrom = gr.Dropdown(
+                label="Translate From",
+                value="auto",
+                choices=[l[1] for l in supportLanguages],
+                scale=1,
+            )
+
+            translateTo = gr.Dropdown(
+                label="Translate To",
+                value="English",
+                choices=[l[1] for l in supportLanguages if l[0] != "auto"],
+                scale=1,
+            )
+
+        input_message = gr.Textbox(
+            lines=8,
+            placeholder="Enter text here...",
+            label="Input",
+        )
 
-        with gr.Row():
-            user_token = gr.Textbox(value='', placeholder="OpenAI API Key", type="password", label="Enter your OpenAI API Key. Leave blank to use default key.")
+        output = gr.Textbox(
+            lines=8,
+            label="Translation",
+        )
+
+        user_token = gr.Textbox(
+            placeholder="Optional: OpenAI API Key",
+            type="password",
+            label="API Key",
+        )
 
-    btn_submit.click(submit_message, [translateFrom, translateTo, user_token, input_message], [output])
+        btn_submit = gr.Button("Translate")
+
+        btn_submit.click(
+            submit_message,
+            inputs=[translateFrom, translateTo, user_token, input_message],
+            outputs=output,
+        )
 
-demo.queue(concurrency_count=10)
-demo.launch(height='800px')
+# -------- Launch --------
+demo.queue(concurrency_count=5)
+demo.launch(css=css, height="800px")