import gradio as gr
import os
import base64

def generate_html(name, code):
    try:
        with open(f"{name}.html", "w") as f:
            f.write(code)
        return f"HTML file {name}.html generated successfully!", {"name": f"{name}.html", "data": base64.b64encode(open(f"{name}.html", "rb").read()).decode("utf-8")}
    except Exception as e:
        return f"Error generating HTML file: {str(e)}", None

demo = gr.Interface(
    fn=generate_html,
    inputs=[
        gr.Textbox(label="File Name"),
        gr.Code(label="HTML Code")
    ],
    outputs=[gr.Text(label="Status"), gr.File(label="Download HTML File")],
    title="HTML Generator"
)

demo.launch()