Upload 5 files

breed_comparison.py

@@ -0,0 +1,125 @@
+import gradio as gr
+
+def create_comparison_tab(dog_breeds, get_dog_description):
+    """创建品种比较标签页
+    
+    Args:
+        dog_breeds: 狗品种列表
+        get_dog_description: 获取品种描述的函数
+    """
+    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,
+                label="Select First Breed",
+                value="Golden_Retriever"
+            )
+            breed2_dropdown = gr.Dropdown(
+                choices=dog_breeds,
+                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;">
+                    <div class="breed-info">
+                        <h2 class="section-title">
+                            <span class="icon">🐕</span> {breed1.replace('_', ' ')}
+                        </h2>
+                        <div class="info-section">
+                            <div class="info-item">
+                                <span class="tooltip">
+                                    <span class="icon">📏</span>
+                                    <span class="label">Size:</span>
+                                    <span class="value">{breed1_info['Size']}</span>
+                                </span>
+                            </div>
+                            <div class="info-item">
+                                <span class="tooltip">
+                                    <span class="icon">🏃</span>
+                                    <span class="label">Exercise Needs:</span>
+                                    <span class="value">{breed1_info['Exercise Needs']}</span>
+                                </span>
+                            </div>
+                            <div class="info-item">
+                                <span class="tooltip">
+                                    <span class="icon">✂️</span>
+                                    <span class="label">Grooming:</span>
+                                    <span class="value">{breed1_info['Grooming Needs']}</span>
+                                </span>
+                            </div>
+                            <div class="info-item">
+                                <span class="tooltip">
+                                    <span class="icon">👨‍👩‍👧‍👦</span>
+                                    <span class="label">Good with Children:</span>
+                                    <span class="value">{breed1_info['Good with Children']}</span>
+                                </span>
+                            </div>
+                        </div>
+                    </div>
+
+                    <div class="breed-info">
+                        <h2 class="section-title">
+                            <span class="icon">🐕</span> {breed2.replace('_', ' ')}
+                        </h2>
+                        <div class="info-section">
+                            <div class="info-item">
+                                <span class="tooltip">
+                                    <span class="icon">📏</span>
+                                    <span class="label">Size:</span>
+                                    <span class="value">{breed2_info['Size']}</span>
+                                </span>
+                            </div>
+                            <div class="info-item">
+                                <span class="tooltip">
+                                    <span class="icon">🏃</span>
+                                    <span class="label">Exercise Needs:</span>
+                                    <span class="value">{breed2_info['Exercise Needs']}</span>
+                                </span>
+                            </div>
+                            <div class="info-item">
+                                <span class="tooltip">
+                                    <span class="icon">✂️</span>
+                                    <span class="label">Grooming:</span>
+                                    <span class="value">{breed2_info['Grooming Needs']}</span>
+                                </span>
+                            </div>
+                            <div class="info-item">
+                                <span class="tooltip">
+                                    <span class="icon">👨‍👩‍👧‍👦</span>
+                                    <span class="label">Good with Children:</span>
+                                    <span class="value">{breed2_info['Good with Children']}</span>
+                                </span>
+                            </div>
+                        </div>
+                    </div>
+                </div>
+            </div>
+            """
+            return html_output
+
+        compare_btn.click(
+            show_comparison,
+            inputs=[breed1_dropdown, breed2_dropdown],
+            outputs=comparison_output
+        )
+
+    return {
+        'breed1_dropdown': breed1_dropdown,
+        'breed2_dropdown': breed2_dropdown,
+        'compare_btn': compare_btn,
+        'comparison_output': comparison_output
+    }

breed_detection.py

@@ -0,0 +1,38 @@
+import gradio as gr
+from PIL import Image
+
+def create_detection_tab(predict_fn, example_images):
+    """创建品种识别标签页
+    
+    Args:
+        predict_fn: 预测函数
+        example_images: 示例图片路径列表
+    """
+    with gr.TabItem("Breed Detection"):
+        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_fn,
+            inputs=input_image,
+            outputs=[output, output_image, initial_state]
+        )
+
+        gr.Examples(
+            examples=example_images,
+            inputs=input_image
+        )
+
+    return {
+        'input_image': input_image,
+        'output_image': output_image,
+        'output': output,
+        'initial_state': initial_state
+    }

breed_recommendation.py

@@ -0,0 +1,130 @@
+import gradio as gr
+
+def create_recommendation_tab(UserPreferences, get_breed_recommendations, format_recommendation_html, history_component):
+    """创建品种推荐标签页
+    
+    Args:
+        UserPreferences: 用户偏好类
+        get_breed_recommendations: 获取品种推荐的函数
+        format_recommendation_html: 格式化推荐结果的函数
+        history_component: 历史记录组件
+    """
+    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(
+                    choices=["apartment", "house_small", "house_large"],
+                    label="What type of living space do you have?",
+                    info="Choose your current living situation",
+                    value="apartment"
+                )
+
+                exercise_time = gr.Slider(
+                    minimum=0,
+                    maximum=180,
+                    value=60,
+                    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"],
+                    label="Dog ownership experience",
+                    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")
+
+        def on_find_match_click(*args):
+            try:
+                user_prefs = UserPreferences(
+                    living_space=args[0],
+                    exercise_time=args[1],
+                    grooming_commitment=args[2],
+                    experience_level=args[3],
+                    has_children=args[4],
+                    noise_tolerance=args[5],
+                    space_for_play=True if args[0] != "apartment" else False,
+                    other_pets=False,
+                    climate="moderate"
+                )
+
+                recommendations = get_breed_recommendations(user_prefs)
+                
+                history_results = [{
+                    'breed': rec['breed'],
+                    'rank': rec['rank'],
+                    'overall_score': rec['final_score'],
+                    'base_score': rec['base_score'],
+                    'bonus_score': rec['bonus_score'],
+                    'scores': rec['scores']
+                } for rec in recommendations]
+
+                # 保存到历史记录
+                history_component.save_search(
+                    user_preferences={
+                        'living_space': args[0],
+                        'exercise_time': args[1],
+                        'grooming_commitment': args[2],
+                        'experience_level': args[3],
+                        'has_children': args[4],
+                        'noise_tolerance': args[5]
+                    },
+                    results=history_results
+                )
+
+                return format_recommendation_html(recommendations)
+                
+            except Exception as e:
+                print(f"Error in find match: {str(e)}")
+                import traceback
+                print(traceback.format_exc())
+                return "Error getting recommendations"
+
+        get_recommendations_btn.click(
+            fn=on_find_match_click,
+            inputs=[
+                living_space,
+                exercise_time,
+                grooming_commitment,
+                experience_level,
+                has_children,
+                noise_tolerance
+            ],
+            outputs=recommendation_output
+        )
+
+    return {
+        'living_space': living_space,
+        'exercise_time': exercise_time,
+        'grooming_commitment': grooming_commitment,
+        'experience_level': experience_level,
+        'has_children': has_children,
+        'noise_tolerance': noise_tolerance,
+        'get_recommendations_btn': get_recommendations_btn,
+        'recommendation_output': recommendation_output
+    }

history_manager.py

@@ -0,0 +1,92 @@
+
+from datetime import datetime
+import json
+import os
+import pytz
+import traceback
+
+class UserHistoryManager:
+    def __init__(self):
+        """初始化歷史紀錄管理器"""
+        self.history_file = "user_history.json"
+        print(f"Initializing UserHistoryManager with file: {os.path.abspath(self.history_file)}")
+        self._init_file()
+
+    def _init_file(self):
+        """初始化JSON檔案"""
+        try:
+            if not os.path.exists(self.history_file):
+                print(f"Creating new history file: {self.history_file}")
+                with open(self.history_file, 'w', encoding='utf-8') as f:
+                    json.dump([], f)
+            else:
+                print(f"History file exists: {self.history_file}")
+                # 驗證檔案內容
+                with open(self.history_file, 'r', encoding='utf-8') as f:
+                    data = json.load(f)
+                    print(f"Current history entries: {len(data)}")
+        except Exception as e:
+            print(f"Error in _init_file: {str(e)}")
+            print(traceback.format_exc())
+
+
+    def save_history(self, user_preferences: dict, results: list) -> bool:
+        """儲存搜尋歷史，使用台北時間"""
+        try:
+            print("\nSaving history:")
+            print("Results to save:", results)
+            
+            # 使用 pytz 創建台北時區
+            taipei_tz = pytz.timezone('Asia/Taipei')
+            # 獲取當前時間並轉換為台北時間
+            taipei_time = datetime.now(taipei_tz)
+            
+            history_entry = {
+                "timestamp": taipei_time.strftime("%Y-%m-%d %H:%M:%S"),
+                "preferences": user_preferences,
+                "results": results
+            }
+            
+            with open(self.history_file, 'r', encoding='utf-8') as f:
+                history = json.load(f)
+            
+            # 添加新紀錄
+            history.append(history_entry)
+            
+            # 限制保存最近的20筆記錄
+            if len(history) > 20:
+                history = history[-20:]
+            
+            with open(self.history_file, 'w', encoding='utf-8') as f:
+                json.dump(history, f, ensure_ascii=False, indent=2)
+            
+            return True
+        except Exception as e:
+            print(f"Error saving history: {str(e)}")
+            return False
+
+    def get_history(self) -> list:
+        """獲取搜尋歷史"""
+        try:
+            print("Attempting to read history")  # Debug
+            with open(self.history_file, 'r', encoding='utf-8') as f:
+                data = json.load(f)
+                print(f"Read {len(data)} history entries")  # Debug
+                return data if isinstance(data, list) else []
+        except Exception as e:
+            print(f"Error reading history: {str(e)}")
+            print(traceback.format_exc())
+            return []
+
+    def clear_all_history(self) -> bool:
+        """清除所有歷史紀錄"""
+        try:
+            print("Attempting to clear all history")  # Debug
+            with open(self.history_file, 'w', encoding='utf-8') as f:
+                json.dump([], f)
+            print("History cleared successfully")  # Debug
+            return True
+        except Exception as e:
+            print(f"Error clearing history: {str(e)}")
+            print(traceback.format_exc())
+            return False

search_history.py

@@ -0,0 +1,158 @@
+import gradio as gr
+import traceback
+from typing import Optional, Dict, List
+from history_manager import UserHistoryManager
+
+class SearchHistoryComponent:
+    def __init__(self):
+        """初始化搜索歷史組件"""
+        self.history_manager = UserHistoryManager()
+        
+    def format_history_html(self, history_data: Optional[List[Dict]] = None) -> str:
+        """將歷史記錄格式化為HTML"""
+        try:
+            if history_data is None:
+                history_data = self.history_manager.get_history()
+                
+            if not history_data:
+                return """
+                <div style='text-align: center; padding: 20px; color: #666;'>
+                    No search history yet. Try making some breed recommendations!
+                </div>
+                """
+
+            html = "<div class='history-container'>"
+            
+            for entry in reversed(history_data):
+                timestamp = entry.get('timestamp', 'Unknown time')
+                prefs = entry.get('preferences', {})
+                results = entry.get('results', [])
+                
+                html += f"""
+                <div class="history-entry">
+                    <div class="history-header">
+                        <span class="timestamp">🕒 {timestamp}</span>
+                    </div>
+                    
+                    <div class="params-list">
+                        <h4>Search Parameters:</h4>
+                        <ul>
+                            <li><span class="param-label">Living Space:</span> {prefs.get('living_space', 'N/A')}</li>
+                            <li><span class="param-label">Exercise Time:</span> {prefs.get('exercise_time', 'N/A')} minutes</li>
+                            <li><span class="param-label">Grooming:</span> {prefs.get('grooming_commitment', 'N/A')}</li>
+                            <li><span class="param-label">Experience:</span> {prefs.get('experience_level', 'N/A')}</li>
+                            <li><span class="param-label">Children at Home:</span> {"Yes" if prefs.get('has_children') else "No"}</li>
+                            <li><span class="param-label">Noise Tolerance:</span> {prefs.get('noise_tolerance', 'N/A')}</li>
+                        </ul>
+                    </div>
+                    
+                    <div class="results-list">
+                        <h4>Top 5 Breed Matches:</h4>
+                        <div class="breed-list">
+                """
+                
+                if results:
+                    for i, result in enumerate(results[:5], 1):
+                        breed_name = result.get('breed', 'Unknown breed').replace('_', ' ')
+                        score = result.get('overall_score', result.get('final_score', 0))
+                        html += f"""
+                            <div class="breed-item">
+                                <div class="breed-info">
+                                    <span class="breed-rank">#{i}</span>
+                                    <span class="breed-name">{breed_name}</span>
+                                    <span class="breed-score">{score*100:.1f}%</span>
+                                </div>
+                            </div>
+                        """
+                
+                html += """
+                        </div>
+                    </div>
+                </div>
+                """
+            
+            html += "</div>"
+            return html
+            
+        except Exception as e:
+            print(f"Error formatting history: {str(e)}")
+            print(traceback.format_exc())
+            return f"""
+            <div style='text-align: center; padding: 20px; color: #dc2626;'>
+                Error formatting history. Please try refreshing the page.
+                <br>Error details: {str(e)}
+            </div>
+            """
+
+    def clear_history(self) -> str:
+        """清除所有搜尋歷史"""
+        try:
+            success = self.history_manager.clear_all_history()
+            print(f"Clear history result: {success}")
+            return self.format_history_html()
+        except Exception as e:
+            print(f"Error in clear_history: {str(e)}")
+            print(traceback.format_exc())
+            return "Error clearing history"
+
+    def refresh_history(self) -> str:
+        """刷新歷史記錄顯示"""
+        try:
+            return self.format_history_html()
+        except Exception as e:
+            print(f"Error in refresh_history: {str(e)}")
+            return "Error refreshing history"
+
+    def save_search(self, user_preferences: dict, results: list) -> bool:
+        """保存搜索結果"""
+        return self.history_manager.save_history(user_preferences, results)
+
+def create_history_component():
+    """只创建历史组件实例，不创建UI"""
+    return SearchHistoryComponent()
+
+def create_history_tab(history_component: SearchHistoryComponent):
+    """创建历史记录标签页
+    
+    Args:
+        history_component: 已创建的历史组件实例
+    """
+    with gr.TabItem("Recommendation Search History"):
+        gr.HTML("""
+        <div style='text-align: center; padding: 20px;'>
+            <h3 style='color: #2D3748; margin-bottom: 10px;'>Search History</h3>
+            <p style='color: #4A5568;'>View your previous breed recommendations and search preferences</p>
+        </div>
+        """)
+        
+        with gr.Row():
+            with gr.Column(scale=4):
+                history_display = gr.HTML()
+                
+                with gr.Row():
+                    with gr.Column(scale=1):
+                        clear_history_btn = gr.Button(
+                            "🗑️ Clear History", 
+                            variant="secondary",
+                            size="sm"
+                        )
+                    with gr.Column(scale=1):
+                        refresh_btn = gr.Button(
+                            "🔄 Refresh", 
+                            variant="secondary",
+                            size="sm"
+                        )
+
+                history_display.value = history_component.format_history_html()
+
+                clear_history_btn.click(
+                    fn=history_component.clear_history,
+                    outputs=[history_display],
+                    api_name="clear_history"
+                )
+
+                refresh_btn.click(
+                    fn=history_component.refresh_history,
+                    outputs=[history_display],
+                    api_name="refresh_history"
+                )