import gradio as gr
import aiohttp
import asyncio
import threading

# 비동기적으로 외부 URL에 접속하는 함수를 정의합니다.
async def check_connection(url):
    try:
        async with aiohttp.ClientSession() as session:
            async with session.get(url) as response:
                status = f"URL: {url} 상태 코드: {response.status}, 접속 상태: {'접속 성공' if response.status == 200 else '접속 실패'}"
                print(status)
                return status
    except Exception as e:
        status = f"URL: {url} 접속 실패: {str(e)}"
        print(status)
        return status

# 타이머를 사용하여 정해진 주기로 비동기 함수를 반복 실행합니다.
async def start_timer(urls, interval):
    while True:
        for url in urls:
            if url:  # URL이 비어 있지 않으면 체크
                await check_connection(url)
                await asyncio.sleep(1)  # 각 URL 체크 후 1초 대기
        await asyncio.sleep(interval - len(urls))  # 모든 URL 체크 후 interval에서 이미 소비된 시간을 뺀 만큼 대기

# 타이머 시작 함수를 Gradio의 입력과 함께 연결합니다.
def setup_timer(interval, *urls):
    interval_seconds = interval * 60  # 분을 초로 변환
    # 비동기 작업을 별도의 스레드에서 실행
    threading.Thread(target=asyncio.run, args=(start_timer(urls, interval_seconds),)).start()
    return "타이머가 설정되었습니다."

# Gradio UI 컴포넌트를 정의합니다.
url_inputs = [gr.Text(label=f"URL {i+1}", placeholder=f"접속할 URL {i+1}을 입력하세요") for i in range(40)]
interval_input = gr.Slider(minimum=1, maximum=60, step=1, value=5, label="접속 주기(분)")

# Gradio 앱 설정
app = gr.Interface(
    fn=setup_timer,
    inputs=[interval_input] + url_inputs,
    outputs="text",
    title="URL 접속 체커",
    description="최대 40개의 URL과 접속 주기를 입력하고 '시작' 버튼을 클릭하세요. 지정된 주기로 HTTP 상태 코드 및 접속 상태를 확인합니다.",
    examples=[
        [
            15,
            "https://fantaxy-timer1.hf.space",  
#            "https://seawolf2357-FastGPT.hf.space",
#            "https://seawolf2357-NaverTalk.hf.space",
            "https://fantaxy-kai-llm-claude35.hf.space",
            "https://fantaxy-kai-llm-gpt4o.hf.space",
            "https://fantaxy-kai-llm-l370b.hf.space",
            "https://fantaxy-kai-llm-l38b.hf.space",
            "https://fantaxy-kai-llm-phi3.hf.space",
            "https://fantaxy-kai-llm-mix87.hf.space",
            "https://fantaxy-kai-llm-aya.hf.space",
            "https://fantaxy-kai-llm-zepr.hf.space",
            "https://fantaxy-kai-llm-yi.hf.space",
            "https://ginipick-kai-llm-law1.hf.space",
 #           "https://fantaxy-kai-allsend.hf.space",
            "https://fantaxy-kai-charac-yok.hf.space",
            "https://fantaxy-kai-charac-mbti.hf.space",
            "https://fantaxy-kai-charac-girlfriend.hf.space",
            "https://fantaxy-kai-charac-ana.hf.space",
            "https://fantaxy-kai-charac-pluting.hf.space",
            "https://fantaxy-kai-charac-simsim.hf.space",
            "https://fantaxy-kai-charac-boyfriend.hf.space",   
            "https://fantaxy-kai-llm-persona.hf.space",
            "https://fantaxy-kai-llm-trans.hf.space",
            "https://fantaxy-kai-llm-code.hf.space",
            "https://fantaxy-kai-llm-novel.hf.space",
            "https://ginipick-gini-ytb-private-replysend.hf.space",
            "https://seawolf2357-kai-saju.hf.space",
            "https://ginipick-gini-saju.hf.space",
            "https://fantos-opencodeai.hf.space",
            "https://seawolf2357-kai-l3-405b.hf.space",
            "https://ginipick-gini-api-chat.hf.space",
            "https://ginipick-gini-api-blog.hf.space",
            "https://ginipick-gini-api-script.hf.space",
            "https://ginipick-gini-api-market.hf.space",
            "https://seawolf2357-kai-llm-mislarge2.hf.space",
            "https://seawolf2357-kai-llm-upstage.hf.space",
            "https://seawolf2357-kai-llm-medi.hf.space",
            "https://seawolf2357-kai-llm-pharm.hf.space",
            "https://seawolf2357-kai-llm-insu.hf.space",
            "https://seawolf2357-kai-llm-search.hf.space",
            "https://seawolf2357-kai-llm-copy.hf.space",
            "https://ginipick-EXAONE-3.0-7.8B-Instruct.hf.space",
            "https://fantaxy-kai-llm-math.hf.space"
        ]
    ],
    cache_examples=False  # 캐시 비활성화
)

if __name__ == "__main__":
    app.launch()