Update ui.py

ui.py

@@ -1,42 +1,283 @@
 import gradio as gr
 import os
-from config import STATIONS
+import json
+from datetime import datetime
+from config import STATIONS, STATION_NAMES
 from supabase_utils import get_supabase_client
+from api_utils import (
+    api_get_tide_level,
+    api_get_tide_series,
+    api_get_extremes_info,
+    api_check_tide_alert,
+    api_compare_stations,
+    api_health_check
+)
 
 def create_ui(prediction_handler, chatbot_handler):
     """Gradio UI를 생성하고 반환합니다."""
+    
     with gr.Blocks(title="통합 조위 예측 시스템", theme=gr.themes.Soft()) as demo:
         gr.Markdown("# 🌊 통합 조위 예측 시스템 with Gemini")
-
+        
         # 연결 상태 표시
         client = get_supabase_client()
         supabase_status = "🟢 연결됨" if client else "🔴 연결 안됨 (환경변수 확인 필요)"
         gemini_status = "🟢 연결됨" if os.getenv("GEMINI_API_KEY") else "🔴 연결 안됨 (환경변수 확인 필요)"
         gr.Markdown(f"**Supabase 상태**: {supabase_status} | **Gemini 상태**: {gemini_status}")
-
+        
         with gr.Tabs():
-            with gr.TabItem("통합 조위 예측"):
+            # 1. 예측 탭
+            with gr.TabItem("🔮 통합 조위 예측"):
+                gr.Markdown("""
+                ### TimeXer 모델을 활용한 조위 예측
+                - 과거 기상 데이터를 업로드하여 미래 72시간 조위 예측
+                - 잔차 예측 + 조화 예측 = 최종 조위
+                """)
+                
                 with gr.Row():
                     with gr.Column(scale=1):
-                        station_id_input = gr.Dropdown(STATIONS, label="관측소 선택", value="DT_0001")
+                        station_id_input = gr.Dropdown(
+                            choices=[(f"{STATION_NAMES[s]} ({s})", s) for s in STATIONS],
+                            label="관측소 선택",
+                            value="DT_0001"
+                        )
                         input_csv = gr.File(label="과거 데이터 업로드 (.csv)")
-                        predict_btn = gr.Button("예측 실행", variant="primary")
+                        predict_btn = gr.Button("🚀 예측 실행", variant="primary")
+                    
                     with gr.Column(scale=3):
                         output_plot = gr.Plot(label="예측 결과 시각화")
                         output_df = gr.DataFrame(label="예측 결과 데이터")
+                
                 output_log = gr.Textbox(label="실행 로그", lines=5, interactive=False)
-
-            with gr.TabItem("AI 조위 챗봇"):
+            
+            # 2. 챗봇 탭
+            with gr.TabItem("💬 AI 조위 챗봇"):
                 gr.ChatInterface(
                     fn=chatbot_handler,
-                    title="AI 조위 챗봇",
-                    description="조위에 대해 궁금한 점을 물어보세요. (예: '인천 오늘 현재 조위 알려줘')",
-                    #examples=[]
+                    title="",
+                    description="조위에 대해 궁금한 점을 물어보세요.",
+                    examples=[
+                        "인천 현재 조위 알려줘",
+                        "내일 오후 3시 평택 조위는?",
+                        "오늘 만조 시간은?"
+                    ]
                 )
