Add Gradio app and requirements

app.py

@@ -0,0 +1,245 @@
+# app.py
+import gradio as gr
+from vllm import LLM, SamplingParams
+
+# --------------------------------------------------------------------------
+# 1. 配置和加载模型 (在应用启动时执行一次)
+# --------------------------------------------------------------------------
+
+# !! 重要 !! -> 将此处的 "your-org/your-algharb-model" 替换成你在 Hugging Face Hub 上的模型ID
+model_id = "/mnt/workspace/wanghao/model_saved/Marco-MT-WMT"
+
+# 为 vLLM 配置 GPU 数量
+tensor_parallel_size = 1
+
+print(f"正在加载模型: {model_id}...")
+try:
+    llm = LLM(model=model_id, tensor_parallel_size=tensor_parallel_size)
+    print("模型加载成功!")
+except Exception as e:
+    print(f"模型加载失败: {e}")
+    llm = None # 标记模型加载失败
+
+# 定义采样参数
+sampling_params = SamplingParams(
+    n=1,
+    temperature=0.001,
+    top_p=0.001,
+    max_tokens=512,
+)
+
+# 语言代码到全名的映射
+source_lang_name_map = {
+    "en": "english",
+    "ja": "japanese",
+    "cs": "czech",
+    "de": "german",
+}
+
+target_lang_name_map = {
+    "zh": "chinese",
+    "ko": "korean",
+    "ja": "japanese",
+    "ar": "arabic",
+    "cs": "czech",
+    "ru": "russian",
+    "uk": "ukraine",
+    "et": "estonian",
+    "bho": "bhojpuri",
+    "sr_latin": "serbian",
+    "de": "german",
+}
+# --------------------------------------------------------------------------
+# 2. 定义核心翻译函数
+# --------------------------------------------------------------------------
+def translate(source_text, source_lang_code, target_lang_code):
+    """
+    接收用户输入并返回翻译结果
+    """
+    if llm is None:
+        return "错误：模型未能成功加载，请检查 Space 日志。"
+    
+    # 简单的输入验证
+    if not source_text or not source_text.strip():
+        return ""
+
+    source_language_name = source_lang_name_map.get(source_lang_code, "the source language")
+    target_language_name = target_lang_name_map.get(target_lang_code, "the target language")
+
+    prompt = (
+        f"Human: Please translate the following text into {target_language_name}: \n"
+        f"{source_text}<|im_end|>\n"
+        f"Assistant:"
+    )
+    print(prompt)
+    outputs = llm.generate([prompt], sampling_params)
+    
+    generated_text = outputs[0].outputs[0].text.strip()
+    
+    return generated_text
+
+# --------------------------------------------------------------------------
+# 3. 创建并配置 Gradio 界面 (修改版)
+# --------------------------------------------------------------------------
+
+# <--- 修改 1: 定义自定义 CSS 样式 (高级背景 + 正常字体) --->
+css = """
+/* --- 1. 整体背景 (改为更高级的浅灰蓝渐变) --- */
+.gradio-container {
+    /* 新的背景: 一个更清晰、更现代的浅灰蓝色渐变 */
+    background: linear-gradient(135deg, #F5F7FA 0%, #E8EBEE 100%);
+    padding: 20px !important;
+    
+    /* 新增: 设置全局字体为"正常"的系统默认字体 */
+    font-family: -apple-system, BlinkMacSystemFont, "Segoe UI", Roboto, Helvetica, Arial, sans-serif, "Apple Color Emoji", "Segoe UI Emoji", "Segoe UI Symbol";
+}
+
+/* --- 2. 标题 (保持不变, 它将继承上面的新字体) --- */
+.app-title {
+    font-size: 32px;
+    font-weight: 600; 
+    text-align: center;
+    color: #333333; 
+    margin-bottom: 5px;
+    padding-top: 10px;
+}
+.app-subtitle {
+    text-align: center;
+    font-size: 18px;
+    color: #555555;
+    margin-bottom: 20px;
+}
+
+/* --- 3. "悬浮卡片" 效果 (保持不变, 新背景将使其更突出) --- */
+.gradio-group {
+    border-radius: 20px !important;
+    box-shadow: 0 10px 30px rgba(0,0,0,0.07) !important;
+    border: 0 !important;
+    background: white;
+}
+
+/* --- 4. 增大文本框 (保持不变) --- */
+.gradio-textbox {
+    min-height: 300px !important;
+}
+"""
+
+# <--- 修复: 在这里加回被删除的 choices 定义 ---
+source_lang_choices = [(name.capitalize(), code) for code, name in source_lang_name_map.items()]
+target_lang_choices = [(name.capitalize(), code) for code, name in target_lang_name_map.items()]
+
+
+# <--- 修改 2: 使用 gr.Blocks 并保持主题 --->
+with gr.Blocks(
+    theme=gr.themes.Soft(primary_hue="amber", secondary_hue="amber"), # 保持 "amber" 主题
+    css=css,
+) as demo:
+    
+    # --- 标题 ---
+    gr.HTML(f"""
+    <div class='app-title'>Marco-MT-Algharb</div>
+    <div class='app-subtitle'>Our Submission for the WMT25 General Translation Task</div>
+    """)
+    
+    # --- 翻译器主界面 (两栏布局) ---
+    with gr.Row(variant="panel", equal_height=True):
+        
+        # --- 左侧输入卡片 ---
+        with gr.Group():
+            source_lang_dd = gr.Dropdown(
+                choices=source_lang_choices,
+                value="en", 
+                label="源语言 (Source Language)"
+            )
+            source_text_tb = gr.Textbox(
+                lines=10, 
+                label="源文本 (Source Text)", 
+                placeholder="Enter text to translate here...",
+                elem_classes=["gradio-textbox"]
+            )
+
+        # --- 右侧输出卡片 ---
+        with gr.Group():
+            target_lang_dd = gr.Dropdown(
+                choices=target_lang_choices,
+                value="zh", 
+                label="目标语言 (Target Language)"
+            )
+            output_text_tb = gr.Textbox(
+                lines=10, 
+                label="翻译结果 (Translation)", 
+                interactive=False,
+                elem_classes=["gradio-textbox"]
+            )
+
+    # --- 按钮行 ---
+    with gr.Row():
+        clear_btn = gr.ClearButton(
+            value="清除 (Clear)",
+            components=[source_text_tb, output_text_tb, source_lang_dd, target_lang_dd]
+        )
+        submit_btn = gr.Button("翻译 (Submit)", variant="primary", scale=1)
+
+    # --- 示例 ---
+    example_list = [
+            ["The quick brown fox jumps over the lazy dog.", "en", "zh"],
+            ["The sunset painted the sky with brilliant shades of orange and purple.", "en", "ko"],
+            ["The ancient ruins stand as a silent testament to the rise and fall of a great civilization.", "en", "ja"],
+        ]
+    gr.Examples(
+        examples=example_list,
+        inputs=[source_text_tb, source_lang_dd, target_lang_dd]
+    )
+
+    # --- (新位置) 支持的语向卡片 ---
+    # <--- 修改 3: 此处HTML将自动继承新的全局字体 --->
+    gr.HTML(f"""
+    <div style="color: #444; font-size: 16px; margin-top: 30px; padding: 20px 25px; background-color: #FFFFFF; border-radius: 15px; max-width: 900px; margin-left: auto; margin-right: auto; box-shadow: 0 4px 20px rgba(0,0,0,0.05);">
+        
+        <h3 style="text-align: center; margin-top: 5px; margin-bottom: 20px; color: #444444; font-weight: 600;">Supported Language Pairs</h3>
+        
+        <div style="display: flex; justify-content: space-around; text-align: left; line-height: 1.8;">
+            
+            <div>
+                <strong>From English (en):</strong>
+                <ul style="list-style-type: '» '; margin: 5px 0 0 20px; padding: 0;">
+                    <li>en2zh</li>
+                    <li>en2ja</li>
+                    <li>en2ko</li>
+                    <li>en2ar</li>
+                    <li>en2et</li>
+                    <li>en2sr_latin</li>
+                    <li>en2ru</li>
+                    <li>en2uk</li>
+                    <li>en2cs</li>
+                    <li>en2bho</li>
+                </ul>
+            </div>
+            
+            <div style="margin-left: 20px;">
+                <strong>From Czech (cs):</strong>
+                <ul style="list-style-type: '» '; margin: 5px 0 15px 20px; padding: 0;">
+                    <li>cs2uk</li>
+                    <li>cs2de</li>
+                </ul>
+                
+                <strong>From Japanese (ja):</strong>
+                <ul style="list-style-type: '» '; margin: 5px 0 0 20px; padding: 0;">
+                    <li>ja2zh</li>
+                </ul>
+            </div>
+        </div>
+    </div>
+    """)
+
+    # --- 设定按钮的点击逻辑 ---
+    submit_btn.click(
+        fn=translate,
+        inputs=[source_text_tb, source_lang_dd, target_lang_dd],
+        outputs=[output_text_tb],
+        api_name="translate"
+    )
+
+# 启动应用
+if __name__ == "__main__":
+    demo.launch()

requirements.txt

@@ -0,0 +1,3 @@
+Transformers==4.55.0
+vllm==0.10.0
+gradio==5.49.1