import gradio as gr
import requests
from bs4 import BeautifulSoup

def execute(code_snippet, lang):
    lang_param = None
    
    match lang:
        case "C++": lang_param = "cpp"
        case "C#": lang_param = "cs"
        case _: lang_param = lang.lower()

    # FIXME: ERROR HANDLING
    res = requests.request("POST", f"https://try.w3schools.com/try_{lang_param}.php", data={
        "code": code_snippet
    })
    return BeautifulSoup(res.text, "html.parser").find_all("pre")[0].string

demo = gr.Interface(
    fn=execute,
    inputs=[
        gr.Textbox(
            show_label=True,
            label="Code",
            max_lines=4_294_967_295,
            lines=4_294_967_295,
            value="print('Hello, World!')",
        ),
        gr.Dropdown(
            show_label=True,
            label="Language",
            choices=["Python", "Java", "C", "C++", "C#", "PHP"],
            value="Python"
        ),
    ],
    outputs=gr.Textbox(label="Result"),
    title="HFChat Code Executor",
    description="Enter the code snippet and language that you want to execute.",
)

demo.launch(debug=True)