Update app.py

app.py

@@ -3,7 +3,7 @@ import os
 import shutil
 from docx import Document
 from pptx import Presentation
-from deep_translator import GoogleTranslator
+import openai
 
 def translate_document(file, target_language, target_country, api_key):
     log_messages = []
@@ -18,6 +18,8 @@ def translate_document(file, target_language, target_country, api_key):
         shutil.copy(original_file_path, copied_file_path)
         log_messages.append(f"Copied file to {copied_file_path}")
 
+        openai.api_key = api_key  # 設定 OpenAI API 金鑰
+
         # 處理 .docx 文件
         if copied_file_path.endswith(".docx"):
             doc = Document(copied_file_path)
@@ -25,7 +27,10 @@ def translate_document(file, target_language, target_country, api_key):
             # 翻譯並替換段落文本
             for paragraph in doc.paragraphs:
                 if paragraph.text.strip():  # 只翻譯非空段落
-                    translated_text = GoogleTranslator(source='auto', target=target_language).translate(paragraph.text)
+                    translated_text = openai.ChatCompletion.create(
+                        model="gpt-3.5-turbo",
+                        messages=[{"role": "user", "content": f"Translate this to {target_language}: {paragraph.text}"}]
+                    )["choices"][0]["message"]["content"]
                     paragraph.text = translated_text  # 用翻譯後的文本替換原始文本
             
             # 處理表格中的文本
@@ -33,16 +38,12 @@ def translate_document(file, target_language, target_country, api_key):
                 for row in table.rows:
                     for cell in row.cells:
                         if cell.text.strip():  # 只翻譯非空單元格
-                            translated_text = GoogleTranslator(source='auto', target=target_language).translate(cell.text)
+                            translated_text = openai.ChatCompletion.create(
+                                model="gpt-3.5-turbo",
+                                messages=[{"role": "user", "content": f"Translate this to {target_language}: {cell.text}"}]
+                            )["choices"][0]["message"]["content"]
                             cell.text = translated_text  # 用翻譯後的文本替換原始文本
 
-            # 處理文本框和形狀中的文本
-            for inline_shape in doc.inline_shapes:
-                if inline_shape.type == 3:  # 3 表示文本框
-                    if inline_shape.text_frame and inline_shape.text_frame.text.strip():
-                        translated_text = GoogleTranslator(source='auto', target=target_language).translate(inline_shape.text_frame.text)
-                        inline_shape.text_frame.text = translated_text  # 替換文本框中的文本
-
             output_path = "translated_" + os.path.basename(original_file_path)
             doc.save(output_path)
             log_messages.append(f"Saved translated document as {output_path}.")
@@ -55,20 +56,12 @@ def translate_document(file, target_language, target_country, api_key):
             for slide in prs.slides:
                 for shape in slide.shapes:
                     if hasattr(shape, "text") and shape.text.strip():  # 只翻譯非空文本框
-                        # 保存原始格式信息
-                        original_font_size = None
-                        if shape.has_text_frame:
-                            original_font_size = shape.text_frame.paragraphs[0].runs[0].font.size
-                        
-                        translated_text = GoogleTranslator(source='auto', target=target_language).translate(shape.text)
+                        translated_text = openai.ChatCompletion.create(
+                            model="gpt-3.5-turbo",
+                            messages=[{"role": "user", "content": f"Translate this to {target_language}: {shape.text}"}]
+                        )["choices"][0]["message"]["content"]
                         shape.text = translated_text  # 用翻譯後的文本替換原始文本
 
-                        # 恢復原始格式
-                        if shape.has_text_frame and original_font_size is not None:
-                            for paragraph in shape.text_frame.paragraphs:
-                                for run in paragraph.runs:
-                                    run.font.size = original_font_size
-
             output_path = "translated_" + os.path.basename(original_file_path)
             prs.save(output_path)
             log_messages.append(f"Saved translated presentation as {output_path}.")
@@ -87,9 +80,9 @@ iface = gr.Interface(
     fn=translate_document,
     inputs=[
         gr.File(label="上傳文件 (.docx 或 .pptx)"),
-        gr.Dropdown(choices=["en", "es", "fr", "de", "ja", "ko", "zh-TW", "vi"], label="Target Language"),
+        gr.Dropdown(choices=["English", "Spanish", "French", "German", "Japanese", "Korean", "Chinese", "Vietnamese"], label="Target Language"),
         gr.Dropdown(choices=["US", "UK", "ES", "FR", "DE", "JP", "KR", "TW", "VN"], label="Target Country"),
-        gr.Textbox(label="API 金鑰（可選）", placeholder="如果需要，請輸入您的 API 金鑰")
+        gr.Textbox(label="API 金鑰（必填）", placeholder="請輸入您的 OpenAI API 金鑰")
     ],
     outputs=[
         gr.File(label="Download Translated Document"),