Update app.py

app.py

@@ -1,5 +1,6 @@
 import gradio as gr
 import subprocess
+import os
 
 def save_webpage(url, save_method, javascript):
     # Command to execute
@@ -20,12 +21,20 @@ def save_webpage(url, save_method, javascript):
         command_str = " ".join(command) + f" > {output_file}"
     
     # Execute the command
-    subprocess.run(command_str, shell=True)
+    result = subprocess.run(command_str, shell=True, capture_output=True, text=True)
     
-    with open(output_file, "r", encoding="utf-8") as file:
-        html_content = file.read()
+    # Check for warnings
+    warning_message = ""
+    if "[ WARN ]" in result.stderr:
+        warning_message = result.stderr.split("[ WARN ]")[1].strip()
     
-    return html_content
+    # Read the output HTML file
+    if os.path.exists(output_file):
+        with open(output_file, "r", encoding="utf-8") as file:
+            html_content = file.read()
+        return html_content, warning_message, output_file
+    else:
+        return "", "Error: HTML file not created", ""
 
 # Gradio Interface
 iface = gr.Interface(
@@ -35,7 +44,11 @@ iface = gr.Interface(
         gr.Radio(choices=["From URL", "From Saved File"], label="Save Method"),
         gr.Checkbox(label="Enable JavaScript"),
     ],
-    outputs=gr.HTML(),
+    outputs=[
+        gr.HTML(label="HTML Content"),
+        gr.Textbox(label="Warning/Error Messages"),
+        gr.File(label="Download HTML File")
+    ],
     title="Webpage to HTML Converter",
     description="Convert a webpage to a single HTML file for offline use."
 )