-
+            
+            # 3. API 탭 (새로 추가)
+            with gr.TabItem("🔌 API"):
+                gr.Markdown("""
+                ## RESTful API 엔드포인트
+                실무에서 바로 사용 가능한 API 기능을 제공합니다.
+                """)
+                
+                with gr.Tabs():
+                    # 3-1. 현재 조위
+                    with gr.TabItem("현재 조위"):
+                        with gr.Row():
+                            with gr.Column():
+                                gr.Markdown("### 특정 시간 조위 조회")
+                                api1_station = gr.Dropdown(
+                                    choices=[(f"{STATION_NAMES[s]} ({s})", s) for s in STATIONS],
+                                    label="관측소",
+                                    value="DT_0001"
+                                )
+                                api1_time = gr.Textbox(
+                                    label="조회 시간 (비워두면 현재)",
+                                    placeholder="2024-12-25T15:00:00"
+                                )
+                                api1_fallback = gr.Checkbox(
+                                    label="조화 예측 폴백 사용",
+                                    value=True
+                                )
+                                api1_btn = gr.Button("조회")
+                            
+                            with gr.Column():
+                                api1_output = gr.JSON(label="API 응답")
+                        
+                        api1_btn.click(
+                            fn=lambda s, t, f: api_get_tide_level(s, t if t else None, f),
+                            inputs=[api1_station, api1_time, api1_fallback],
+                            outputs=api1_output,
+                            api_name="tide_level"
+                        )
+                    
+                    # 3-2. 시계열 데이터
+                    with gr.TabItem("시계열"):
+                        with gr.Row():
+                            with gr.Column():
+                                gr.Markdown("### 시간대별 조위 데이터")
+                                api2_station = gr.Dropdown(
+                                    choices=[(f"{STATION_NAMES[s]} ({s})", s) for s in STATIONS],
+                                    label="관측소",
+                                    value="DT_0001"
+                                )
+                                api2_start = gr.Textbox(
+                                    label="시작 시간",
+                                    placeholder="2024-12-25T00:00:00"
+                                )
+                                api2_end = gr.Textbox(
+                                    label="종료 시간",
+                                    placeholder="2024-12-26T00:00:00"
+                                )
+                                api2_interval = gr.Number(
+                                    label="간격(분)",
+                                    value=60,
+                                    minimum=5,
+                                    maximum=360
+                                )
+                                api2_btn = gr.Button("조회")
+                            
+                            with gr.Column():
+                                api2_output = gr.JSON(label="API 응답 (공공 API 형식)")
+                        
+                        api2_btn.click(
+                            fn=lambda s, st, et, i: api_get_tide_series(
+                                s, st if st else None, et if et else None, int(i)
+                            ),
+                            inputs=[api2_station, api2_start, api2_end, api2_interval],
+                            outputs=api2_output,
+                            api_name="tide_series"
+                        )
+                    
+                    # 3-3. 만조/간조
+                    with gr.TabItem("만조/간조"):
+                        with gr.Row():
+                            with gr.Column():
+                                gr.Markdown("### 만조/간조 정보")
+                                api3_station = gr.Dropdown(
+                                    choices=[(f"{STATION_NAMES[s]} ({s})", s) for s in STATIONS],
+                                    label="관측소",
+                                    value="DT_0001"
+                                )
+                                api3_date = gr.Textbox(
+                                    label="날짜 (YYYY-MM-DD)",
+                                    placeholder=datetime.now().strftime("%Y-%m-%d")
+                                )
+                                api3_secondary = gr.Checkbox(
+                                    label="부차 만조/간조 포함",
+                                    value=False
+                                )
+                                api3_btn = gr.Button("조회")
+                            
+                            with gr.Column():
+                                api3_output = gr.JSON(label="만조/간조 정보")
+                        
+                        api3_btn.click(
+                            fn=lambda s, d, sec: api_get_extremes_info(
+                                s, d if d else None, sec
+                            ),
+                            inputs=[api3_station, api3_date, api3_secondary],
+                            outputs=api3_output,
+                            api_name="extremes"
+                        )
+                    
+                    # 3-4. 위험 수위
+                    with gr.TabItem("위험 알림"):
+                        with gr.Row():
+                            with gr.Column():
+                                gr.Markdown("### 위험 수위 체크")
+                                api4_station = gr.Dropdown(
+                                    choices=[(f"{STATION_NAMES[s]} ({s})", s) for s in STATIONS],
+                                    label="관측소",
+                                    value="DT_0001"
+                                )
+                                api4_hours = gr.Number(
+                                    label="확인 시간(시간)",
+                                    value=24,
+                                    minimum=1,
+                                    maximum=72
+                                )
+                                api4_warning = gr.Number(
+                                    label="주의 수위(cm)",
+                                    value=700
+                                )
+                                api4_danger = gr.Number(
+                                    label="경고 수위(cm)",
+                                    value=750
+                                )
+                                api4_btn = gr.Button("체크")
+                            
+                            with gr.Column():
+                                api4_output = gr.JSON(label="위험 수위 정보")
+                        
+                        api4_btn.click(
+                            fn=lambda s, h, w, d: api_check_tide_alert(s, int(h), w, d),
+                            inputs=[api4_station, api4_hours, api4_warning, api4_danger],
+                            outputs=api4_output,
+                            api_name="alert"
+                        )
+                    
+                    # 3-5. 관측소 비교
+                    with gr.TabItem("비교"):
+                        with gr.Row():
+                            with gr.Column():
+                                gr.Markdown("### 다중 관측소 비교")
+                                api5_stations = gr.CheckboxGroup(
+                                    choices=[(f"{STATION_NAMES[s]}", s) for s in STATIONS],
+                                    label="비교할 관측소 선택",
+                                    value=["DT_0001", "DT_0002", "DT_0003"]
+                                )
+                                api5_time = gr.Textbox(
+                                    label="비교 시간 (비워두면 현재)",
+                                    placeholder="2024-12-25T15:00:00"
+                                )
+                                api5_btn = gr.Button("비교")
+                            
+                            with gr.Column():
+                                api5_output = gr.JSON(label="비교 결과")
+                        
+                        api5_btn.click(
+                            fn=lambda s, t: api_compare_stations(s, t if t else None),
+                            inputs=[api5_stations, api5_time],
+                            outputs=api5_output,
+                            api_name="compare"
+                        )
+                    
+                    # 3-6. 상태 체크
+                    with gr.TabItem("상태"):
+                        gr.Markdown("### API 및 시스템 상태")
+                        api6_btn = gr.Button("🔍 상태 확인", variant="secondary")
+                        api6_output = gr.JSON(label="시스템 상태")
+                        
+                        api6_btn.click(
+                            fn=api_health_check,
+                            inputs=[],
+                            outputs=api6_output,
+                            api_name="health"
+                        )
+                
+                # API 사용 안내
+                gr.Markdown("""
+                ---
+                ### 📚 API 사용 안내
+                
+                **Python 예제:**
+                ```python
+                from gradio_client import Client
+                
+                client = Client("https://your-space.hf.space/")
+                
+                # 현재 조위 조회
+                result = client.predict(
+                    "DT_0001",  # station_id
+                    "",         # time (empty = current)
+                    True,       # use_fallback
+                    api_name="/tide_level"
+                )
+                ```
+                
+                **특징:**
+                - 🔄 **조화 예측 폴백**: 최종 예측이 없을 때 자동으로 조화 예측 사용
+                - 📊 **공공 API 호환**: 기상청 API와 유사한 응답 형식
+                - ⚠️ **위험 수위 알림**: 2단계 경보 시스템
+                - 🏢 **다중 관측소**: 여러 관측소 동시 비교
+                """)
+        
+        # 이벤트 핸들러 연결
         predict_btn.click(
             fn=prediction_handler,
             inputs=[station_id_input, input_csv],
             outputs=[output_plot, output_df, output_log]
         )
+    
     return demo