Make a bilingual version: Egglish and Chinese

app.py

@@ -48,16 +48,16 @@ def validate_json(output: str) -> tuple:
     try:
         json_match = re.search(r'\{[\s\S]*\}', output)
         if not json_match:
-            return False, None, "未找到JSON"
+            return False, None, "No JSON found"
         
         json_str = json_match.group(0)
         json_str = re.sub(r',\s*\}', '}', json_str)
         parsed = json.loads(json_str)
-        return True, parsed, "有效的JSON"
+        return True, parsed, "Valid JSON"
     except json.JSONDecodeError:
-        return False, None, "無效的JSON格式"
+        return False, None, "Invalid JSON format"
     except Exception:
-        return False, None, "解析JSON時出錯"
+        return False, None, "Error parsing JSON"
 
 def extract_reservation_info(text: str):
     """Extract reservation information from text"""
@@ -65,7 +65,7 @@ def extract_reservation_info(text: str):
     model, tokenizer = load_model()
     
     if model is None or tokenizer is None:
-        return {"error": "模型未加載成功，請刷新頁面重試"}, ""
+        return {"error": "Model not loaded, please refresh the page"}, ""
     
     try:
         # Create chat template
@@ -107,65 +107,157 @@ def extract_reservation_info(text: str):
             return {"error": message}, assistant_output
             
     except Exception as e:
-        return {"error": f"處理時出錯: {str(e)}"}, ""
+        return {"error": f"Processing error: {str(e)}"}, ""
 
 # Create Gradio interface
 def create_interface():
     """Create the Gradio interface"""
