Update app.py

app.py

@@ -9,6 +9,7 @@ import torch.nn.functional as F
 from torchvision import transforms
 from PIL import Image, ImageDraw, ImageFont, ImageFilter
 from data_manager import get_dog_description, UserPreferences, get_breed_recommendations, format_recommendation_html
+from breed_health_info import breed_health_info, default_health_note
 from urllib.parse import quote
 from ultralytics import YOLO
 import asyncio
@@ -274,7 +275,7 @@ def create_breed_comparison(breed1: str, breed2: str) -> dict:
     """比較兩個狗品種的特性"""
     breed1_info = get_dog_description(breed1)
     breed2_info = get_dog_description(breed2)
-    
+
     # 標準化數值轉換
     value_mapping = {
         'Size': {'Small': 1, 'Medium': 2, 'Large': 3, 'Giant': 4},
@@ -282,12 +283,12 @@ def create_breed_comparison(breed1: str, breed2: str) -> dict:
         'Care_Level': {'Low': 1, 'Moderate': 2, 'High': 3},
         'Grooming_Needs': {'Low': 1, 'Moderate': 2, 'High': 3}
     }
-    
+
     comparison_data = {
         breed1: {},
         breed2: {}
     }
-    
+
     for breed, info in [(breed1, breed1_info), (breed2, breed2_info)]:
         comparison_data[breed] = {
             'Size': value_mapping['Size'].get(info['Size'], 2),  # 預設 Medium
@@ -297,7 +298,7 @@ def create_breed_comparison(breed1: str, breed2: str) -> dict:
             'Good_with_Children': info['Good with Children'] == 'Yes',
             'Original_Data': info
         }
-    
+
     return comparison_data
 
 
@@ -1260,6 +1261,90 @@ with gr.Blocks(css="""
         .bonus-score .progress {
             background: linear-gradient(90deg, #48bb78, #68d391);
         }
+
+        .health-section {
+            margin: 25px 0;
+            padding: 24px;
+            background-color: #f8f9fb;
+            border-radius: 12px;
+            border: 1px solid #e1e4e8;
+        }
+
+        .health-section .subsection-title {
+            font-size: 1.3em;
+            font-weight: 600;
+            margin-bottom: 20px;
+            display: flex;
+            align-items: center;
+            gap: 8px;
+            color: #2c3e50;
+        }
+
+        .health-info {
+            background-color: white;
+            padding: 24px;
+            border-radius: 8px;
+            margin: 15px 0;
+            border: 1px solid #e1e4e8;
+        }
+
+        .health-details {
+            font-size: 1.1rem;
+            line-height: 1.6;
+        }
+
+        .health-details h4 {
+            color: #2c3e50;
+            font-size: 1.15rem;
+            font-weight: 600;
+            margin: 20px 0 15px 0;
+        }
+
+        .health-details h4:first-child {
+            margin-top: 0;
+        }
+
+        .health-details ul {
+            list-style-type: none;
+            padding-left: 0;
+            margin: 0 0 25px 0;
+        }
+
+        .health-details ul li {
+            margin-bottom: 12px;
+            padding-left: 20px;
+            position: relative;
+        }
+
+        .health-details ul li:before {
+            content: "•";
+            position: absolute;
+            left: 0;
+            color: #2c3e50;
+        }
+
+        .health-disclaimer {
+            margin-top: 20px;
+            color: #666;
+            font-size: 1.05rem;
+            line-height: 1.6;
+            padding-top: 20px;
+            border-top: 1px solid #e1e4e8;
+        }
+
+        .health-disclaimer p {
+            margin: 8px 0;
+            padding-left: 15px;
+            position: relative;
+        }
+
+        .health-disclaimer p:before {
+            content: "»";
+            position: absolute;
+            left: 0;
+            color: #666;
+        }
+
         """) as iface:
 
         gr.HTML("""
@@ -1276,40 +1361,36 @@ with gr.Blocks(css="""
             </p>
         </header>
         """)
-        
+
         # 使用 Tabs 來分隔兩個功能
         with gr.Tabs():
             # 第一個 Tab：原有的辨識功能
             with gr.TabItem("Breed Detection"):
-                gr.HTML("<p style='text-align: center;'>Upload a picture of a dog, the model will predict its breed and provide detailed information!</p>")
+                gr.HTML("<p style='text-align: center;'>Upload a picture of a dog, and the model will predict its breed and provide detailed information!</p>")
                 gr.HTML("<p style='text-align: center; color: #666; font-size: 0.9em;'>Note: The model's predictions may not always be 100% accurate, and it is recommended to use the results as a reference.</p>")
-                
+
                 with gr.Row():
                     input_image = gr.Image(label="Upload a dog image", type="pil")
                     output_image = gr.Image(label="Annotated Image")
-                
+
                 output = gr.HTML(label="Prediction Results")
                 initial_state = gr.State()
-                
+
                 input_image.change(
                     predict,
                     inputs=input_image,
                     outputs=[output, output_image, initial_state]
                 )
-                
+
                 gr.Examples(
-                    examples=['Border_Collie.jpg', 
-                              'Golden_Retriever.jpeg', 
-                              'Saint_Bernard.jpeg',
-                              'Samoyed.jpg',
-                              'French_Bulldog.jpeg'],
+                    examples=['Border_Collie.jpg', 'Golden_Retriever.jpeg', 'Saint_Bernard.jpeg',  'French_Bulldog.jpeg', 'Samoyed.jpg'],
                     inputs=input_image
                 )
-            
+
             # 第二個 Tab：品種比較功能
             with gr.TabItem("Breed Comparison"):
                 gr.HTML("<p style='text-align: center;'>Select two dog breeds to compare their characteristics and care requirements.</p>")
-                
+
                 with gr.Row():
                     breed1_dropdown = gr.Dropdown(
                         choices=dog_breeds,
@@ -1321,17 +1402,17 @@ with gr.Blocks(css="""
                         label="Select Second Breed",
                         value="Border_Collie"
                     )
