Update app.py

app.py

@@ -1,13 +1,14 @@
 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!"
+        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)}"
+        return f"Error generating HTML file: {str(e)}", None
 
 demo = gr.Interface(
     fn=generate_html,
@@ -15,7 +16,7 @@ demo = gr.Interface(
         gr.Textbox(label="File Name"),
         gr.Code(label="HTML Code")
     ],
-    outputs="text",
+    outputs=[gr.Text(label="Status"), gr.File(label="Download HTML File")],
     title="HTML Generator"
 )