-    examples = [
+    chinese_examples = [
         "你好，我想訂明天晚上7點的位子，四位成人，電話是0912-345-678",
         "週六下午三點，兩位，電話0987654321",
         "預約下週三中午12點半，5人用餐，聯絡電話0912345678",
         "我要訂位，3個人，今天下午6點"
     ]
     
+    english_examples = [
+        "Hello, I'd like to reserve a table for 4 people tomorrow at 7 PM, phone number is 0912-345-678",
+        "Saturday 3 PM, 2 people, phone 0987654321",
+        "Reservation for next Wednesday at 12:30 PM, 5 people, contact number 0912345678",
+        "I want to make a reservation, 3 people, today at 6 PM"
+    ]
+    
+    # Language-specific text dictionaries
+    text_en = {
+        "title": "🍽️ Restaurant Reservation Info Extractor",
+        "description": "Use AI to automatically extract reservation information from messages",
+        "input_label": "Input reservation message",
+        "input_placeholder": "e.g., Hello, I'd like to reserve a table for 4 people tomorrow at 7 PM, phone number is 0912-345-678",
+        "button_text": "Extract Information",
+        "json_label": "Extracted Result",
+        "raw_label": "Raw Output",
+        "instructions_title": "ℹ️ Instructions",
+        "instructions": """**Supported information:**
+- 👥 Number of people (num_people)
+- 📅 Reservation date/time (reservation_date)
+- 📞 Phone number (phone_num)
+
+**Notes:**
+- First-time model loading may take a few minutes
+- If you encounter errors, try refreshing the page
+- The model outputs results in JSON format""",
+        "footer": "Powered by [Together AI](https://together.ai) | Model: Luigi/gemma-3-270m-it-dinercall-ner",
+        "examples_title": "Examples",
+        "chinese_examples": "Chinese Examples",
+        "english_examples": "English Examples"
+    }
+    
+    text_zh = {
+        "title": "🍽️ 餐廳訂位資訊提取器",
+        "description": "使用AI從中文訊息中自動提取訂位資訊",
+        "input_label": "輸入訂位訊息",
+        "input_placeholder": "例如: 你好，我想訂明天晚上7點的位子，四位成人，電話是0912-345-678",
+        "button_text": "提取資訊",
+        "json_label": "提取結果",
+        "raw_label": "原始輸出",
+        "instructions_title": "ℹ️ 使用說明",
+        "instructions": """**支援提取的資訊:**
+- 👥 人數 (num_people)
+- 📅 預訂日期/時間 (reservation_date)
+- 📞 電話號碼 (phone_num)
+
+**注意事項:**
+- 首次加載模型可能需要幾分鐘時間
+- 如果遇到錯誤，請嘗試刷新頁面
+- 模型會輸出JSON格式的結果""",
+        "footer": "由 [Together AI](https://together.ai) 提供技術支持 | 模型: Luigi/gemma-3-270m-it-dinercall-ner",
+        "examples_title": "示例",
+        "chinese_examples": "中文示例",
+        "english_examples": "英文示例"
+    }
+    
+    def update_interface(language):
+        """Update interface based on selected language"""
+        texts = text_en if language == "English" else text_zh
+        return [
+            gr.Markdown.update(value=f"# {texts['title']}"),
+            gr.Markdown.update(value=texts['description']),
+            gr.Textbox.update(
+                label=texts['input_label'],
+                placeholder=texts['input_placeholder']
+            ),
+            gr.Button.update(value=texts['button_text']),
+            gr.JSON.update(label=texts['json_label']),
+            gr.Textbox.update(label=texts['raw_label']),
+            gr.Accordion.update(label=texts['instructions_title']),
+            gr.Markdown.update(value=texts['instructions']),
+            gr.Markdown.update(value=texts['footer']),
+            gr.Markdown.update(value=f"### {texts['examples_title']}"),
+            gr.Markdown.update(value=f"### {texts['chinese_examples']}"),
+            gr.Markdown.update(value=f"### {texts['english_examples']}")
+        ]
+    
     with gr.Blocks(
-        title="🍽️ 餐廳訂位資訊提取器",
+        title="Restaurant Reservation Info Extractor",
         theme=gr.themes.Soft()
     ) as demo:
-        gr.Markdown("# 🍽️ 餐廳訂位資訊提取器")
-        gr.Markdown("使用AI從中文訊息中自動提取訂位資訊")
+        # Language selector
+        language = gr.Radio(
+            choices=["English", "中文"],
+            value="English",
+            label="Language / 语言",
+            interactive=True
+        )
+        
+        # Main interface components
+        title_md = gr.Markdown("# Restaurant Reservation Info Extractor")
+        description_md = gr.Markdown("Use AI to automatically extract reservation information from messages")
         
         with gr.Row():
             with gr.Column():
                 input_text = gr.Textbox(
-                    label="輸入訂位訊息",
-                    placeholder="例如: 你好，我想訂明天晚上7點的位子，四位成人，電話是0912-345-678",
+                    label="Input reservation message",
+                    placeholder="e.g., Hello, I'd like to reserve a table for 4 people tomorrow at 7 PM, phone number is 0912-345-678",
                     lines=3
                 )
-                submit_btn = gr.Button("提取資訊", variant="primary")
+                submit_btn = gr.Button("Extract Information", variant="primary")
+                
+                examples_title_md = gr.Markdown("### Examples")
+                chinese_examples_title_md = gr.Markdown("### Chinese Examples")
+                gr.Examples(
+                    examples=chinese_examples,
+                    inputs=input_text,
+                    label="Chinese Examples"
+                )
                 
+                english_examples_title_md = gr.Markdown("### English Examples")
                 gr.Examples(
-                    examples=examples,
+                    examples=english_examples,
                     inputs=input_text,
-                    label="示例訊息"
+                    label="English Examples"
                 )
                 
             with gr.Column():
-                json_output = gr.JSON(label="提取結果")
+                json_output = gr.JSON(label="Extracted Result")
                 raw_output = gr.Textbox(
-                    label="原始輸出",
+                    label="Raw Output",
                     interactive=False,
                     lines=3
                 )
         
         # Info panel
-        with gr.Accordion("ℹ️ 使用說明", open=False):
-            gr.Markdown("""
-            **支援提取的資訊:**
-            - 👥 人數 (num_people)
-            - 📅 預訂日期/時間 (reservation_date)
-            - 📞 電話號碼 (phone_num)
-            
-            **注意事項:**
-            - 首次加載模型可能需要幾分鐘時間
-            - 如果遇到錯誤，請嘗試刷新頁面
-            - 模型會輸出JSON格式的結果
-            """)
+        with gr.Accordion("ℹ️ Instructions", open=False) as instructions_accordion:
+            instructions_md = gr.Markdown("""**Supported information:**
+- 👥 Number of people (num_people)
+- 📅 Reservation date/time (reservation_date)
+- 📞 Phone number (phone_num)
+
+**Notes:**
+- First-time model loading may take a few minutes
+- If you encounter errors, try refreshing the page
+- The model outputs results in JSON format""")
         
         # Footer
-        gr.Markdown("---")
-        gr.Markdown("由 [Together AI](https://together.ai) 提供技術支持 | 模型: Luigi/gemma-3-270m-it-dinercall-ner")
+        footer_md = gr.Markdown("Powered by [Together AI](https://together.ai) | Model: Luigi/gemma-3-270m-it-dinercall-ner")
         
         # Connect the function to the button
         submit_btn.click(
@@ -173,6 +265,26 @@ def create_interface():
             inputs=input_text,
             outputs=[json_output, raw_output]
         )
+        
+        # Connect language selector to update interface
+        language.change(
+            fn=update_interface,
+            inputs=language,
+            outputs=[
+                title_md,
+                description_md,
+                input_text,
+                submit_btn,
+                json_output,
+                raw_output,
+                instructions_accordion,
+                instructions_md,
+                footer_md,
+                examples_title_md,
+                chinese_examples_title_md,
+                english_examples_title_md
+            ]
+        )
     
     return demo
 
@@ -186,4 +298,4 @@ if __name__ == "__main__":
         server_name="0.0.0.0",
         server_port=7860,
         share=False
-    )
+    )