Update app.py

app.py

@@ -1,5 +1,7 @@
 import gradio as gr
 import subprocess
+import os
+import time
 
 def save_webpage(url, save_method, javascript):
     # Command to execute
@@ -20,12 +22,27 @@ 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)
     
+    # Check for warnings and errors
+    warning_message = ""
+    if "[ WARN ]" in result.stderr:
+        warning_message = result.stderr.split("[ WARN ]")[1].strip()
+    
+    error_message = result.stderr.strip() if result.returncode != 0 else ""
+    
+    # Wait for the file to be created, with a timeout of 3 seconds
+    start_time = time.time()
+    while not os.path.exists(output_file):
+        if time.time() - start_time > 3:
+            return "", f"Error: HTML file not created within the timeout period. {error_message}", ""
+        time.sleep(0.1)
+    
+    # Read the output HTML file
     with open(output_file, "r", encoding="utf-8") as file:
         html_content = file.read()
     
-    return html_content
+    return html_content, warning_message if warning_message else error_message, output_file
 
 # Gradio Interface
 iface = gr.Interface(
@@ -35,7 +52,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."
 )