Spaces:
Running
on
Zero
Running
on
Zero
Update breed_recommendation.py
Browse files- breed_recommendation.py +73 -15
breed_recommendation.py
CHANGED
|
@@ -1,5 +1,7 @@
|
|
| 1 |
import sqlite3
|
| 2 |
import gradio as gr
|
|
|
|
|
|
|
| 3 |
from dog_database import get_dog_description, dog_data
|
| 4 |
from breed_health_info import breed_health_info
|
| 5 |
from breed_noise_info import breed_noise_info
|
|
@@ -528,33 +530,89 @@ def create_recommendation_tab(UserPreferences, get_breed_recommendations, format
|
|
| 528 |
elem_id="recommendation-output"
|
| 529 |
)
|
| 530 |
|
| 531 |
-
def on_find_match_click(*args):
|
|
|
|
|
|
|
|
|
|
|
|
|
| 532 |
try:
|
| 533 |
-
#
|
| 534 |
-
|
| 535 |
<div style="
|
| 536 |
text-align: center;
|
| 537 |
padding: 20px;
|
| 538 |
background: linear-gradient(to right, rgba(66, 153, 225, 0.1), rgba(72, 187, 120, 0.1));
|
| 539 |
border-radius: 10px;
|
| 540 |
margin: 20px 0;
|
|
|
|
| 541 |
">
|
| 542 |
<div style="font-size: 40px; margin-bottom: 10px;">🐕</div>
|
| 543 |
-
<
|
| 544 |
color: #2D3748;
|
| 545 |
font-size: 1.2em;
|
| 546 |
-
margin:
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 547 |
">
|
| 548 |
-
|
| 549 |
-
|
| 550 |
-
|
| 551 |
-
|
| 552 |
-
|
| 553 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 554 |
</div>
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 555 |
""")
|
| 556 |
-
|
| 557 |
-
#
|
|
|
|
|
|
|
|
|
|
| 558 |
user_prefs = UserPreferences(
|
| 559 |
living_space=args[0],
|
| 560 |
yard_access=args[1],
|
|
@@ -604,13 +662,13 @@ def create_recommendation_tab(UserPreferences, get_breed_recommendations, format
|
|
| 604 |
)
|
| 605 |
|
| 606 |
# 返回最終結果
|
| 607 |
-
|
| 608 |
|
| 609 |
except Exception as e:
|
| 610 |
print(f"Error in find match: {str(e)}")
|
| 611 |
import traceback
|
| 612 |
print(traceback.format_exc())
|
| 613 |
-
|
| 614 |
<div style="
|
| 615 |
text-align: center;
|
| 616 |
padding: 20px;
|
|
|
|
| 1 |
import sqlite3
|
| 2 |
import gradio as gr
|
| 3 |
+
import asyncio
|
| 4 |
+
from typing import Generator
|
| 5 |
from dog_database import get_dog_description, dog_data
|
| 6 |
from breed_health_info import breed_health_info
|
| 7 |
from breed_noise_info import breed_noise_info
|
|
|
|
| 530 |
elem_id="recommendation-output"
|
| 531 |
)
|
| 532 |
|
| 533 |
+
def on_find_match_click(*args) -> Generator:
|
| 534 |
+
"""
|
| 535 |
+
處理推薦按鈕點擊事件的函數
|
| 536 |
+
使用生成器來處理中間狀態和最終結果
|
| 537 |
+
"""
|
| 538 |
try:
|
| 539 |
+
# 首先返回 loading 狀態
|
| 540 |
+
yield gr.HTML("""
|
| 541 |
<div style="
|
| 542 |
text-align: center;
|
| 543 |
padding: 20px;
|
| 544 |
background: linear-gradient(to right, rgba(66, 153, 225, 0.1), rgba(72, 187, 120, 0.1));
|
| 545 |
border-radius: 10px;
|
| 546 |
margin: 20px 0;
|
| 547 |
+
animation: fadeIn 0.5s ease-in-out;
|
| 548 |
">
|
| 549 |
<div style="font-size: 40px; margin-bottom: 10px;">🐕</div>
|
| 550 |
+
<div style="
|
| 551 |
color: #2D3748;
|
| 552 |
font-size: 1.2em;
|
| 553 |
+
margin-bottom: 15px;
|
| 554 |
+
font-weight: 500;
|
| 555 |
+
">Finding your perfect match...</div>
|
| 556 |
+
|
| 557 |
+
<!-- 進度指示動畫 -->
|
| 558 |
+
<div style="
|
| 559 |
+
display: flex;
|
| 560 |
+
justify-content: center;
|
| 561 |
+
gap: 8px;
|
| 562 |
+
margin: 15px 0;
|
| 563 |
">
|
| 564 |
+
<div class="paw-print" style="
|
| 565 |
+
width: 12px;
|
| 566 |
+
height: 12px;
|
| 567 |
+
background: #4299e1;
|
| 568 |
+
border-radius: 50%;
|
| 569 |
+
animation: bounce 0.6s infinite ease-in-out;
|
| 570 |
+
animation-delay: 0s;
|
| 571 |
+
"></div>
|
| 572 |
+
<div class="paw-print" style="
|
| 573 |
+
width: 12px;
|
| 574 |
+
height: 12px;
|
| 575 |
+
background: #4299e1;
|
| 576 |
+
border-radius: 50%;
|
| 577 |
+
animation: bounce 0.6s infinite ease-in-out;
|
| 578 |
+
animation-delay: 0.2s;
|
| 579 |
+
"></div>
|
| 580 |
+
<div class="paw-print" style="
|
| 581 |
+
width: 12px;
|
| 582 |
+
height: 12px;
|
| 583 |
+
background: #4299e1;
|
| 584 |
+
border-radius: 50%;
|
| 585 |
+
animation: bounce 0.6s infinite ease-in-out;
|
| 586 |
+
animation-delay: 0.4s;
|
| 587 |
+
"></div>
|
| 588 |
+
</div>
|
| 589 |
+
|
| 590 |
+
<!-- 提示訊息 -->
|
| 591 |
+
<div style="
|
| 592 |
+
color: #4A5568;
|
| 593 |
+
font-size: 0.9em;
|
| 594 |
+
font-style: italic;
|
| 595 |
+
margin-top: 10px;
|
| 596 |
+
">Analyzing your preferences and our database of good boys and girls...</div>
|
| 597 |
</div>
|
| 598 |
+
|
| 599 |
+
<style>
|
| 600 |
+
@keyframes bounce {
|
| 601 |
+
0%, 100% { transform: translateY(0); }
|
| 602 |
+
50% { transform: translateY(-10px); }
|
| 603 |
+
}
|
| 604 |
+
|
| 605 |
+
@keyframes fadeIn {
|
| 606 |
+
from { opacity: 0; }
|
| 607 |
+
to { opacity: 1; }
|
| 608 |
+
}
|
| 609 |
+
</style>
|
| 610 |
""")
|
| 611 |
+
|
| 612 |
+
# 添加短暫延遲使 loading 動畫可見
|
| 613 |
+
yield asyncio.sleep(1)
|
| 614 |
+
|
| 615 |
+
# 創建用戶偏好對象並獲取推薦
|
| 616 |
user_prefs = UserPreferences(
|
| 617 |
living_space=args[0],
|
| 618 |
yard_access=args[1],
|
|
|
|
| 662 |
)
|
| 663 |
|
| 664 |
# 返回最終結果
|
| 665 |
+
yield format_recommendation_html(recommendations, is_description_search=False)
|
| 666 |
|
| 667 |
except Exception as e:
|
| 668 |
print(f"Error in find match: {str(e)}")
|
| 669 |
import traceback
|
| 670 |
print(traceback.format_exc())
|
| 671 |
+
yield gr.HTML("""
|
| 672 |
<div style="
|
| 673 |
text-align: center;
|
| 674 |
padding: 20px;
|