app.py

import gradio as gr
import requests
import logging
import time

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

css = """
footer { visibility: hidden; }
.status-normal { background-color: green; color: white; padding: 10px; border-radius: 5px; }
.status-abnormal { background-color: red; color: white; padding: 10px; border-radius: 5px; }
"""

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

# 상태 업데이트 함수
def update_status():
    while True:
        status, status_class = check_url_status()
        yield f"<div class='status-{status_class}'>AI 개인 비서 상태: {status}</div>"
        time.sleep(6)  # 6초 대기

# Gradio 인터페이스 설정
def create_dashboard():
    with gr.Blocks(css=css) as app:
        gr.Image("banner.png", show_label=False)
        gr.Markdown("# 24시간 모니터링 시스템")
        gr.Markdown("## \n")
        gr.Markdown("### 홈 설명: [링크](https://seawolf2357-bnews1.hf.space)")
        gr.Markdown("## \n")
        
        status_html = gr.HTML("AI 개인 비서 상태: 로딩 중...")
        
        gr.Loop(update_status, outputs=status_html, every=6)
    
    return app

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