app.py

import gradio as gr
import requests
import logging
import threading
import time

# 로깅 설정
logging.basicConfig(level=logging.INFO)

css = """
footer { visibility: hidden; }
.status-button { 
    display: inline-block;
    padding: 5px 10px;
    margin: 5px;
    border-radius: 5px;
    font-size: 12px;
}
.status-normal { background-color: green; color: white; }
.status-abnormal { background-color: red; color: white; }
"""

# 모니터링 대상 목록
TARGETS = [
    {"name": "[GINI/DS]테스트", "url": "http://hugpu.ai:8888"},
    {"name": "[OFAI/DS]테스트", "url": "http://hugpu.ai:8888"},
    {"name": "[HF]테스트", "url": "http://hugpu.ai:8888"},    
    {"name": "[H100]AI 개인비서 JinJAVIS", "url": "http://hugpu.ai:8000"},
    
    {"name": "[HUGPU]FLUX 확장 이미지 생성", "url": "https://fantaxy-ofai-flx-expl-store.hf.space"},
    {"name": "[HUGPU]FLUX GIF 생성", "url": "https://aiqtech-flxgif.hf.space"},     
    {"name": "[HUGPU]LOGO 디자인 생성 ", "url": "https://fantaxy-ofai-flx-logo.hf.space"},     
    {"name": "[HUGPU]FLUX 프롬프트 생성 ", "url": "https://ginipick-flux-prompt-generator.hf.space"},     
    {"name": "[HUGPU]프롬프트+이미지로 이미지 생성 ", "url": "https://.hf.space"},  ##잘못된 URL: INDEX.HTML도 수정할것   
    {"name": "[HUGPU]LoRA: 기블리 스타일 ", "url": "https://aiqtech-flux-ghibli-studio-lora.hf.space"},  
    {"name": "[HUGPU]LoRA: 카툰 스타일 ", "url": "https://aiqtech-flxani.hf.space"},     
    {"name": "[HUGPU]FLUX 8step LoRA 이미지 생성 ", "url": "https://fantos-flx8lora.hf.space"},     
    {"name": "[HUGPU]FLUX 텍스트 TO 비디오 생성 ", "url": "https://fantos-cogvidx.hf.space"},     
    {"name": "[HUGPU]FLUX 패션모델 생성 ", "url": "https://fantos-flxfashmodel.hf.space"},     
    {"name": "[HUGPU]FLUX Controlnet ", "url": "https://fantos-flxcontrol.hf.space"},  
    {"name": "[HUGPU]FLUX NF4 이미지 생성 ", "url": "https://ginipick-ofai-flxnf4.hf.space"},     
    
    {"name": "[HUGPU] ", "url": "https://.hf.space"},     
    {"name": "[HUGPU] ", "url": "https://.hf.space"},     
    {"name": "[HUGPU] ", "url": "https://.hf.space"},     
    {"name": "[HUGPU] ", "url": "https://.hf.space"},      
    {"name": "[HUGPU] ", "url": "https://.hf.space"},     
    {"name": "[HUGPU] ", "url": "https://.hf.space"},     
    {"name": "[HUGPU] ", "url": "https://.hf.space"},     
    {"name": "[HUGPU] ", "url": "https://.hf.space"},     
    {"name": "[HUGPU] ", "url": "https://.hf.space"},  
    {"name": "[HUGPU] ", "url": "https://.hf.space"},     
    {"name": "[HUGPU] ", "url": "https://.hf.space"},     
    {"name": "[HUGPU] ", "url": "https://.hf.space"},     
    {"name": "[HUGPU] ", "url": "https://.hf.space"},     
    {"name": "[HUGPU] ", "url": "https://.hf.space"},  
    {"name": "[HUGPU] ", "url": "https://.hf.space"},     
    {"name": "[HUGPU] ", "url": "https://.hf.space"},     
    {"name": "[HUGPU] ", "url": "https://.hf.space"},     
    {"name": "[HUGPU] ", "url": "https://.hf.space"},     
    {"name": "[HUGPU] ", "url": "https://.hf.space"},  
    {"name": "[HUGPU] ", "url": "https://.hf.space"},     
    {"name": "[HUGPU] ", "url": "https://.hf.space"},     
    {"name": "[HUGPU] ", "url": "https://.hf.space"},     
    {"name": "[HUGPU] ", "url": "https://.hf.space"},     
    {"name": "[HUGPU] ", "url": "https://.hf.space"},  
    {"name": "[HUGPU] ", "url": "https://.hf.space"},     
    {"name": "[HUGPU] ", "url": "https://.hf.space"},     
    {"name": "[HUGPU] ", "url": "https://.hf.space"},     
    {"name": "[HUGPU] ", "url": "https://.hf.space"},     
    {"name": "[HUGPU] ", "url": "https://.hf.space"},  
    {"name": "[HUGPU] ", "url": "https://.hf.space"},     
    {"name": "[HUGPU] ", "url": "https://.hf.space"},     
    {"name": "[HUGPU] ", "url": "https://.hf.space"},     
    {"name": "[HUGPU] ", "url": "https://.hf.space"},     
    {"name": "[HUGPU] ", "url": "https://.hf.space"},       
]


# URL 상태 확인 함수
def check_url_status(url):
    try:
        response = requests.get(url, timeout=5)
        if response.status_code == 200:
            logging.info(f"URL 접속 성공: {url}")
            return "정상", "normal"
        else:
            logging.error(f"URL 접속 오류: {url}, 상태 코드 {response.status_code}")
            return "비정상", "abnormal"
    except requests.exceptions.RequestException as e:
        logging.exception(f"서버에 연결할 수 없습니다: {url}")
        return "비정상", "abnormal"

# 상태 업데이트 함수
def update_status():
    status_html = ""
    for target in TARGETS:
        status, status_class = check_url_status(target["url"])
        status_html += f'<span class="status-button status-{status_class}">{target["name"]}: {status}</span>'
    return status_html

# 주기적 업데이트를 위한 함수
def periodic_update(status_html):
    while True:
        time.sleep(10)  # 10초 대기
        new_status = update_status()
        status_html.update(value=new_status)

# Gradio 인터페이스 설정
def create_dashboard():
    with gr.Blocks(css=css) as app:
        gr.Image("banner.png", show_label=False)
        gr.Markdown("# 24시간 모니터링 시스템")
        gr.Markdown("## \n")
        
        status_html = gr.HTML()
        
        refresh_button = gr.Button("상태 새로고침")
        refresh_button.click(fn=update_status, outputs=status_html)

        # 초기 상태 업데이트
        app.load(fn=update_status, outputs=status_html)

        # 백그라운드에서 주기적 업데이트 시작
        threading.Thread(target=periodic_update, args=(status_html,), daemon=True).start()
    
    return app

if __name__ == "__main__":
    dashboard = create_dashboard()
    dashboard.launch()