Update app.py

app.py

@@ -1,6 +1,6 @@
 # -*- coding: utf-8 -*-
 """
-Road Buddy - Vietnamese Traffic Assistant
+Road Buddy - Traffic Assistant
 Powered by Vision-Language Model with LoRA adapter
 Analyzes road situations, explains traffic signs, and provides transportation law advice
 """
@@ -20,23 +20,15 @@ from peft import PeftModel
 BASE_MODEL = "unsloth/Qwen3-VL-4B-Instruct"
 ADAPTER_NAME = "hanguyen1907/Qwen3-4B-SFT"
 
-SYSTEM_PROMPT = """Bạn là "Road Buddy" - trợ lý giao thông thông minh chuyên về:
-1. Phân tích tình huống giao thông trên đường
-2. Nhận diện và giải thích biển báo giao thông
-3. Tư vấn luật giao thông đường bộ Việt Nam
-4. Hướng dẫn xử lý các tình huống trên đường
-
-Hãy trả lời chi tiết, dễ hiểu và đưa ra lời khuyên an toàn cho người dùng.
-Nếu nhìn thấy hình ảnh/video về giao thông, hãy phân tích kỹ và đưa ra nhận xét.
-
-You are "Road Buddy" - an intelligent traffic assistant specializing in:
+SYSTEM_PROMPT = """You are "Road Buddy" - an intelligent traffic assistant specializing in:
 1. Analyzing traffic situations on the road
 2. Recognizing and explaining traffic signs
-3. Advising on Vietnamese road traffic laws
-4. Guiding users through road situations
+3. Advising on road traffic laws and regulations
+4. Guiding users through various road situations
 
-Provide detailed, easy-to-understand answers and safe advice.
-If you see traffic images/videos, analyze carefully and provide observations."""
+Provide detailed, easy-to-understand answers and safe driving advice.
+If you see traffic images or videos, analyze them carefully and provide observations.
+Always prioritize safety in your recommendations."""
 
 # ====================== Global Model ======================
 MODEL = None
