Update app.py

app.py

@@ -5,26 +5,22 @@ 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"{name}.html"  # Return the file path directly
     except Exception as e:
-        return f"Error generating HTML file: {str(e)}"
-
-def download_file(name):
-    try:
-        with open(f"{name}.html", "rb") as f:
-            return f.read()
-    except Exception as e:
-        return None
+        return None  # Returning None for the Gradio interface to handle errors gracefully
 
 def generate_and_download(name, code):
-    generate_html(name, code)
-    return download_file(name)
+    file_path = generate_html(name, code)
+    if file_path and os.path.exists(file_path):
+        return file_path  # Return the path for download
+    else:
+        return None  # Handle if the file couldn't be created
 
 demo = gr.Interface(
     fn=generate_and_download,
     inputs=[
-        gr.Textbox(label="File Name"),
-        gr.Code(label="HTML Code")
+        gr.Textbox(label="File Name", placeholder="Enter file name without extension"),
+        gr.Code(label="HTML Code", language="html")
     ],
     outputs=gr.File(label="Download HTML File"),
     title="HTML Generator"