Update app.py

app.py

@@ -36,24 +36,65 @@ def mistral_inference(prompt, image_url):
     
     return result
 
-# Gradio インターフェース
+# 言語によるUIラベルの設定
+def get_labels(language):
+    labels = {
+        'en': {
+            'title': "Pixtral Model Image Description",
+            'text_prompt': "Text Prompt",
+            'image_url': "Image URL",
+            'output': "Model Output",
+            'image_display': "Input Image",
+            'submit': "Run Inference"
+        },
+        'zh': {
+            'title': "Pixtral模型图像描述",
+            'text_prompt': "文本提示",
+            'image_url': "图片网址",
+            'output': "模型输出",
+            'image_display': "输入图片",
+            'submit': "运行推理"
+        },
+        'jp': {
+            'title': "Pixtralモデルによる画像説明生成",
+            'text_prompt': "テキストプロンプト",
+            'image_url': "画像URL",
+            'output': "モデルの出力結果",
+            'image_display': "入力された画像",
+            'submit': "推論を実行"
+        }
+    }
+    return labels[language]
+
+# Gradioインターフェース
 def process_input(text, image_url):
     result = mistral_inference(text, image_url)
     return result, image_url
 
+def update_ui(language):
+    labels = get_labels(language)
+    return labels['title'], labels['text_prompt'], labels['image_url'], labels['output'], labels['image_display'], labels['submit']
+
 with gr.Blocks() as demo:
-    gr.Markdown("## Pixtralモデルによる画像説明生成")
+    language_choice = gr.Dropdown(choices=['en', 'zh', 'jp'], label="Select Language", value='en')
     
+    title = gr.Markdown("## Pixtral Model Image Description")
     with gr.Row():
-        text_input = gr.Textbox(label="テキストプロンプト", placeholder="例: Describe the image.")
-        image_input = gr.Textbox(label="画像URL", placeholder="例: https://example.com/image.png")
+        text_input = gr.Textbox(label="Text Prompt", placeholder="e.g. Describe the image.")
+        image_input = gr.Textbox(label="Image URL", placeholder="e.g. https://example.com/image.png")
     
-    result_output = gr.Textbox(label="モデルの出力結果", lines=8, max_lines=20)  # 高さを500ピクセルに相当するように調整
-    image_output = gr.Image(label="入力された画像", type="auto")  # 入力画像URLを表示するための場所
+    result_output = gr.Textbox(label="Model Output", lines=8, max_lines=20)  # 高さ500ピクセルに相当するように調整
+    image_output = gr.Image(label="Input Image", type="auto")  # 入力画像URLを表示するための場所
 
-    submit_button = gr.Button("推論を実行")
+    submit_button = gr.Button("Run Inference")
 
-    # ボタンをクリックすると、モデルの結果と画像を表示
     submit_button.click(process_input, inputs=[text_input, image_input], outputs=[result_output, image_output])
 
+    # 言語変更時にUIラベルを更新
+    language_choice.change(
+        fn=update_ui, 
+        inputs=[language_choice], 
+        outputs=[title, text_input, image_input, result_output, image_output, submit_button]
+    )
+
 demo.launch()