@@ -288,7 +280,7 @@ def process_video_chat(
         
         # Default message if empty
         if not message.strip():
-            message = "Hãy phân tích tình huống giao thông trong video này và đưa ra nhận xét."
+            message = "Analyze the traffic situation in this video and provide your observations."
         
         # Convert history
         history_messages = []
@@ -342,7 +334,7 @@ def process_image_chat(
         
         # Default message if empty
         if not message.strip():
-            message = "Hãy phân tích hình ảnh giao thông này và đưa ra nhận xét."
+            message = "Analyze this traffic image and provide your observations."
         
         # Convert history
         history_messages = []
@@ -379,20 +371,20 @@ def clear_chat():
 # ====================== Gradio Interface ======================
 
 EXAMPLE_QUESTIONS = [
-    "Biển báo này có ý nghĩa gì?",
     "What does this traffic sign mean?",
-    "Tình huống này tôi nên xử lý như thế nào?",
-    "Ai có quyền ưu tiên trong tình huống này?",
-    "Đây có phải là vi phạm luật giao thông không?",
-    "Tốc độ tối đa cho phép trên đường này là bao nhiêu?",
-    "Luật giao thông quy định gì về việc vượt xe?",
-    "Khi nào được phép quay đầu xe?",
+    "How should I handle this situation?",
+    "Who has the right of way here?",
+    "Is this a traffic violation?",
+    "What's the speed limit on this road?",
+    "What are the rules for overtaking?",
+    "When is it legal to make a U-turn?",
+    "What should I do at this intersection?",
 ]
 
 
 def create_demo():
     with gr.Blocks(
-        title="Road Buddy - Vietnamese Traffic Assistant",
+        title="Road Buddy - Traffic Assistant",
         theme=gr.themes.Soft(),
         css="""
         .chat-container { min-height: 400px; }
@@ -401,13 +393,13 @@ def create_demo():
     ) as demo:
         
         gr.Markdown("""
-        # 🚗 Road Buddy - Trợ Lý Giao Thông Thông Minh
+        # 🚗 Road Buddy - Your Intelligent Traffic Assistant
         
-        **Chào mừng bạn!** Road Buddy có thể giúp bạn:
-        - 🚦 **Phân tích tình huống giao thông** từ video/hình ảnh
-        - 🪧 **Nhận diện & giải thích biển báo** giao thông
-        - 📜 **Tư vấn luật giao thông** đường bộ Việt Nam
-        - 💡 **Hướng dẫn xử lý** các tình huống trên đường
+        **Welcome!** Road Buddy can help you with:
+        - 🚦 **Analyzing traffic situations** from videos and images
+        - 🪧 **Recognizing & explaining traffic signs**
+        - 📜 **Advising on traffic laws** and regulations
+        - 💡 **Guiding you through** various road situations
         
         ---
         """)
@@ -418,7 +410,7 @@ def create_demo():
         with gr.Tabs():
             # =================== TEXT CHAT TAB ===================
             with gr.TabItem("💬 Chat"):
-                gr.Markdown("### Hỏi về luật giao thông, quy tắc đường bộ...")
+                gr.Markdown("### Ask about traffic laws, road rules, driving tips...")
                 
                 chatbot = gr.Chatbot(
                     label="Road Buddy Chat",
@@ -428,14 +420,14 @@ def create_demo():
                 
                 with gr.Row():
                     msg_input = gr.Textbox(
-                        label="Tin nhắn của bạn",
-                        placeholder="Hỏi về luật giao thông, cách xử lý tình huống...",
+                        label="Your Message",
+                        placeholder="Ask about traffic laws, how to handle situations...",
                         lines=2,
                         scale=4,
                     )
-                    send_btn = gr.Button("Gửi 📤", variant="primary", scale=1)
+                    send_btn = gr.Button("Send 📤", variant="primary", scale=1)
                 
-                gr.Markdown("**Câu hỏi mẫu:**")
+                gr.Markdown("**Example Questions:**")
                 with gr.Row():
                     for q in EXAMPLE_QUESTIONS[:4]:
                         gr.Button(q, size="sm").click(
@@ -448,10 +440,10 @@ def create_demo():
                         )
                 
                 with gr.Accordion("⚙️ Settings", open=False):
-                    max_tokens_chat = gr.Slider(128, 1024, value=512, step=64, label="Max tokens")
+                    max_tokens_chat = gr.Slider(128, 1024, value=512, step=64, label="Max Tokens")
                     temp_chat = gr.Slider(0.0, 1.0, value=0.7, step=0.1, label="Temperature")
                 
-                clear_btn = gr.Button("🗑️ Xóa cuộc trò chuyện")
+                clear_btn = gr.Button("🗑️ Clear Conversation")
                 
                 # Chat handlers
                 send_btn.click(
@@ -467,8 +459,8 @@ def create_demo():
                 clear_btn.click(clear_chat, outputs=[chatbot, chat_state])
             
             # =================== VIDEO TAB ===================
-            with gr.TabItem("📹 Phân Tích Video"):
-                gr.Markdown("### Upload video giao thông để phân tích")
+            with gr.TabItem("📹 Video Analysis"):
+                gr.Markdown("### Upload a traffic video for analysis")
                 
                 chatbot_v = gr.Chatbot(label="Road Buddy", height=350)
                 
@@ -477,29 +469,29 @@ def create_demo():
                         video_input = gr.Video(label="Upload Video")
                         num_frames = gr.Slider(
                             minimum=1, maximum=8, value=4, step=1,
-                            label="Số frame trích xuất"
+                            label="Number of Frames to Extract"
                         )
                         frame_gallery = gr.Gallery(
-                            label="Frames trích xuất", 
+                            label="Extracted Frames", 
                             columns=4, 
                             height=150
                         )
                     
                     with gr.Column(scale=1):
                         question_v = gr.Textbox(
-                            label="Câu hỏi về video",
-                            placeholder="VD: Phân tích tình huống này...",
+                            label="Question about the video",
+                            placeholder="E.g., Analyze this traffic situation...",
                             lines=3,
                         )
                         with gr.Accordion("⚙️ Settings", open=False):
-                            max_tokens_v = gr.Slider(128, 1024, value=512, step=64, label="Max tokens")
+                            max_tokens_v = gr.Slider(128, 1024, value=512, step=64, label="Max Tokens")
                             temp_v = gr.Slider(0.0, 1.0, value=0.7, step=0.1, label="Temperature")
                         
                         status_v = gr.Markdown("")
                 
                 with gr.Row():
-                    analyze_btn_v = gr.Button("🔍 Phân Tích Video", variant="primary")
-                    clear_btn_v = gr.Button("🗑️ Xóa")
+                    analyze_btn_v = gr.Button("🔍 Analyze Video", variant="primary")
+                    clear_btn_v = gr.Button("🗑️ Clear")
                 
                 chat_state_v = gr.State([])
                 
@@ -514,36 +506,36 @@ def create_demo():
                 )
             
             # =================== IMAGE TAB ===================
-            with gr.TabItem("🖼️ Phân Tích Hình Ảnh"):
-                gr.Markdown("### Upload hình ảnh giao thông (biển báo, tình huống đường...)")
+            with gr.TabItem("🖼️ Image Analysis"):
+                gr.Markdown("### Upload traffic images (signs, road situations, etc.)")
                 
                 chatbot_i = gr.Chatbot(label="Road Buddy", height=350)
                 
                 with gr.Row():
                     with gr.Column(scale=1):
-                        gr.Markdown("**Upload 1-4 hình ảnh:**")
+                        gr.Markdown("**Upload 1-4 images:**")
                         with gr.Row():
-                            img1 = gr.Image(label="Ảnh 1", type="pil")
-                            img2 = gr.Image(label="Ảnh 2", type="pil")
+                            img1 = gr.Image(label="Image 1", type="pil")
+                            img2 = gr.Image(label="Image 2", type="pil")
                         with gr.Row():
-                            img3 = gr.Image(label="Ảnh 3", type="pil")
-                            img4 = gr.Image(label="Ảnh 4", type="pil")
+                            img3 = gr.Image(label="Image 3", type="pil")
+                            img4 = gr.Image(label="Image 4", type="pil")
                     
                     with gr.Column(scale=1):
                         question_i = gr.Textbox(
-                            label="Câu hỏi về hình ảnh",
-                            placeholder="VD: Biển báo này có ý nghĩa gì?",
+                            label="Question about the image(s)",
+                            placeholder="E.g., What does this sign mean?",
                             lines=3,
                         )
                         with gr.Accordion("⚙️ Settings", open=False):
-                            max_tokens_i = gr.Slider(128, 1024, value=512, step=64, label="Max tokens")
+                            max_tokens_i = gr.Slider(128, 1024, value=512, step=64, label="Max Tokens")
                             temp_i = gr.Slider(0.0, 1.0, value=0.7, step=0.1, label="Temperature")
                         
                         status_i = gr.Markdown("")
                 
                 with gr.Row():
-                    analyze_btn_i = gr.Button("🔍 Phân Tích Hình Ảnh", variant="primary")
-                    clear_btn_i = gr.Button("🗑️ Xóa")
+                    analyze_btn_i = gr.Button("🔍 Analyze Image", variant="primary")
+                    clear_btn_i = gr.Button("🗑️ Clear")
                 
                 chat_state_i = gr.State([])
                 
@@ -558,38 +550,53 @@ def create_demo():
                 )
             
             # =================== QUICK REFERENCE TAB ===================
-            with gr.TabItem("📚 Tra Cứu Nhanh"):
+            with gr.TabItem("📚 Quick Reference"):
                 gr.Markdown("""
-                ### Một số quy định giao thông cơ bản
+                ### Common Traffic Rules & Guidelines
+                
+                #### 🚦 Traffic Light Basics:
+                - **Red**: Stop completely behind the stop line
+                - **Yellow**: Prepare to stop (don't accelerate to beat it)
+                - **Green**: Proceed if the intersection is clear
+                - **Flashing Red**: Treat as a stop sign
+                - **Flashing Yellow**: Proceed with caution
                 
-                #### 🚗 Tốc độ tối đa trong khu vực đông dân cư:
-                - Xe máy, xe gắn máy: **40 km/h**
-                - Ô tô con, xe tải < 3.5 tấn: **50 km/h**
-                - Ô tô khách, xe tải > 3.5 tấn: **40 km/h**
+                #### 🪧 Common Sign Types:
+                - **Red Circle**: Prohibition (No entry, No parking, etc.)
+                - **Red Triangle**: Warning (Danger ahead)
+                - **Blue Circle**: Mandatory (Must follow)
+                - **Blue Rectangle**: Information
+                - **Green**: Direction and highway signs
                 
-                #### 🛣️ Tốc độ tối đa ngoài khu vực đông dân cư:
-                - Đường đôi có dải phân cách: **90 km/h** (ô tô con), **80 km/h** (xe máy)
-                - Đường 2 chiều không có dải phân cách: **80 km/h** (ô tô), **70 km/h** (xe máy)
+                #### 🚗 Right of Way Rules:
+                - Yield to vehicles already in the intersection
+                - At 4-way stops: First to arrive, first to go
+                - When arriving simultaneously: Yield to the right
+                - Always yield to emergency vehicles
+                - Pedestrians have right of way at crosswalks
                 
-                #### 🍺 Nồng độ cồn:
-                - **Xe máy**: 0 mg/lít khí thở hoặc 0 mg/100ml máu
-                - **Ô tô**: 0 mg/lít khí thở hoặc 0 mg/100ml máu
+                #### 🛣️ Lane Discipline:
+                - Keep right except when passing
+                - Use turn signals before changing lanes
+                - Check mirrors and blind spots
+                - Don't weave between lanes
                 
-                #### 📱 Sử dụng điện thoại:
-                - **Cấm** sử dụng điện thoại khi đang điều khiển phương tiện
-                - Có thể sử dụng thiết bị rảnh tay (hands-free)
+                #### 📱 Distracted Driving:
+                - No handheld phone use while driving
+                - Pull over safely if you need to use your phone
+                - Use hands-free devices if necessary
                 
                 ---
-                *Lưu ý: Đây chỉ là thông tin tham khảo. Hãy hỏi Road Buddy để được tư vấn chi tiết!*
+                *Note: Traffic laws vary by location. Always follow local regulations. Ask Road Buddy for specific advice!*
                 """)
         
         gr.Markdown(f"""
         ---
-        ### ℹ️ Thông tin
+        ### ℹ️ About
         - **Base Model:** [{BASE_MODEL}](https://huggingface.co/{BASE_MODEL})
         - **LoRA Adapter:** [{ADAPTER_NAME}](https://huggingface.co/{ADAPTER_NAME})
         
-        *Road Buddy cung cấp thông tin tham khảo. Vui lòng xác minh với nguồn chính thức khi cần thiết.*
+        *Road Buddy provides general information. Please verify with official sources when needed.*
         """)
     
     return demo