import spaces
import gradio as gr
import requests

def check_internet_gr(url: str):
    try:
        response = requests.get(url, timeout=30, verify=False)
        if response.status_code == 200:
            print(f"{url} is working. Status code:", response.status_code)
            gr.Info(f"{url} is working. Status code: {response.status_code}")
        else:
            print(f"Connected to {url}, but the server returned an error. Status code:", response.status_code)
            gr.Info(f"Connected to {url}, but the server returned an error. Status code: {response.status_code}")
    except requests.ConnectionError as e:
        print(f"No connection.", e)
        gr.Info(f"No connection. {e}")
    except requests.Timeout as e:
        print(f"Request timed out.", e)
        gr.Info(f"Request timed out. {e}")
    except Exception as e:
        print(f"An error occurred: {e}")
        gr.Info(f"An error occurred: {e}")

@spaces.GPU
def test(url: str):
    check_internet_gr(url)
    return ""

with gr.Blocks() as demo:
    url = gr.Textbox(label="URL", value="https://api.openai.com")
    run_button = gr.Button("Submit", variant="primary")
    info_md = gr.Markdown("<br><br><br>")

    run_button.click(test, [url], [info_md])

demo.launch()