Update app.py

app.py

@@ -30,15 +30,17 @@ pipe = pipeline(
 def translate_text(input_text, target_language):
     prompt = f"請將以下文字翻譯成{target_language}：\n\n{input_text}"
     
-    response = openai.ChatCompletion.create(
-        model="gpt-4o",
-        messages=[
-            {"role": "user", "content": prompt}
-        ]
-    )
-    
-    translated_text = response['choices'][0]['message']['content'].strip()
-    return translated_text
+    try:
+        response = openai.ChatCompletion.create(
+            model="gpt-3.5-turbo",  # 更改模型名稱
+            messages=[
+                {"role": "user", "content": prompt}
+            ]
+        )
+        translated_text = response['choices'][0]['message']['content'].strip()
+        return translated_text
+    except Exception as e:
+        raise gr.Error(f"翻譯過程中出現錯誤：{str(e)}")
 
 @spaces.GPU
 def transcribe(inputs, task, translate_option, target_language):
@@ -115,13 +117,15 @@ def yt_transcribe(yt_url, task, translate_option, target_language, max_filesize=
 
 demo = gr.Blocks(theme=gr.themes.Ocean())
 
+language_options = ["英文", "日文", "法文", "德文", "西班牙文", "繁體中文", "簡體中文", "越南文", "泰文"]
+
 mf_transcribe = gr.Interface(
     fn=transcribe,
     inputs=[
         gr.Audio(sources="microphone", type="filepath"),
         gr.Radio(["transcribe", "translate"], label="任務", value="transcribe"),
         gr.Radio(["是", "否"], label="是否翻譯轉錄結果", value="否"),
-        gr.Dropdown(["英文", "日文", "法文", "德文", "西班牙文"], label="目標語言", value="英文")
+        gr.Dropdown(language_options, label="目標語言", value="英文")
     ],
     outputs="text",
     title="清華大學多模態課程＆廖老師嫡傳弟子-第二組 「語音轉文字」model",
@@ -138,10 +142,10 @@ file_transcribe = gr.Interface(
         gr.Audio(sources="upload", type="filepath", label="音訊檔案"),
         gr.Radio(["transcribe", "translate"], label="任務", value="transcribe"),
         gr.Radio(["是", "否"], label="是否翻譯轉錄結果", value="否"),
-        gr.Dropdown(["英文", "日文", "法文", "德文", "西班牙文"], label="目標語言", value="英文")
+        gr.Dropdown(language_options, label="目標語言", value="英文")
     ],
     outputs="text",
-    title="Whisper Large V3: 轉錄音訊",
+    title="清華大學多模態課程＆廖老師嫡傳弟子-第二組 「語音轉文字」model：上傳音檔",
     description=(
         "只需點擊一下按鈕，即可轉錄長篇的麥克風或音訊輸入！演示使用了"
         f"檢查點 [{MODEL_NAME}](https://huggingface.co/{MODEL_NAME}) 和 🤗 Transformers 來轉錄任意長度的音訊文件。"
@@ -155,10 +159,10 @@ yt_transcribe = gr.Interface(
         gr.Textbox(lines=1, placeholder="在此處貼上 YouTube 視頻的 URL", label="YouTube URL"),
         gr.Radio(["transcribe", "translate"], label="任務", value="transcribe"),
         gr.Radio(["是", "否"], label="是否翻譯轉錄結果", value="否"),
-        gr.Dropdown(["英文", "日文", "法文", "德文", "西班牙文"], label="目標語言", value="英文")
+        gr.Dropdown(language_options, label="目標語言", value="英文")
     ],
     outputs=["html", "text"],
-    title="Whisper Large V3: 轉錄 YouTube",
+    title="清華大學多模態課程＆廖老師嫡傳弟子-第二組 「語音轉文字」model: 轉錄 YouTube",
     description=(
         "只需點擊一下按鈕，即可轉錄長篇的 YouTube 視頻！演示使用了"
         f"檢查點 [{MODEL_NAME}](https://huggingface.co/{MODEL_NAME}) 和 🤗 Transformers 來轉錄任意長度的視頻文件。"