Update app.py

app.py

@@ -58,13 +58,22 @@ def compare_models(prompt):
 
 # --- 4. 建立介面 ---
 with gr.Blocks(theme=gr.themes.Soft()) as demo:
-    gr.Markdown("#SFT Demo)")
-    gr.Markdown("展示 原始模型與微調後模型，針對同一問題的回答差異。")
+    gr.Markdown("#SFT Demo")
+    gr.Markdown("展示原始模型與微調後模型 針對同一問題的回答差異。")
     
     with gr.Row():
-        input_box = gr.Textbox(label="輸入測試問題", placeholder="例如：請推薦台灣的旅遊景點。", lines=2)
+        input_box = gr.Textbox(label="輸入測試問題 (Prompt)", placeholder="例如：請推薦台灣的旅遊景點。", lines=2)
         submit_btn = gr.Button("開始比對", variant="primary")
     
+    with gr.Row():
+        with gr.Column():
+            output_base = gr.Textbox(label="❌ 訓練前 (Base Model)", lines=10, interactive=False)
+            gr.Markdown("*特徵：容易答非所問、不斷重複問題、或像機器人一樣講廢話。*")
+        
+        with gr.Column():
+            output_chat = gr.Textbox(label="✅ 訓練後 (SFT Model)", lines=10, interactive=False)
+            gr.Markdown("*特徵：能理解指令、條理分明、回答符合人類邏輯。*")
+
     submit_btn.click(compare_models, inputs=input_box, outputs=[output_base, output_chat])
 
 # 啟動