-                
+
                 compare_btn = gr.Button("Compare Breeds")
                 comparison_output = gr.HTML(label="Comparison Results")
-                
+
                 def show_comparison(breed1, breed2):
                     if not breed1 or not breed2:
                         return "Please select two breeds to compare"
-                    
+
                     breed1_info = get_dog_description(breed1)
                     breed2_info = get_dog_description(breed2)
-                    
+
                     html_output = f"""
                     <div class="dog-info-card">
                         <div class="comparison-grid" style="display: grid; grid-template-columns: 1fr 1fr; gap: 20px;">
@@ -1370,7 +1451,7 @@ with gr.Blocks(css="""
                                     </div>
                                 </div>
                             </div>
-                            
+
                             <div class="breed-info">
                                 <h2 class="section-title">
                                     <span class="icon">🐕</span> {breed2.replace('_', ' ')}
@@ -1410,17 +1491,17 @@ with gr.Blocks(css="""
                     </div>
                     """
                     return html_output
-                
+
                 compare_btn.click(
                     show_comparison,
                     inputs=[breed1_dropdown, breed2_dropdown],
                     outputs=comparison_output
                 )
-            
+
             # 第三個 Tab：品種推薦功能
             with gr.TabItem("Breed Recommendation"):
                 gr.HTML("<p style='text-align: center;'>Tell us about your lifestyle, and we'll recommend the perfect dog breeds for you!</p>")
-                
+
                 with gr.Row():
                     with gr.Column():
                         living_space = gr.Radio(
@@ -1429,7 +1510,7 @@ with gr.Blocks(css="""
                             info="Choose your current living situation",
                             value="apartment"
                         )
-                        
+
                         exercise_time = gr.Slider(
                             minimum=0,
                             maximum=180,
@@ -1437,14 +1518,14 @@ with gr.Blocks(css="""
                             label="Daily exercise time (minutes)",
                             info="Consider walks, play time, and training"
                         )
-                        
+
                         grooming_commitment = gr.Radio(
                             choices=["low", "medium", "high"],
                             label="Grooming commitment level",
                             info="Low: monthly, Medium: weekly, High: daily",
                             value="medium"
                         )
-                        
+
                     with gr.Column():
                         experience_level = gr.Radio(
                             choices=["beginner", "intermediate", "advanced"],
@@ -1452,20 +1533,20 @@ with gr.Blocks(css="""
                             info="Be honest - this helps find the right match",
                             value="beginner"
                         )
-                        
+
                         has_children = gr.Checkbox(
                             label="Have children at home",
                             info="Helps recommend child-friendly breeds"
                         )
-                        
+
                         noise_tolerance = gr.Radio(
                             choices=["low", "medium", "high"],
                             label="Noise tolerance level",
                             info="Some breeds are more vocal than others",
                             value="medium"
                         )
-                
-                
+
+
                 # 設置按鈕的點擊事件
                 get_recommendations_btn = gr.Button("Find My Perfect Match! 🔍", variant="primary")
                 recommendation_output = gr.HTML(label="Breed Recommendations")
@@ -1484,27 +1565,27 @@ with gr.Blocks(css="""
                             other_pets=False,
                             climate="moderate"
                         )
-                        
+
                         recommendations = get_breed_recommendations(user_prefs)
                         return format_recommendation_html(recommendations)
                     except Exception as e:
                         print(f"Error in process_recommendations: {str(e)}")
                         return f"An error occurred: {str(e)}"
-                
+
                 # 這行是關鍵 - 確保按鈕點擊事件有正確連接到處理函數
                 get_recommendations_btn.click(
                     fn=process_recommendations,  # 處理函數
                     inputs=[
                         living_space,
-                        exercise_time, 
+                        exercise_time,
                         grooming_commitment,
-                        experience_level, 
-                        has_children, 
+                        experience_level,
+                        has_children,
                         noise_tolerance
                     ],
                     outputs=recommendation_output  # 輸出結果的位置
                 )
-                        
+
         gr.HTML('For more details on this project and other work, feel free to visit my GitHub <a href="https://github.com/Eric-Chung-0511/Learning-Record/tree/main/Data%20Science%20Projects/Dog_Breed_Classifier">Dog Breed Classifier</a>')
 
 if __name__ == "__main__":