fix bug

app.py

@@ -3,65 +3,65 @@ import base64
 import gradio as gr
 from gradio_client import Client
 
-client = Client("https://s5k.cn/api/v1/studio/yangbaosong/Qwen_Turbo_MT/gradio/")
-source = "English"
-target = "Chinese"
+API_URL = "https://s5k.cn/api/v1/studio/yangbaosong/Qwen_Turbo_MT/gradio/"
+client = Client(API_URL)
+
 
 def pdf2base64(file):
+    """将PDF文件转换为base64编码的字符串。"""
     if file is None:
         return "从上传PDF文件开始..."
     else:
-        with open(file, "rb") as f:
-            encoded_pdf = base64.b64encode(f.read()).decode("utf-8")
-        return f"""
-                    <iframe src="data:application/pdf;base64,{encoded_pdf}"
-                            width="100%"
-                            height="800px">
-                    </iframe>
-                    """
-
-
-def change_source_language(sorce_language):
-    global source
-    if sorce_language == "英文":
-        source = "English"
-    elif sorce_language == "中文":
-        source = "Chinese"
-
-
-def change_target_language(target_language):
-    global target
-    if target_language == "中文":
-        target = "Chinese"
-    elif target_language == "英文":
-        target = "English"
-
-
-def translate_and_display(text, chat_history):
-    global source, target
-    result = client.predict(
-        text,  # str  in 'Input' Textbox component
-        source,  # Literal['Chinese', 'English', 'Japanese', 'Korean', 'Thai', 'French', 'German', 'Spanish', 'Arabic', 'Indonesian', 'Vietnamese', 'Portuguese']  in 'Source Language' Dropdown component
-        target,  # Literal['Chinese', 'English', 'Japanese', 'Korean', 'Thai', 'French', 'German', 'Spanish', 'Arabic', 'Indonesian', 'Vietnamese', 'Portuguese']  in 'Target Language' Dropdown component
-        "",
-        "",
-        "",
-        "",
-        "",
-        "",
-        api_name="/model_chat",
-    )
-    chat_history.append((text, result))
-    return "", chat_history
+        try:
+            with open(file, "rb") as f:
+                encoded_pdf = base64.b64encode(f.read()).decode("utf-8")
+            return f"""
+                        <iframe src="data:application/pdf;base64,{encoded_pdf}"
+                                width="100%"
+                                height="800px">
+                        </iframe>
+                        """
+        except Exception as e:
+            return f"PDF文件加载失败: {e}"
+
+
+def translate_text(text, source_language, target_language):
+    """调用翻译API进行翻译。"""
+    try:
+        result = client.predict(
+            text,
+            source_language,
+            target_language,
+            "",
+            "",
+            "",
+            "",
+            "",
+            "",
+            api_name="/model_chat",
+        )
+        return result
+    except Exception as e:
+        return f"翻译失败: {e}"
+
+
+def update_chat_history(text, translation, chat_history):
+    """更新聊天记录。"""
+    chat_history.append((text, translation))
+    return chat_history
 
 
 def clear_text(txt2translate, chatbox):
+    """清空文本框和聊天记录。"""
     return "", [], gr.update(visible=False)
 
 
 with gr.Blocks() as demo:
+    source_language = gr.State("English")
+    target_language = gr.State("Chinese")
+
     with gr.Row():
-        with gr.Column(scale=2):  # PDF栏占比更大
+        with gr.Column(scale=2):
             output_pdf = gr.HTML("从上传PDF文件开始...")
         with gr.Column():
             with gr.Tab(label="传递PDF文件"):
@@ -73,10 +73,10 @@ with gr.Blocks() as demo:
                 with gr.Row():
                     with gr.Column():
                         with gr.Row():
-                            sorce_language = gr.Dropdown(
+                            source_dropdown = gr.Dropdown(
                                 value="英文", choices=["英文", "中文"], label="源语言"
                             )
-                            target_language = gr.Dropdown(
+                            target_dropdown = gr.Dropdown(
                                 value="中文", choices=["中文", "英文"], label="目标语言"
                             )
                 with gr.Row():
@@ -84,7 +84,6 @@ with gr.Blocks() as demo:
                         label="输入",
                         placeholder="从PDF复制或输入文本以进行翻译",
                     )
-
                     with gr.Column():
                         with gr.Row():
                             translate_btn = gr.Button(value="翻译")
@@ -94,16 +93,32 @@ with gr.Blocks() as demo:
                         show_copy_button=True,
                     )
 
+    def update_source_language(value):
+        if value == "英文":
+            return "English"
+        elif value == "中文":
+            return "Chinese"
+
+    def update_target_language(value):
+        if value == "中文":
+            return "Chinese"
+        elif value == "英文":
+            return "English"
+
     input_pdf.change(pdf2base64, input_pdf, output_pdf)
     explorer.change(pdf2base64, explorer, output_pdf)
-    sorce_language.change(change_source_language, sorce_language)
-    target_language.change(change_target_language, target_language)
+    source_dropdown.change(update_source_language, source_dropdown, source_language)
+    target_dropdown.change(update_target_language, target_dropdown, target_language)
+
+    def translate_and_update(text, chat_history, source, target):
+        translation = translate_text(text, source, target)
+        return "", update_chat_history(text, translation, chat_history)
+
     translate_btn.click(
-        translate_and_display,
-        [txt2translate, chatbox],
+        translate_and_update,
+        [txt2translate, chatbox, source_language, target_language],
         [txt2translate, chatbox],
     )
     clear_btn.click(clear_text, [txt2translate, chatbox], [txt2translate, chatbox])
 
-
 demo.launch()