import sqlite3
import gradio as gr
import asyncio
from typing import Generator
from dog_database import get_dog_description, dog_data
from breed_health_info import breed_health_info
from breed_noise_info import breed_noise_info
from scoring_calculation_system import UserPreferences, calculate_compatibility_score
from recommendation_html_format import format_recommendation_html, get_breed_recommendations
from search_history import create_history_tab, create_history_component
# def create_recommendation_tab(UserPreferences, get_breed_recommendations, format_recommendation_html, history_component):
# with gr.TabItem("Breed Recommendation"):
# with gr.Tabs():
# with gr.Tab("Find by Criteria"):
# gr.HTML("""
#
#
#
BETA
#
#
# Tell us about your lifestyle, and we'll recommend the perfect dog breeds for you!
#
#
#
# 🔬
# Beta Feature: Our matching algorithm is continuously improving. Results are for reference only.
#
#
BETA
Tell us about your lifestyle, and we'll recommend the perfect dog breeds for you!
🔬
Beta Feature: Our matching algorithm is continuously improving. Results are for reference only.
""")
# 添加 Loading 狀態顯示
loading_html = gr.HTML("""
🐕
Sniffing out your perfect match...
Checking all the good boys and girls...
""", visible=False)
# 用戶輸入區域
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"
)
yard_access = gr.Radio(
choices=["no_yard", "shared_yard", "private_yard"],
label="Yard Access Type",
info="Available outdoor space",
value="no_yard"
)
exercise_time = gr.Slider(
minimum=0,
maximum=180,
value=60,
label="Daily exercise time (minutes)",
info="Consider walks, play time, and training"
)
exercise_type = gr.Radio(
choices=["light_walks", "moderate_activity", "active_training"],
label="Exercise Style",
info="What kind of activities do you prefer?",
value="moderate_activity"
)
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"
)
time_availability = gr.Radio(
choices=["limited", "moderate", "flexible"],
label="Time Availability",
info="Time available for dog care daily",
value="moderate"
)
has_children = gr.Checkbox(
label="Have children at home",
info="Helps recommend child-friendly breeds"
)
children_age = gr.Radio(
choices=["toddler", "school_age", "teenager"],
label="Children's Age Group",
info="Helps match with age-appropriate breeds",
visible=False
)
noise_tolerance = gr.Radio(
choices=["low", "medium", "high"],
label="Noise tolerance level",
info="Some breeds are more vocal than others",
value="medium"
)
# 控制 children_age 顯示邏輯
def update_children_age_visibility(has_children):
return gr.update(visible=has_children)
has_children.change(
fn=update_children_age_visibility,
inputs=has_children,
outputs=children_age
)
# 推薦按鈕
get_recommendations_btn = gr.Button(
"Find My Perfect Match! 🔍",
variant="primary"
)
# 推薦結果顯示區域
recommendation_output = gr.HTML(
label="Breed Recommendations",
visible=True,
elem_id="recommendation-output"
)
def on_find_match_click(*args) -> Generator:
"""
處理推薦按鈕點擊事件的函數
使用生成器來處理中間狀態和最終結果
"""
try:
# 首先返回 loading 狀態
yield gr.HTML("""
🐕
Finding your perfect match...
Analyzing your preferences and our database of good boys and girls...
""")
# 添加短暫延遲使 loading 動畫可見
yield asyncio.sleep(1)
# 創建用戶偏好對象並獲取推薦
user_prefs = UserPreferences(
living_space=args[0],
yard_access=args[1],
exercise_time=args[2],
exercise_type=args[3],
grooming_commitment=args[4],
experience_level=args[5],
time_availability=args[6],
has_children=args[7],
children_age=args[8] if args[7] else None,
noise_tolerance=args[9],
space_for_play=True if args[0] != "apartment" else False,
other_pets=False,
climate="moderate",
health_sensitivity="medium",
barking_acceptance=args[9]
)
# 獲取推薦結果
recommendations = get_breed_recommendations(user_prefs, top_n=10)
# 儲存搜尋歷史
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],
'yard_access': args[1],
'exercise_time': args[2],
'exercise_type': args[3],
'grooming_commitment': args[4],
'experience_level': args[5],
'time_availability': args[6],
'has_children': args[7],
'children_age': args[8] if args[7] else None,
'noise_tolerance': args[9],
'search_type': 'Criteria'
},
results=history_results
)
# 返回最終結果
yield format_recommendation_html(recommendations, is_description_search=False)
except Exception as e:
print(f"Error in find match: {str(e)}")
import traceback
print(traceback.format_exc())
yield gr.HTML("""
🐾 Oops! Something went wrong while finding your perfect match.
Please try again!
""")
# 設置按鈕點擊事件
get_recommendations_btn.click(
fn=on_find_match_click,
inputs=[
living_space,
yard_access,
exercise_time,
exercise_type,
grooming_commitment,
experience_level,
time_availability,
has_children,
children_age,
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,
}