Update chatbot.py

chatbot.py

@@ -6,7 +6,7 @@ import traceback
 from dateutil import parser as date_parser
 
 # Local imports
-from supabase_utils import get_supabase_client
+from supabase_utils import get_supabase_client, get_tide_predictions
 
 # Attempt to import Gemini
 try:
@@ -72,31 +72,23 @@ def retrieve_context_from_db(intent: dict) -> str:
         return "질문에서 시간 정보를 찾을 수 없습니다."
 
     try:
-        start_time = date_parser.parse(start_time_str)
-        end_time = date_parser.parse(end_time_str)
+        # KST 시간을 UTC로 변환하여 쿼리
+        result_data = get_tide_predictions(station_id, start_time_str, end_time_str)
 
-        start_query_str = start_time.strftime('%Y-%m-%d %H:%M:%S')
-        end_query_str = end_time.strftime('%Y-%m-%d %H:%M:%S')
-        
-        result = supabase.table('tide_predictions')\
-            .select('*')\
-            .eq('station_id', station_id)\
-            .gte('predicted_at', start_query_str)\
-            .lte('predicted_at', end_query_str)\
-            .order('predicted_at')\
-            .execute()
-
-        if result.data:
+        if result_data:
             info_text = f"'{station_name}'의 '{start_time_str}'부터 '{end_time_str}'까지 조위 정보입니다.\n\n"
             
-            if len(result.data) > 10:
-                levels = [d['final_tide_level'] for d in result.data]
+            if len(result_data) > 10:
+                levels = [d['final_tide_level'] for d in result_data]
                 max_level = max(levels)
                 min_level = min(levels)
                 info_text += f"- 최고 조위: {max_level:.1f}cm\n- 최저 조위: {min_level:.1f}cm"
             else:
-                for d in result.data:
-                    time_kst = date_parser.parse(d['predicted_at']).strftime('%H:%M')
+                for d in result_data:
+                    # UTC 시간을 KST로 변환하여 표시
+                    utc_time = date_parser.parse(d['predicted_at']).replace(tzinfo=pytz.UTC)
+                    kst_time = utc_time.astimezone(pytz.timezone('Asia/Seoul'))
+                    time_kst = kst_time.strftime('%H:%M')
                     info_text += f"- {time_kst}: 최종 조위 {d['final_tide_level']:.1f}cm (잔차 {d['predicted_residual']:.1f}cm)\n"
             return info_text
         else: